aboutsummaryrefslogtreecommitdiffstats
path: root/python/report_ops/custodians.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/report_ops/custodians.py')
-rw-r--r--python/report_ops/custodians.py23
1 files changed, 9 insertions, 14 deletions
diff --git a/python/report_ops/custodians.py b/python/report_ops/custodians.py
index 4524e14f..bb939e86 100644
--- a/python/report_ops/custodians.py
+++ b/python/report_ops/custodians.py
@@ -1,14 +1,15 @@
-from serenitas.utils.exchange import ExchangeMessage, FileAttachment
-from serenitas.utils.env import DAILY_DIR
-import warnings
import datetime
-from .misc import get_dir, _recipients, _cc_recipients
-from .scotia import download_scotia_report
+from typing import ClassVar
+from dataclasses import dataclass
import gpg
+
+from serenitas.utils.exchange import ExchangeMessage, FileAttachment
+from serenitas.utils.env import DAILY_DIR
from serenitas.ops.trade_dataclasses import BondDeal
from serenitas.ops.funds import Service
-from typing import ClassVar
-from dataclasses import dataclass
+
+from .misc import get_dir, _recipients, _cc_recipients, em_date_filter
+from .scotia import download_scotia_report
def upload_to_custodian(account, trade_date, upload):
@@ -105,13 +106,7 @@ class Custodian:
@staticmethod
def em_date_filter(em, date):
- start = datetime.datetime.combine(date, datetime.time.min).replace(
- tzinfo=em._account.default_timezone
- )
- end = datetime.datetime.combine(date, datetime.time.max).replace(
- tzinfo=em._account.default_timezone
- )
- return {"datetime_received__gte": start, "datetime_received__lte": end}
+ return em_date_filter(em, date, date)
class NT(Custodian, account="NT"):