from ccxt import ExchangeError
import ccxt_wrapper as ccxt
import time
from store import *
class Market:
debug = False
ccxt = None
report = None
trades = None
balances = None
def __init__(self, ccxt_instance, debug=False):
self.debug = debug
self.ccxt = ccxt_instance
self.ccxt._market = self
self.report = ReportStore(self)
self.trades = TradeStore(self)
self.balances = BalanceStore(self)
@classmethod
def from_config(cls, config, debug=False):
ccxt_instance = ccxt.poloniexE(config)
# For requests logging
ccxt_instance.session.origin_request = ccxt_instance.session.request
ccxt_instance.session._parent = ccxt_instance
def request_wrap(self, *args, **kwargs):
r = self.origin_request(*args, **kwargs)
self._parent._market.report.log_http_request(args[0],
args[1], kwargs["data"], kwargs["headers"], r)
return r
ccxt_instance.session.request = request_wrap.__get__(ccxt_instance.session,
ccxt_instance.session.__class__)
return cls(ccxt_instance, debug=debug)
def move_balances(self):
needed_in_margin = {}
moving_to_margin = {}
for currency in self.balances.all:
if self.balances.all[currency].margin_free != 0:
needed_in_margin[currency] = 0
for trade in self.trades.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 = self.balances.all[currency].margin_free
moving_to_margin[currency] = (needed - current_balance)
delta = moving_to_margin[currency].value
if self.debug:
self.report.log_debug_action("Moving {} from exchange to margin".format(moving_to_margin[currency]))
continue
if delta > 0:
self.ccxt.transfer_balance(currency, delta, "exchange", "margin")
elif delta < 0:
self.ccxt.transfer_balance(currency, -delta, "margin", "exchange")
self.report.log_move_balances(needed_in_margin, moving_to_margin)
self.balances.fetch_balances()
fees_cache = {}
def fetch_fees(self):
if self.ccxt.__class__ not in self.fees_cache:
self.fees_cache[self.ccxt.__class__] = self.ccxt.fetch_fees()
return self.fees_cache[self.ccxt.__class__]
ticker_cache = {}
ticker_cache_timestamp = time.time()
def get_ticker(self, c1, c2, refresh=False):
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() - self.ticker_cache_timestamp > 5:
self.ticker_cache = {}
self.ticker_cache_timestamp = time.time()
elif not refresh:
if (c1, c2, self.ccxt.__class__) in self.ticker_cache:
return self.ticker_cache[(c1, c2, self.ccxt.__class__)]
if (c2, c1, self.ccxt.__class__) in self.ticker_cache:
return invert(self.ticker_cache[(c2, c1, self.ccxt.__class__)])
try:
self.ticker_cache[(c1, c2, self.ccxt.__class__)] = self.ccxt.fetch_ticker("{}/{}".format(c1, c2))
augment_ticker(self.ticker_cache[(c1, c2, self.ccxt.__class__)])
except ExchangeError:
try:
self.ticker_cache[(c2, c1, self.ccxt.__class__)] = self.ccxt.fetch_ticker("{}/{}".format(c2, c1))
augment_ticker(self.ticker_cache[(c2, c1, self.ccxt.__class__)])
except ExchangeError:
self.ticker_cache[(c1, c2, self.ccxt.__class__)] = None
return self.get_ticker(c1, c2)
def follow_orders(self, sleep=None):
if sleep is None:
sleep = 7 if self.debug else 30
if self.debug:
self.report.log_debug_action("Set follow_orders tick to {}s".format(sleep))
tick = 0
self.report.log_stage("follow_orders_begin")
while len(self.trades.all_orders(state="open")) > 0:
time.sleep(sleep)
tick += 1
open_orders = self.trades.all_orders(state="open")
self.report.log_stage("follow_orders_tick_{}".format(tick))
self.report.log_orders(open_orders, tick=tick)
for order in open_orders:
if order.get_status() != "open":
self.report.log_order(order, tick, finished=True)
else:
order.trade.update_order(order, tick)
self.report.log_stage("follow_orders_end")
def prepare_trades(self, base_currency="BTC", liquidity="medium", compute_value="average"):
self.report.log_stage("prepare_trades")
values_in_base = self.balances.in_currency(base_currency, compute_value=compute_value)
total_base_value = sum(values_in_base.values())
new_repartition = self.balances.dispatch_assets(total_base_value, liquidity=liquidity)
# Recompute it in case we have new currencies
values_in_base = self.balances.in_currency(base_currency, compute_value=compute_value)
self.trades.compute_trades(values_in_base, new_repartition)
def update_trades(self, base_currency="BTC", liquidity="medium", compute_value="average", only=None):
self.report.log_stage("update_trades")
values_in_base = self.balances.in_currency(base_currency, compute_value=compute_value)
total_base_value = sum(values_in_base.values())
new_repartition = self.balances.dispatch_assets(total_base_value, liquidity=liquidity)
self.trades.compute_trades(values_in_base, new_repartition, only=only)
def prepare_trades_to_sell_all(self, base_currency="BTC", compute_value="average"):
self.report.log_stage("prepare_trades_to_sell_all")
values_in_base = self.balances.in_currency(base_currency, compute_value=compute_value)
total_base_value = sum(values_in_base.values())
new_repartition = self.balances.dispatch_assets(total_base_value, repartition={ base_currency: (1, "long") })
self.trades.compute_trades(values_in_base, new_repartition)
