diff options
| -rw-r--r-- | python/citco_ops/utils.py | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/python/citco_ops/utils.py b/python/citco_ops/utils.py index 1a4c467a..0ada5f47 100644 --- a/python/citco_ops/utils.py +++ b/python/citco_ops/utils.py @@ -28,7 +28,7 @@ def next_business_days(date, offset): def get_file_status(s): orig_name, submit_date, status, process_date = re.match( - "([^\d]*)(\d*)-(PROCESSED|FAILED)_([^-]*)", s + r"([^\d]*)(\d*)-(PROCESSED|FAILED)_([^-]*)", s ).groups() zone = ZoneInfo("America/New_York") submit_date = datetime.datetime.strptime(submit_date, "%Y%m%d%H%M%S").replace( @@ -95,14 +95,14 @@ class CitcoSubmission(Deal, deal_type=None, table_name="citco_submission2"): with cls._sftp.client.open(fname) as fh: next(fh) for row in csv.reader(fh): - try: - trade = cls( - "failed", row[-1], row[2], submit_date, process_date - ) - except IndexError: - trade = cls( - "failed", row[-1], row[-1], submit_date, process_date - ) + id_or_error = row[2] if len(row) > 2 else row[-1] + trade = cls( + "failed", + row[-1], + id_or_error, + submit_date, + process_date, + ) trade.stage() @classmethod |