diff options
Diffstat (limited to 'python/process_queue.py')
| -rw-r--r-- | python/process_queue.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/python/process_queue.py b/python/process_queue.py index f9ae6d2a..9f0e6f47 100644 --- a/python/process_queue.py +++ b/python/process_queue.py @@ -179,21 +179,23 @@ def generate_csv(l, queue_name='bond_trades'): else: return output.getvalue() +def get_filename(timestamp, queue_name): + d = {'bond_trades':'Mortgages', + 'cds_trades':'CreditDefaultSwapDeal'} + return 'Serenitas.ALL.{0:%Y%m%d.%H%M%S}.{1}.csv'.format(timestamp, d[queue_name]) + def upload_file(timestamp, queue_name='bond_trades'): ftp = FTP('ftp.globeop.com') ftp.login('srntsftp', config.ftp_password) ftp.cwd('incoming') - filename = 'Serenitas.ALL.{0:%Y%m%d.%H%M%S}.{1}.csv'.format(timestamp, "Mortgages" if name=='bond_trades' - else "CreditDefaultSwapDeal") + filename = get_filename(timestamp, queue_name) cmd = 'STOR {0}'.format(filename) with open(os.path.join('/home/share/Daily', str(timestamp.date()), filename), 'rb') as fh: ftp.storbinary(cmd, fh) def write_buffer(buf, queue_name='bond_trades'): - d = {'bond_trades':'Mortgages', - 'cds_trades':'CreditDefaultSwapDeal'} timestamp = pd.datetime.now() - filename = 'Serenitas.ALL.{0:%Y%m%d.%H%M%S}.{1}.csv'.format(timestamp, d[queue_name]) + filename = get_filename(timestamp, queue_name) with open(os.path.join('/home/share/Daily', str(timestamp.date()), filename), 'wb') as fh: fh.write(buf) return timestamp |