diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/collateral/baml_isda.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/python/collateral/baml_isda.py b/python/collateral/baml_isda.py index 34e7bb67..5ddfd8da 100644 --- a/python/collateral/baml_isda.py +++ b/python/collateral/baml_isda.py @@ -73,11 +73,22 @@ def download_files(em, d=None, count=20, *, fund="Serenitas", **kwargs): if ( msg.sender.name == "us_otc_client_valuation@baml.com" or msg.sender.name == "us_otc_client_valuation@bofa.com" - or msg.sender.name - == "bank_of_america_collateral_operations@bankofamerica.com" ): + soup = BeautifulSoup(msg.body, features="lxml") + a = soup.find("a") + url = urlsplit(a["href"]) + query = parse_qs(url.query) + base_url = urlunsplit(url[:2] + ("",) * 3) + try: + download_from_secure_id( + query["id"][0], query["brand"][0], DATA_DIR, base_url + ) + except ValueError as e: + logger.error(e) + continue + if msg.sender.name == "bank_of_america_collateral_operations@bankofamerica.com": for attach in msg.attachments: - if attach.name.endswith("xls") or attach.name.endswith("pdf"): + if attach.name.endswith("xls"): p = DATA_DIR / attach.name if not p.exists(): p.write_bytes(attach.content) |