diff options
| -rw-r--r-- | python/ack_checker.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/python/ack_checker.py b/python/ack_checker.py index 3c2269ce..fbfdd9ff 100644 --- a/python/ack_checker.py +++ b/python/ack_checker.py @@ -1,5 +1,6 @@ import codecs import datetime +import re from csv import reader from io import BytesIO from remote import FtpClient @@ -20,7 +21,8 @@ def ack_check(date: datetime.date, conn): for serenitas_id, action, dealtype, result, globeop_id, _, _ in csv: if action == "NEW" and result == "Loaded": globeop_id = int(globeop_id) - serenitas_id = int(serenitas_id[5:]) + if m := re.match("[^0-9]*([0-9]*)", serenitas_id): + serenitas_id = int(m.groups()[0]) if dealtype == "CreditDefaultSwapDeal": with conn.cursor() as c: c.execute( @@ -38,7 +40,7 @@ def ack_check(date: datetime.date, conn): if dealtype == "SwaptionDeal": with conn.cursor() as c: c.execute( - "UPDATE swaptions SET globeop_id=%s WHERE dealid=%s", + "UPDATE swaptions SET globeop_id=%s WHERE id=%s", (globeop_id, serenitas_id), ) conn.commit() |