aboutsummaryrefslogtreecommitdiff
path: root/main.py
diff options
context:
space:
mode:
Diffstat (limited to 'main.py')
-rw-r--r--main.py13
1 files changed, 11 insertions, 2 deletions
diff --git a/main.py b/main.py
index 37f485d..856d449 100644
--- a/main.py
+++ b/main.py
@@ -117,6 +117,8 @@ def parse_args(argv):
117 parser.add_argument("--action", 117 parser.add_argument("--action",
118 action='append', 118 action='append',
119 help="Do a different action than trading (add several times to chain)") 119 help="Do a different action than trading (add several times to chain)")
120 parser.add_argument("--parallel", action='store_true', default=True, dest="parallel")
121 parser.add_argument("--no-parallel", action='store_false', dest="parallel")
120 122
121 args = parser.parse_args(argv) 123 args = parser.parse_args(argv)
122 124
@@ -139,8 +141,15 @@ def main(argv):
139 141
140 pg_config, report_path = parse_config(args.config) 142 pg_config, report_path = parse_config(args.config)
141 143
142 for market_config, user_id in fetch_markets(pg_config, args.user): 144 if args.parallel:
143 process(market_config, user_id, report_path, args) 145 import threading
146 market.Portfolio.start_worker()
147
148 for market_config, user_id in fetch_markets(pg_config, args.user):
149 threading.Thread(target=process, args=[market_config, user_id, report_path, args]).start()
150 else:
151 for market_config, user_id in fetch_markets(pg_config, args.user):
152 process(market_config, user_id, report_path, args)
144 153
145if __name__ == '__main__': # pragma: no cover 154if __name__ == '__main__': # pragma: no cover
146 main(sys.argv[1:]) 155 main(sys.argv[1:])