-from datetime import datetime
import configargparse
import psycopg2
import os
import threading
market.Portfolio.start_worker()
+ threads = []
def process_(*args):
- threading.Thread(target=process, args=args).start()
+ thread = threading.Thread(target=process, args=args)
+ thread.start()
+ threads.append(thread)
else:
process_ = process
for market_id, market_config, user_id in fetch_markets(pg_config, args.user):
process_(market_config, market_id, user_id, args, pg_config)
+ if args.parallel:
+ for thread in threads:
+ thread.join()
+ market.Portfolio.stop_worker()
+
if __name__ == '__main__': # pragma: no cover
main(sys.argv[1:])