aboutsummaryrefslogtreecommitdiffstats
path: root/python
diff options
context:
space:
mode:
Diffstat (limited to 'python')
-rw-r--r--python/collateral_calc.py13
1 files changed, 11 insertions, 2 deletions
diff --git a/python/collateral_calc.py b/python/collateral_calc.py
index 603e9668..d1279ca8 100644
--- a/python/collateral_calc.py
+++ b/python/collateral_calc.py
@@ -40,7 +40,14 @@ def download_baml_files(d=None):
DATA_DIR = DAILY_DIR / "BAML_reports"
sftp = get_baml_sftp_client()
for f in sftp.listdir('outgoing'):
- sftp.get(f"outgoing/{f}", localpath= DATA_DIR / f)
+ sftp.get(f"outgoing/{f}", localpath=DATA_DIR / f)
+
+def download_wells_files(d=None):
+ DATA_DIR = DAILY_DIR / "Wells_reports"
+ sftp = get_wells_sftp_client()
+ base_dir = "/RECEIVE/339425_DATO2"
+ for f in sftp.listdir(base_dir):
+ sftp.get(f"{base_dir}/{f}", localpath=DATA_DIR / f)
def download_sftp_files(d=None,
@@ -395,10 +402,12 @@ if __name__ == "__main__":
download_ms_emails()
download_gs_emails()
download_citi_emails()
- d = (pd.Timestamp.today() - BDay()).normalize()
+
#download_sftp_files(d)
download_baml_files()
+ download_wells_files()
dawn_trades = get_dawn_trades()
+ d = (pd.Timestamp.today() - BDay()).normalize()
try:
df_ms = ms_collateral(d, dawn_trades)
except FileNotFoundError as e: