diff options
Diffstat (limited to 'python/process_queue.py')
| -rw-r--r-- | python/process_queue.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/python/process_queue.py b/python/process_queue.py index 07669e09..4c99f946 100644 --- a/python/process_queue.py +++ b/python/process_queue.py @@ -17,6 +17,7 @@ from bbg_helpers import init_bbg_session, retrieve_data, BBG_IP import re import psycopg2 import logging +import argparse def decode_dict(d): return {k.decode() if isinstance(k, bytes) else k: \ @@ -166,6 +167,9 @@ def write_buffer(buf): return timestamp if __name__=="__main__": + parser = argparse.ArgumentParser() + parser.add_argument("-n", "--no-upload", action="store_true", help="do not upload to Globeop") + args = parser.parse_args() engine = create_engine('postgresql://dawn_user@debian/dawndb') conn = engine.raw_connection() q = get_redis_queue() @@ -176,5 +180,6 @@ if __name__=="__main__": for trade in l: bbg_process(conn, session, trade) timestamp = write_buffer(buf) - upload_file(timestamp) + if not args.no_upload: + upload_file(timestamp) q.delete('trades') |