diff options
Diffstat (limited to 'python/bowdst_wire.py')
| -rw-r--r-- | python/bowdst_wire.py | 24 |
1 files changed, 21 insertions, 3 deletions
diff --git a/python/bowdst_wire.py b/python/bowdst_wire.py index b53e4624..8153ecb7 100644 --- a/python/bowdst_wire.py +++ b/python/bowdst_wire.py @@ -1,6 +1,7 @@ from bowdst import get_dir, download_messages from csv import DictReader -from citco_ops.bowdst import BowdstWire +from citco_ops.bowdst import BowdstWire, IsoselWire +from citco_ops.cash import IsoselCashReport, dt_from_fname as nt_key import datetime @@ -10,7 +11,7 @@ def dt_from_fname(f): ) -def load_wire_report(workdate: datetime.date): +def load_bowdst_wire_report(workdate: datetime.date): p = max( [f for f in get_dir(workdate).iterdir() if "BowdstWires" in f.name], key=dt_from_fname, @@ -24,6 +25,21 @@ def load_wire_report(workdate: datetime.date): BowdstWire.commit() +def load_isosel_wire_report(workdate: datetime.date): + p = max( + [f for f in get_dir(workdate).iterdir() if "custodian_wires" in f.name], + key=nt_key, + ) + if not p: # No files available + return + with open(p) as fh: + reader = DictReader(fh) + for line in reader: + if "sponsor" in line["narrative"].lower(): + IsoselWire.from_passport_line(line).stage() + IsoselWire.commit() + + if __name__ == "__main__": import argparse from serenitas.utils.exchange import ExchangeMessage @@ -39,4 +55,6 @@ if __name__ == "__main__": args = parser.parse_args() em = ExchangeMessage() download_messages(em) - load_wire_report(args.workdate) + IsoselCashReport.download_reports(args.workdate) + load_bowdst_wire_report(args.workdate) + load_isosel_wire_report(args.workdate) |