f2b9decf27
- Made the storage model slightly more flexible - Made a small P-o-C GUI application in GTK - Polished accounting.client - models.Transaction.id is now a str - Fixed transaction.id marshalling for storage.ledgercli
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from datetime import datetime
|
|
|
|
from flask import json
|
|
|
|
from accounting.models import Amount, Transaction, Posting, Account
|
|
|
|
|
|
class AccountingEncoder(json.JSONEncoder):
|
|
def default(self, o):
|
|
if isinstance(o, Account):
|
|
return dict(
|
|
__type__=o.__class__.__name__,
|
|
name=o.name,
|
|
amounts=o.amounts,
|
|
accounts=o.accounts
|
|
)
|
|
elif isinstance(o, Transaction):
|
|
return dict(
|
|
__type__=o.__class__.__name__,
|
|
id=o.id,
|
|
date=o.date.strftime('%Y-%m-%d'),
|
|
payee=o.payee,
|
|
postings=o.postings,
|
|
metadata=o.metadata
|
|
)
|
|
elif isinstance(o, Posting):
|
|
return dict(
|
|
__type__=o.__class__.__name__,
|
|
account=o.account,
|
|
amount=o.amount,
|
|
metadata=o.metadata
|
|
)
|
|
elif isinstance(o, Amount):
|
|
return dict(
|
|
__type__=o.__class__.__name__,
|
|
amount=str(o.amount),
|
|
symbol=o.symbol
|
|
)
|
|
elif isinstance(o, Exception):
|
|
return dict(
|
|
__type__=o.__class__.__name__,
|
|
args=o.args
|
|
)
|
|
|
|
return json.JSONEncoder.default(self, o)
|
|
|
|
|
|
class AccountingDecoder(json.JSONDecoder):
|
|
def __init__(self):
|
|
json.JSONDecoder.__init__(self, object_hook=self.dict_to_object)
|
|
|
|
def dict_to_object(self, d):
|
|
if '__type__' not in d:
|
|
return d
|
|
|
|
types = {c.__name__: c for c in [Amount, Transaction, Posting,
|
|
Account]}
|
|
|
|
_type = d.pop('__type__')
|
|
|
|
if _type == 'Transaction':
|
|
d['date'] = datetime.strptime(d['date'], '%Y-%m-%d')
|
|
|
|
return types[_type](**d)
|