diff options
author | Ismaël Bouya <ismael.bouya@normalesup.org> | 2018-03-09 15:37:10 +0100 |
---|---|---|
committer | Ismaël Bouya <ismael.bouya@normalesup.org> | 2018-03-09 19:12:57 +0100 |
commit | dc1ca9a306f09886c6c57f8d426c59a9d084b2b3 (patch) | |
tree | d1a94be893451a4e182f3e75e9afb01749172bb4 /main.py | |
parent | 81d1db5117b61c2bb9f114cfc59f015e97c6eb9b (diff) | |
download | Trader-dc1ca9a306f09886c6c57f8d426c59a9d084b2b3.tar.gz Trader-dc1ca9a306f09886c6c57f8d426c59a9d084b2b3.tar.zst Trader-dc1ca9a306f09886c6c57f8d426c59a9d084b2b3.zip |
Add parallelization
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:]) |