diff options
Diffstat (limited to 'python/task_server/globeop.py')
| -rw-r--r-- | python/task_server/globeop.py | 30 |
1 files changed, 19 insertions, 11 deletions
diff --git a/python/task_server/globeop.py b/python/task_server/globeop.py index 7d0d3f39..295509c0 100644 --- a/python/task_server/globeop.py +++ b/python/task_server/globeop.py @@ -1,6 +1,5 @@ import os
import os.path
-import datetime
from ftplib import FTP
import gnupg
import config
@@ -44,10 +43,24 @@ def key_fun(s): def run_date(s):
return datetime.datetime.strptime(s.split("_")[2], "%Y%m%d.%H%M%S")
-def download_data(workdate):
+def get_ftp(folder):
ftp = FTP('ftp.globeop.com')
ftp.login('srntsftp', config.ftp_password)
- ftp.cwd('outgoing')
+ ftp.cwd(folder)
+ return ftp
+
+def get_gpg():
+ if os.name=='nt':
+ gpg = gnupg.GPG(gpgbinary = r'"c:\\Program Files (x86)\\GNU\\GnuPG\\gpg2.exe"',
+ gnupghome = os.path.join(os.getenv('APPDATA'), "gnupg"))
+ elif os.name == 'posix':
+ gpg = gnupg.GPG(gnupghome = '/home/guillaume/.gnupg')
+ gpg.encoding = 'utf8'
+ return gpg
+
+def download_data(workdate):
+ ftp = get_ftp('outgoing')
+ workdate = workdate.date()
files = ftp.nlst()
pnlfiles = [filename for filename in files if "csv" in filename and \
"Profit" in filename if get_ped(filename) < workdate]
@@ -76,12 +89,7 @@ def download_data(workdate): ftp.retrbinary('RETR ' + filename, fh.write)
logging.info("downloaded {0}".format(filename))
- if os.name=='nt':
- gpg = gnupg.GPG(gpgbinary = r'"c:\\Program Files (x86)\\GNU\\GnuPG\\gpg2.exe"',
- gnupghome = os.path.join(os.getenv('APPDATA'), "gnupg"))
- elif os.name == 'posix':
- gpg = gnupg.GPG(gnupghome = '/home/guillaume/.gnupg')
- gpg.encoding = 'utf8'
+ gpg = get_gpg()
for filename in available_files:
if "Profit" in filename:
newfilename = "Pnl.csv"
@@ -145,8 +153,8 @@ if __name__=="__main__": help="download reports from GlobeOp")
group.add_argument("-u", "--upload", action="store_true",
help="upload marks to GlobeOp")
- parser.add_argument("date", nargs='?', type=date_from_string,
- default=datetime.date.today())
+ 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:
|