diff options
Diffstat (limited to 'python/Dawn')
| -rw-r--r-- | python/Dawn/utils.py | 3 | ||||
| -rw-r--r-- | python/Dawn/views.py | 10 |
2 files changed, 11 insertions, 2 deletions
diff --git a/python/Dawn/utils.py b/python/Dawn/utils.py index 9a2cf37f..731d6dbb 100644 --- a/python/Dawn/utils.py +++ b/python/Dawn/utils.py @@ -9,11 +9,12 @@ def bump_rev(filename): return f"{begin} rev{rev_number}.pdf" -def simple_serialize(obj): +def simple_serialize(obj, **kwargs): 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}) + d.update(kwargs) return dumps(d) diff --git a/python/Dawn/views.py b/python/Dawn/views.py index 086d7d99..1bda06e1 100644 --- a/python/Dawn/views.py +++ b/python/Dawn/views.py @@ -365,8 +365,16 @@ def trade_manage(tradeid, kind): trade.cashaccount = trade.fcm_account.cash_account session.commit() if form.upload_globeop.data: + if kind == "bond": + buf = simple_serialize(trade, upload=True) + else: + buf = simple_serialize(trade) q = get_queue() - q.rpush(f'{kind}_trades', simple_serialize(trade)) + q.rpush(f"{kind}_trades", buf) + else: + if kind == "bond": + q = get_queue() + q.rpush(f"{kind}_trades", simple_serialize(trade)) return redirect(url_for('list_trades', kind=kind)) else: if form.errors: |