diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/load_globeop_report.py | 2 | ||||
| -rw-r--r-- | python/task_server/globeop.py | 5 |
2 files changed, 4 insertions, 3 deletions
diff --git a/python/load_globeop_report.py b/python/load_globeop_report.py index 01ffb5a7..d45c3f85 100644 --- a/python/load_globeop_report.py +++ b/python/load_globeop_report.py @@ -19,7 +19,7 @@ def get_globs(fname, years=['2013', '2014', '2015', '2016', '2017']): return globs def read_valuation_report(f): - date = pd.Timestamp(f.rsplit('/', 3)[1]) + date = pd.Timestamp(f.parts[4]) if date >= pd.Timestamp('2013-02-06'): df = pd.read_csv(f, parse_dates=['KnowledgeDate', 'PeriodEndDate']) else: diff --git a/python/task_server/globeop.py b/python/task_server/globeop.py index 272d03f6..7a26f170 100644 --- a/python/task_server/globeop.py +++ b/python/task_server/globeop.py @@ -97,8 +97,9 @@ def download_data(workdate): reports_dir.mkdir(parents=True) for filename in available_files: - ftp.retrbinary('RETR ' + filename, (reports_dir / filename).write_bytes) - logger.info(f"downloaded {filename}") + with (reports_dir / filename).open("wb") as fh: + ftp.retrbinary('RETR ' + filename, fh.write) + logger.info(f"downloaded {filename}") gpg = get_gpg() for filename in available_files: |