core/bfxbot/bfxbot.py

91 lines
2.9 KiB
Python
Raw Normal View History

2020-11-30 14:38:28 +00:00
from time import sleep
2020-12-10 16:29:26 +00:00
from typing import Dict, List
2020-11-30 09:12:43 +00:00
from bfxbot.bfxwrapper import BfxWrapper
2020-11-30 14:38:28 +00:00
from bfxbot.currency import Symbol
2020-12-10 16:29:26 +00:00
from bfxbot.models import SymbolStatus, Ticker, EventHandler, Strategy, Event, EventKind
2020-11-30 09:12:43 +00:00
class BfxBot:
2020-12-10 16:29:26 +00:00
def __init__(self, api_key: str, api_secret: str, symbols: List[Symbol], tick_duration: int = 1, ):
if api_key is None:
print("API_KEY is not set!")
raise ValueError
if api_secret is None:
print("API_SECRET is not set!")
raise ValueError
self.__bfx: BfxWrapper = BfxWrapper(api_key, api_secret)
self.__ticker: Ticker = Ticker(tick_duration)
self.__status: Dict[Symbol, SymbolStatus] = {}
2020-11-30 09:12:43 +00:00
2020-12-10 16:29:26 +00:00
if isinstance(symbols, Symbol):
symbols = [symbols]
self.symbols: List[Symbol] = symbols
# init symbol statuses
for s in self.symbols:
self.__status[s] = SymbolStatus(s)
2020-12-10 16:29:26 +00:00
2020-11-30 09:12:43 +00:00
async def __update_status__(self):
active_positions = await self.__bfx.get_active_position()
2020-11-30 09:12:43 +00:00
for symbol in self.__status:
2020-12-10 16:29:26 +00:00
# updating tick
self.__status[symbol].current_tick = self.__ticker.current_tick
2020-12-10 16:29:26 +00:00
# updating last price
last_price = await self.__bfx.get_current_prices(symbol)
2020-12-10 16:29:26 +00:00
last_price = last_price[0]
2020-11-30 09:12:43 +00:00
self.__status[symbol].set_tick_price(self.__ticker.current_tick, last_price)
2020-11-30 14:38:28 +00:00
2020-12-10 16:29:26 +00:00
# updating positions
symbol_positions = [x for x in active_positions if x.symbol == str(symbol)]
for p in symbol_positions:
await self.__status[Symbol.from_str(p.symbol)].add_position(p)
2020-11-30 09:12:43 +00:00
# updating orders
active_orders = await self.__bfx.get_active_orders(str(symbol))
2020-11-30 09:12:43 +00:00
for o in active_orders:
self.__status[symbol].add_order(o)
2020-11-30 14:38:28 +00:00
# emitting new tick event
2020-12-15 16:34:36 +00:00
## TODO: handle _on_new_tick() from Strategy
await self.__status[symbol].__add_event__(Event(EventKind.NEW_TICK, self.__ticker.current_tick))
2020-11-30 09:12:43 +00:00
def close_order(self, symbol: Symbol, order_id: int):
raise NotImplementedError
def close_position(self, symbol: Symbol, position_id: int):
raise NotImplementedError
def set_strategy(self, symbol, strategy: Strategy):
if symbol in self.__status:
self.__status[symbol].strategy = strategy
else:
self.__status[symbol] = SymbolStatus(symbol, strategy)
async def start(self):
await self.__update_status__()
def symbol_event_handler(self, symbol) -> EventHandler:
if symbol not in self.__status:
2020-11-30 14:38:28 +00:00
return None
return self.__status[symbol].eh
2020-11-30 14:38:28 +00:00
def symbol_status(self, symbol: Symbol) -> SymbolStatus:
if symbol not in self.__status:
2020-11-30 14:38:28 +00:00
return None
return self.__status[symbol]
2020-11-30 14:38:28 +00:00
2020-11-30 09:12:43 +00:00
async def update(self):
sleep(self.__ticker.seconds)
self.__ticker.inc()
2020-11-30 09:12:43 +00:00
await self.__update_status__()