diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/globeop.py | 3 | ||||
| -rw-r--r-- | python/task_server/globeop.py | 2 |
2 files changed, 3 insertions, 2 deletions
diff --git a/python/globeop.py b/python/globeop.py index 80515b83..e0c56fbb 100644 --- a/python/globeop.py +++ b/python/globeop.py @@ -2,6 +2,7 @@ import logging import argparse import pandas as pd from task_server.globeop import download_data, upload_data +from sqlalchemy import create_engine logging.basicConfig(filename='/home/share/CorpCDOs/logs/globeop.log', level=logging.INFO, @@ -23,4 +24,4 @@ if args.download: download_data(args.date) elif args.upload: engine = create_engine('postgresql://dawn_user@debian/dawndb') - upload_data(args.date) + upload_data(engine, args.date) diff --git a/python/task_server/globeop.py b/python/task_server/globeop.py index 47cede1d..e123933f 100644 --- a/python/task_server/globeop.py +++ b/python/task_server/globeop.py @@ -106,7 +106,7 @@ def download_data(workdate): df.to_csv(os.path.join(reports_dir, "CDS_Report.csv"), index=False)
os.remove(os.path.join(reports_dir, "CDS_Report.xls"))
-def upload_data(workdate):
+def upload_data(engine, workdate):
df = pd.read_sql_query("SELECT * from list_marks(%s)", engine, params = (workdate.date(),))
df.rename(columns = {'identifier': 'IDENTIFIER',
'price': 'Price'}, inplace=True)
|