store.py

   1 import portfolio
   2
   3 __all__ = ["BalanceStore", "TradeStore"]
   4
   5 class BalanceStore:
   6     all = {}
   7
   8     @classmethod
   9     def currencies(cls):
  10         return cls.all.keys()
  11
  12     @classmethod
  13     def in_currency(cls, other_currency, market, compute_value="average", type="total"):
  14         amounts = {}
  15         for currency, balance in cls.all.items():
  16             other_currency_amount = getattr(balance, type)\
  17                     .in_currency(other_currency, market, compute_value=compute_value)
  18             amounts[currency] = other_currency_amount
  19         return amounts
  20
  21     @classmethod
  22     def fetch_balances(cls, market):
  23         all_balances = market.fetch_all_balances()
  24         for currency, balance in all_balances.items():
  25             if balance["exchange_total"] != 0 or balance["margin_total"] != 0 or \
  26                     currency in cls.all:
  27                 cls.all[currency] = portfolio.Balance(currency, balance)
  28
  29     @classmethod
  30     def dispatch_assets(cls, amount, repartition=None):
  31         if repartition is None:
  32             repartition = portfolio.Portfolio.repartition()
  33         sum_ratio = sum([v[0] for k, v in repartition.items()])
  34         amounts = {}
  35         for currency, (ptt, trade_type) in repartition.items():
  36             amounts[currency] = ptt * amount / sum_ratio
  37             if trade_type == "short":
  38                 amounts[currency] = - amounts[currency]
  39             if currency not in BalanceStore.all:
  40                 cls.all[currency] = portfolio.Balance(currency, {})
  41         return amounts
  42
  43 class TradeStore:
  44     all = []
  45     debug = False
  46
  47     @classmethod
  48     def compute_trades(cls, values_in_base, new_repartition, only=None, market=None, debug=False):
  49         cls.debug = cls.debug or debug
  50         base_currency = sum(values_in_base.values()).currency
  51         for currency in BalanceStore.currencies():
  52             if currency == base_currency:
  53                 continue
  54             value_from = values_in_base.get(currency, portfolio.Amount(base_currency, 0))
  55             value_to = new_repartition.get(currency, portfolio.Amount(base_currency, 0))
  56             if value_from.value * value_to.value < 0:
  57                 trade_1 = portfolio.Trade(value_from, portfolio.Amount(base_currency, 0), currency, market=market)
  58                 if only is None or trade_1.action == only:
  59                     cls.all.append(trade_1)
  60                 trade_2 = portfolio.Trade(portfolio.Amount(base_currency, 0), value_to, currency, market=market)
  61                 if only is None or trade_2.action == only:
  62                     cls.all.append(trade_2)
  63             else:
  64                 trade = portfolio.Trade(
  65                     value_from,
  66                     value_to,
  67                     currency,
  68                     market=market
  69                     )
  70                 if only is None or trade.action == only:
  71                     cls.all.append(trade)
  72
  73     @classmethod
  74     def prepare_orders(cls, only=None, compute_value="default"):
  75         for trade in cls.all:
  76             if only is None or trade.action == only:
  77                 trade.prepare_order(compute_value=compute_value)
  78
  79     @classmethod
  80     def print_all_with_order(cls):
  81         for trade in cls.all:
  82             trade.print_with_order()
  83
  84     @classmethod
  85     def run_orders(cls):
  86         for order in cls.all_orders(state="pending"):
  87             order.run()
  88
  89     @classmethod
  90     def all_orders(cls, state=None):
  91         all_orders = sum(map(lambda v: v.orders, cls.all), [])
  92         if state is None:
  93             return all_orders
  94         else:
  95             return list(filter(lambda o: o.status == state, all_orders))
  96
  97     @classmethod
  98     def update_all_orders_status(cls):
  99         for order in cls.all_orders(state="open"):
 100             order.get_status()
 101
 102