advanced dstruct usage

In [1]:

    

from dstruct import DataStructFromJSON, DataField, datafield


    

In [2]:

    

# taken from the basic example
class AccountSummaryFromFile(DataStructFromJSON):
    
    user = DataField()
    type = DataField('account', 'account-type')
    ballance = DataField('account', 'account-ballance')
    # you can pass functions under the keyword "parser" to parse raw data parsing
    account_number = DataField('account', 'account-number', parser=lambda s: 'X'*len(s[:-4])+s[-4:])


    

In [3]:

    

AccountSummaryFromFile('data_files/bank_data.json')


    

    
Out[3]:





{"ballance": 1234.56, "type": "checking", "user": "John F. Doe", "account_number": "XXXXX6789"}

In [4]:

    

from dstruct import DataStruct
from datetime import datetime


    

In [5]:

    

class Transaction(DataStruct):
    
    amount = DataField()

    @datafield(path=None)
    def time(self, data):
        s = data['utc-unix']
        tz = data['time-zone']
        if 'UTC' not in tz:
            raise ValueError("Unknow time-zone standard: '%s'" % tz)
        else:
            dif = tz.replace('UTC', '')
            # trick for adding time-zone
            s = eval(str(s)+dif+'*60*60')

        dt = datetime.fromtimestamp(s)
        # return the parsed epoch time
        # for the given time-zone
        return dt.strftime('%Y-%m-%d %H:%M:%S')
    
    @datafield('source')
    def source(self, data):
        t = data['type']
        # we use the transaction type
        # to identify which kind of
        # `DataStruct` should be used
        if '-' in t:
            s = ''
            for sub in t.split('-'):
                s += sub.capitalize()
        else:
            s = t.capitalize()
        
        # we use `eval` to grab
        # the appropriate struct
        cls = eval(s)
        return cls(data)
    

class TransactionSource(DataStruct):
    
    ref = DataField()
    
class Purchase(TransactionSource):

    type = DataField()
    at = DataField('name')
    card = DataField(parser=lambda s: 'X'*len(s[:-4])+s[-4:])
    
class MobileDeposit(TransactionSource):

    type = DataField()
    note = DataField()
    check_number = DataField('check-number')


class DetailedAccountSummary(AccountSummaryFromFile):
    
    last_withdraw = DataField('account', 'withdrawn', '0', parser=Transaction)
    last_deposit = DataField('account', 'deposited', '0', parser=Transaction)


    

In [6]:

    

print(DetailedAccountSummary('data_files/bank_data.json'))


    

    




{
    "account_number": "XXXXX6789",
    "ballance": 1234.56,
    "last_deposit": {
        "amount": 1057.21,
        "source": {
            "check_number": "1229361",
            "note": "bi-weekly paycheck",
            "ref": " #IB3GFRZG31",
            "type": "mobile-deposit"
        },
        "time": "2016-03-08 06:34:51"
    },
    "last_withdraw": {
        "amount": 23.03,
        "source": {
            "at": "Average-Restaurant",
            "card": "XXXXXXXXXXXX1112",
            "ref": "S567013305806010",
            "type": "purchase"
        },
        "time": "2016-03-08 06:34:51"
    },
    "type": "checking",
    "user": "John F. Doe"
}

In [ ]: