aboutsummaryrefslogtreecommitdiffstats
path: root/python/Dawn/utils.py
blob: 8fa45e93a7374d0a1c56c650d70f7211464648a4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import pandas as pd
import os
import re
from pickle import dumps

def bump_rev(filename):
    pattern = "([^r]*)(\srev(\d)|).pdf"
    begin, _, rev_number =  re.match(pattern, filename).groups()
    rev_number = int(rev_number) + 1 if rev_number else 1
    return "{0} rev{1}.pdf".format(begin, rev_number)

def simple_serialize(obj):
    d = {c.name: getattr(obj, c.name) for c in  obj.__table__.columns}
    # add data from foreign relationship
    if obj.__table__.name == 'wires':
        d.update({'Counterparty': obj.account.counterparty,
                 'cashaccount': obj.account.cash_account,
                 'custodian': obj.account.custodian})
    return dumps(d)