aboutsummaryrefslogtreecommitdiffstats
path: root/python/report_ops
diff options
context:
space:
mode:
Diffstat (limited to 'python/report_ops')
-rw-r--r--python/report_ops/cash.py6
-rw-r--r--python/report_ops/custodians.py6
2 files changed, 6 insertions, 6 deletions
diff --git a/python/report_ops/cash.py b/python/report_ops/cash.py
index 3bebbc5a..6e8aff87 100644
--- a/python/report_ops/cash.py
+++ b/python/report_ops/cash.py
@@ -8,7 +8,6 @@ from .custodians import NT, UMB, BNY, SCOTIA
from functools import partial
from dataclasses import dataclass
-from selenium.common.exceptions import StaleElementReferenceException
from serenitas.utils.env import DAILY_DIR
@@ -144,10 +143,7 @@ class SeleneScotiaCashReport(
self._staging_queue.clear()
def get_cash_report(self, prefix=None):
- try:
- self.download_reports(self.date)
- except StaleElementReferenceException:
- raise ValueError(f"File is not ready to download yet")
+ self.download_reports(self.date)
REPORT_DIR = DAILY_DIR / "Selene" / "Scotia_reports"
try:
return next(
diff --git a/python/report_ops/custodians.py b/python/report_ops/custodians.py
index babbf85d..288f81f6 100644
--- a/python/report_ops/custodians.py
+++ b/python/report_ops/custodians.py
@@ -2,6 +2,7 @@ import datetime
from typing import ClassVar
from dataclasses import dataclass
import gpg
+from selenium.common.exceptions import StaleElementReferenceException
from serenitas.utils.exchange import ExchangeMessage, FileAttachment
from serenitas.utils.env import DAILY_DIR
@@ -206,4 +207,7 @@ class BNY(Custodian, account="BONY2"):
class SCOTIA(Custodian, account="SCOTIA"):
@staticmethod
def download_reports(date=datetime.date.today()):
- return download_scotia_report(date)
+ try:
+ return download_scotia_report(date)
+ except StaleElementReferenceException:
+ raise ValueError(f"Scotia report not ready yet for {date}")