parser.add_argument("--user",
default=None, required=False, help="Only run for that user")
parser.add_argument("--action",
- default=None, required=False,
- help="Do a different action than trading")
+ action='append',
+ help="Do a different action than trading (add several times to chain)")
args = parser.parse_args(argv)
for row in cursor:
yield row
-def main_process_market(user_market, action, before=False, after=False):
- if action is None:
+def main_process_market(user_market, actions, before=False, after=False):
+ if len(actions or []) == 0:
if before:
process_sell_all__1_all_sell(user_market)
if after:
portfolio.Portfolio.wait_for_recent(user_market)
process_sell_all__2_all_buy(user_market)
- elif action == "print_balances":
- print_balances(user_market)
- elif action == "print_orders":
- print_orders(user_market)
else:
- raise NotImplementedError("Unknown action {}".format(action))
+ for action in actions:
+ if action in globals():
+ (globals()[action])(user_market)
+ else:
+ raise NotImplementedError("Unknown action {}".format(action))
def main_store_report(report_path, user_id, user_market):
try: