diff options
Diffstat (limited to 'python/report_ops')
| -rw-r--r-- | python/report_ops/custodians.py | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/python/report_ops/custodians.py b/python/report_ops/custodians.py index 76d01d6b..8da504d7 100644 --- a/python/report_ops/custodians.py +++ b/python/report_ops/custodians.py @@ -14,14 +14,9 @@ from dataclasses import dataclass class Custodian: date: datetime.date account: ClassVar[str] - upload_fun: ClassVar - def __init_subclass__(cls, account, upload_fun=None): + def __init_subclass__(cls, account): cls.account = account - cls.upload_fun = staticmethod(upload_fun) - - def upload_to_custodian(self, upload): - self.upload_fun(upload, account=self.account, trade_date=self.date) class NT(Custodian, account="NT"): @@ -49,7 +44,7 @@ class NT(Custodian, account="NT"): csvFile.write(text) -class UMB(Custodian, account="UMB", upload_fun=upload_to_umb): +class UMB(Custodian, account="UMB"): @classmethod def download_reports(cls, date=datetime.date.today()): em = ExchangeMessage() @@ -115,5 +110,5 @@ class BNY(Custodian, account="BONY2"): p.write_bytes(attach.content) -class BBH(Custodian, account="BBH", upload_fun=upload_to_bbh): +class BBH(Custodian, account="BBH"): pass |