diff options
Diffstat (limited to 'python/upload_bbh_trades.py')
| -rw-r--r-- | python/upload_bbh_trades.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/python/upload_bbh_trades.py b/python/upload_bbh_trades.py index 69cba8df..3f0d5e29 100644 --- a/python/upload_bbh_trades.py +++ b/python/upload_bbh_trades.py @@ -50,10 +50,22 @@ def _serialize(obj): if __name__ == "__main__": + import argparse + + parser = argparse.ArgumentParser(description="Upload trades to BBH") + parser.add_argument( + "date", + nargs="?", + type=datetime.date.fromisoformat, + default=(datetime.date.today() - datetime.timedelta(days=7)), + ) + args = parser.parse_args() conn = dbconn("dawndb") with conn.cursor() as c: - sql_query = "SELECT bond_trades.*, counterparties.dtc_number FROM bond_trades LEFT JOIN counterparties ON cp_code=code WHERE cash_counterparty AND trade_date >= '2022-04-05' AND fund='BRINKER' and faceamount is not null and faceamount >0;" - c.execute(sql_query) # We don't want to upload trades before 2022-04-05 + sql_query = "SELECT bond_trades.*, counterparties.dtc_number FROM bond_trades LEFT JOIN counterparties ON cp_code=code WHERE cash_counterparty AND trade_date >= '2022-04-05' AND trade_date >=%s AND fund='BRINKER' and faceamount is not null and faceamount >0;" + c.execute( + sql_query, (args.date,) + ) # We don't want to upload trades before 2022-04-05 so we're filtering on the trade_date twice for row in c: obj = row._asdict() |