aboutsummaryrefslogtreecommitdiffstats
path: root/python/parse_emails.py
blob: aed452259bea2cc54c81407c05ad4987cb1f68e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import pandas as pd
import re
from pathlib import Path

emails = [f for f in Path("../../data/swaptions").iterdir() if f.is_file()]

def makedf(r, indextype, ref):
    if indextype=='IG':
        cols = ['Strike', 'RecBid', 'RecOffer', 'DeltaRec', 'PayBid',
                'PayOffer', 'DeltaPay', 'Vol', 'Gamma']
    else:
        cols = ['Strike', 'RecBid', 'RecOffer', 'DeltaRec', 'PayBid',
                'PayOffer', 'DeltaPay', 'Vol', 'PxVol', 'Gamma']
    df = pd.DataFrame.from_records(r, columns = cols)
    df['ref'] = ref
    for col in ['DeltaRec', 'DeltaPay', 'Vol', 'PxVol', 'Gamma']:
        if col in df:
            df[col] = df[col].str.strip("%").astype('float')/100
    df = df.convert_objects(convert_numeric=True)
    df.set_index('Strike', inplace=True)
    return df

masterdf = {}
for f in emails:
    with f.open("rb") as fh:
        subject = fh.readline()
        m = re.match("(?:Fwd:)?(\w{2})([0-9]{1,2})\s", subject.decode('utf-8'))
        if m:
            indextype, series = m.groups()
            series = int(series)
        else:
            print("can't parse subject line for {0}".format(f))
            print(subject.decode("utf-8"))
            continue
        flag = False
        allexpiriesdf = {}
        for line in fh:
            line = line.decode('utf-8', 'ignore')
            line = line.rstrip()
            if line.startswith("At"):
                quotedate = pd.to_datetime(line[4:])
                if quotedate >= pd.to_datetime('2015-04-01'):
                    continue
            if line.startswith("Ref"):
                m = re.match("Ref:(\S+)\s+Fwd(?: Spd)?:(\S+)\s+Fwd Bpv:(\S+)\s+Expiry:(\S+)", line)
                if m:
                    ref, fwspread, fwbpv, expiry = m.groups()
                    expiry = pd.datetime.strptime(expiry, '%d-%b-%y')
                else:
                    print("something wrong with {0}".format(f))
                continue
            if line.startswith("Strike"):
                if "Px Vol" in line:
                    indextype='HY'
                else:
                    indextype='IG'
                flag = True
                r = []
                continue
            if flag:
                if line:
                    line = re.sub("[/|]", " ", line)
                    vals = re.sub(" +", " ", line).rstrip().split(" ")
                    r.append(vals)
                    continue
                else:
                    allexpiriesdf[expiry] = makedf(r, indextype, ref)
                    flag = False
                    r = []
                    continue
    if flag:
        allexpiriesdf[expiry] = makedf(r, indextype, ref)
    masterdf[(quotedate, indextype, series)] = pd.concat(allexpiriesdf, names=['expiry', 'Strike'])
masterdf = pd.concat(masterdf, names=['indextype', 'series', 'quotedate'])
masterdf.to_hdf('swaptions.hdf', key='swaptions')