aboutsummaryrefslogtreecommitdiffstats
path: root/python/task_server
diff options
context:
space:
mode:
Diffstat (limited to 'python/task_server')
-rw-r--r--python/task_server/globeop.py26
1 files changed, 1 insertions, 25 deletions
diff --git a/python/task_server/globeop.py b/python/task_server/globeop.py
index afd6a194..47cede1d 100644
--- a/python/task_server/globeop.py
+++ b/python/task_server/globeop.py
@@ -2,10 +2,9 @@ import os
import os.path
from ftplib import FTP
import gnupg
-import task_server.config
+from . import config
import re
import logging
-import argparse
import shutil
import pandas as pd
from sqlalchemy import create_engine
@@ -118,26 +117,3 @@ def upload_data(workdate):
with open(fullpath, "rb") as fh:
ftp.storbinary('STOR ' + filename, fh)
logging.info("upload done")
-
-if __name__=="__main__":
- logging.basicConfig(filename='/home/share/CorpCDOs/logs/globeop.log',
- level=logging.INFO,
- format='%(asctime)s %(message)s')
-
- parser = argparse.ArgumentParser()
- group = parser.add_mutually_exclusive_group(required=True)
- ## options are technically not exclusive, but we will be running them
- ## at different times of the day
- group.add_argument("-d", "--download", action="store_true",
- help="download reports from GlobeOp")
- group.add_argument("-u", "--upload", action="store_true",
- help="upload marks to GlobeOp")
- parser.add_argument("date", nargs='?', type=lambda s: pd.datetime.strptime(s, "%Y-%m-%d"),
- default=pd.datetime.today())
- args = parser.parse_args()
-
- if args.download:
- download_data(args.date)
- elif args.upload:
- engine = create_engine('postgresql://dawn_user@debian/dawndb')
- upload_data(args.date)