import time from ccxt import ExchangeError from store import * def move_balances(market, debug=False): needed_in_margin = {} moving_to_margin = {} for currency in BalanceStore.all: if BalanceStore.all[currency].margin_free != 0: needed_in_margin[currency] = 0 for trade in TradeStore.all: if trade.value_to.currency not in needed_in_margin: needed_in_margin[trade.value_to.currency] = 0 if trade.trade_type == "short": needed_in_margin[trade.value_to.currency] += abs(trade.value_to) for currency, needed in needed_in_margin.items(): current_balance = BalanceStore.all[currency].margin_free moving_to_margin[currency] = (needed - current_balance) delta = moving_to_margin[currency].value if debug: ReportStore.log_debug_action("Moving {} from exchange to margin".format(moving_to_margin[currency])) continue if delta > 0: market.transfer_balance(currency, delta, "exchange", "margin") elif delta < 0: market.transfer_balance(currency, -delta, "margin", "exchange") ReportStore.log_move_balances(needed_in_margin, moving_to_margin, debug) BalanceStore.fetch_balances(market) ticker_cache = {} ticker_cache_timestamp = time.time() def get_ticker(c1, c2, market, refresh=False): global ticker_cache, ticker_cache_timestamp def invert(ticker): return { "inverted": True, "average": (1/ticker["bid"] + 1/ticker["ask"]) / 2, "original": ticker, } def augment_ticker(ticker): ticker.update({ "inverted": False, "average": (ticker["bid"] + ticker["ask"] ) / 2, }) if time.time() - ticker_cache_timestamp > 5: ticker_cache = {} ticker_cache_timestamp = time.time() elif not refresh: if (c1, c2, market.__class__) in ticker_cache: return ticker_cache[(c1, c2, market.__class__)] if (c2, c1, market.__class__) in ticker_cache: return invert(ticker_cache[(c2, c1, market.__class__)]) try: ticker_cache[(c1, c2, market.__class__)] = market.fetch_ticker("{}/{}".format(c1, c2)) augment_ticker(ticker_cache[(c1, c2, market.__class__)]) except ExchangeError: try: ticker_cache[(c2, c1, market.__class__)] = market.fetch_ticker("{}/{}".format(c2, c1)) augment_ticker(ticker_cache[(c2, c1, market.__class__)]) except ExchangeError: ticker_cache[(c1, c2, market.__class__)] = None return get_ticker(c1, c2, market) fees_cache = {} def fetch_fees(market): global fees_cache if market.__class__ not in fees_cache: fees_cache[market.__class__] = market.fetch_fees() return fees_cache[market.__class__] def prepare_trades(market, base_currency="BTC", liquidity="medium", compute_value="average", debug=False): ReportStore.log_stage("prepare_trades") values_in_base = BalanceStore.in_currency(base_currency, market, compute_value=compute_value) total_base_value = sum(values_in_base.values()) new_repartition = BalanceStore.dispatch_assets(total_base_value, liquidity=liquidity) # Recompute it in case we have new currencies values_in_base = BalanceStore.in_currency(base_currency, market, compute_value=compute_value) TradeStore.compute_trades(values_in_base, new_repartition, market=market, debug=debug) def update_trades(market, base_currency="BTC", liquidity="medium", compute_value="average", only=None, debug=False): ReportStore.log_stage("update_trades") values_in_base = BalanceStore.in_currency(base_currency, market, compute_value=compute_value) total_base_value = sum(values_in_base.values()) new_repartition = BalanceStore.dispatch_assets(total_base_value, liquidity=liquidity) TradeStore.compute_trades(values_in_base, new_repartition, only=only, market=market, debug=debug) def prepare_trades_to_sell_all(market, base_currency="BTC", compute_value="average", debug=False): ReportStore.log_stage("prepare_trades_to_sell_all") values_in_base = BalanceStore.in_currency(base_currency, market, compute_value=compute_value) total_base_value = sum(values_in_base.values()) new_repartition = BalanceStore.dispatch_assets(total_base_value, repartition={ base_currency: (1, "long") }) TradeStore.compute_trades(values_in_base, new_repartition, market=market, debug=debug) def follow_orders(sleep=None): if sleep is None: sleep = 7 if TradeStore.debug else 30 if TradeStore.debug: ReportStore.log_debug_action("Set follow_orders tick to {}s".format(sleep)) tick = 0 ReportStore.log_stage("follow_orders_begin") while len(TradeStore.all_orders(state="open")) > 0: time.sleep(sleep) tick += 1 open_orders = TradeStore.all_orders(state="open") ReportStore.log_stage("follow_orders_tick_{}".format(tick)) ReportStore.log_orders(open_orders, tick=tick) for order in open_orders: if order.get_status() != "open": ReportStore.log_order(order, tick, finished=True) else: order.trade.update_order(order, tick) ReportStore.log_stage("follow_orders_end") def print_orders(market, base_currency="BTC"): ReportStore.log_stage("print_orders") BalanceStore.fetch_balances(market, tag="print_orders") prepare_trades(market, base_currency=base_currency, compute_value="average", debug=True) TradeStore.prepare_orders(compute_value="average") def print_balances(market, base_currency="BTC"): BalanceStore.fetch_balances(market) if base_currency is not None: ReportStore.print_log("total:") ReportStore.print_log(sum(BalanceStore.in_currency(base_currency, market).values())) def process_sell_needed__1_sell(market, liquidity="medium", base_currency="BTC", debug=False): ReportStore.log_stage("process_sell_needed__1_sell_begin") BalanceStore.fetch_balances(market, tag="process_sell_needed__1_sell_begin") prepare_trades(market, liquidity=liquidity, base_currency=base_currency, debug=debug) TradeStore.prepare_orders(compute_value="average", only="dispose") TradeStore.run_orders() follow_orders() BalanceStore.fetch_balances(market, tag="process_sell_needed__1_sell_end") ReportStore.log_stage("process_sell_needed__1_sell_end") def process_sell_needed__2_buy(market, liquidity="medium", base_currency="BTC", debug=False): ReportStore.log_stage("process_sell_needed__2_buy_begin") BalanceStore.fetch_balances(market, tag="process_sell_needed__2_buy_begin") update_trades(market, base_currency=base_currency, liquidity=liquidity, debug=debug, only="acquire") TradeStore.prepare_orders(compute_value="average", only="acquire") move_balances(market, debug=debug) TradeStore.run_orders() follow_orders() BalanceStore.fetch_balances(market, tag="process_sell_needed__2_buy_end") ReportStore.log_stage("process_sell_needed__2_buy_end") def process_sell_all__1_all_sell(market, base_currency="BTC", debug=False, liquidity="medium"): ReportStore.log_stage("process_sell_all__1_all_sell_begin") BalanceStore.fetch_balances(market, tag="process_sell_all__1_all_sell_begin") prepare_trades_to_sell_all(market, base_currency=base_currency, debug=debug) TradeStore.prepare_orders(compute_value="average") TradeStore.run_orders() follow_orders() BalanceStore.fetch_balances(market, tag="process_sell_all__1_all_sell_end") ReportStore.log_stage("process_sell_all__1_all_sell_end") def process_sell_all__2_all_buy(market, base_currency="BTC", debug=False, liquidity="medium"): ReportStore.log_stage("process_sell_all__2_all_buy_begin") BalanceStore.fetch_balances(market, tag="process_sell_all__2_all_buy_begin") prepare_trades(market, liquidity=liquidity, base_currency=base_currency, debug=debug) TradeStore.prepare_orders(compute_value="average") move_balances(market, debug=debug) TradeStore.run_orders() follow_orders() BalanceStore.fetch_balances(market, tag="process_sell_all__2_all_buy_end") ReportStore.log_stage("process_sell_all__2_all_buy_end")