diff options
Diffstat (limited to 'main.py')
-rw-r--r-- | main.py | 13 |
1 files changed, 11 insertions, 2 deletions
@@ -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 | ||
145 | if __name__ == '__main__': # pragma: no cover | 154 | if __name__ == '__main__': # pragma: no cover |
146 | main(sys.argv[1:]) | 155 | main(sys.argv[1:]) |