class TaskExchanges(object):
def __init__(self, data={}):
self.period = 120
self.exchanges = Exchanges()
self.yahoo = Yahoo()
self.sina = Sina()
self.handler = None
data_type = ["orderbook", "ticker", "rate"]
for _type in data_type:
if _type not in data:
data[_type] = {}
self.data = data
def set_period(self, sec):
self.period = sec
async def fetch_orderbook(self, name, quote, coro, *args):
time_end = int(time.time())
orderbook = self.data["orderbook"]
while True:
time_begin = time_end
_orderbook = await coro(*args)
time_end = int(time.time())
if _orderbook:
orderbook[name] = _orderbook
orderbook[name]["quote"] = quote
if "time" not in _orderbook:
orderbook[name]["time"] = time_end
if self.handler:
self.handler("orderbook", name, orderbook[name])
time_left = self.period - (time_end - time_begin)
if time_left <= 1:
time_left = 1
time_end += time_left
await asyncio.sleep(time_left)
async def fetch_ticker(self, name, quote, coro, *args):
time_end = int(time.time())
ticker = self.data["ticker"]
while True:
time_begin = time_end
_ticker = await coro(*args)
time_end = int(time.time())
if _ticker:
_ticker["quote"] = quote
if "time" not in _ticker:
_ticker["time"] = time_end
ticker[name] = _ticker
if self.handler:
self.handler("ticker", name, _ticker)
time_left = self.period - (time_end - time_begin)
if time_left <= 1:
time_left = 1
time_end += time_left
await asyncio.sleep(time_left)
async def fetch_yahoo_rate(self):
time_end = int(time.time())
async with self.yahoo.create_session() as session:
rate = self.data["rate"]
while True:
time_begin = time_end
_rate = await self.yahoo.fetch_price(session)
time_end = int(time.time())
if _rate:
_rate["time"] = time_end
rate["yahoo"] = _rate
if self.handler:
self.handler("rate", "yahoo", _rate)
time_left = self.period - (time_end - time_begin)
if time_left <= 1:
time_left = 1
time_end += time_left
await asyncio.sleep(time_left)
async def fetch_sina_rate(self):
time_end = int(time.time())
async with self.sina.create_session() as session:
rate = self.data["rate"]
while True:
time_begin = time_end
_rate = await self.sina.fetch_price(session)
time_end = int(time.time())
if _rate:
_rate["time"] = time_end
rate["Sina"] = _rate
if self.handler:
self.handler("rate", "Sina", _rate)
time_left = self.period - (time_end - time_begin)
if time_left <= 1:
time_left = 1
time_end += time_left
await asyncio.sleep(time_left)
def run_tasks_ticker(self, loop):
return [
loop.create_task(self.fetch_ticker(
"poloniex", "USD",
self.exchanges.ticker_poloniex, "USDT", "BTC")),
# loop.create_task(self.fetch_ticker(
# "btce", "USD",
# self.exchanges.ticker_btce, "usd", "btc")),
loop.create_task(self.fetch_ticker(
"bitstamp", "USD",
self.exchanges.ticker_bitstamp, "usd", "btc")),
loop.create_task(self.fetch_ticker(
"gdax", "USD",
self.exchanges.ticker_gdax, "usd", "btc")),
# loop.create_task(self.fetch_ticker(
# "btcchina", "CNY",
# self.exchanges.ticker_btcchina, "cny", "btc")),
# loop.create_task(self.fetch_ticker(
# "huobi", "CNY",
# self.exchanges.ticker_huobi, "btc")),
# loop.create_task(self.fetch_ticker(
# "okcoin_cn", "CNY",
# self.exchanges.ticker_okcoin_cn, "cny", "btc")),
loop.create_task(self.fetch_ticker(
"okcoin_com", "USD",
self.exchanges.ticker_okcoin_com, "usd", "btc")),
loop.create_task(self.fetch_ticker(
"bitfinex", "USD",
self.exchanges.ticker_bitfinex, "usd", "btc")),
loop.create_task(self.fetch_ticker(
"kraken", "EUR",
self.exchanges.ticker_kraken, "eur", "btc")),
loop.create_task(self.fetch_ticker(
"bitflyer_usd", "USD",
self.exchanges.ticker_bitflyer, "usd", "btc")),
loop.create_task(self.fetch_ticker(
"bitflyer_jpy", "JPY",
self.exchanges.ticker_bitflyer, "jpy", "btc")),
]
def run_tasks_orderbook(self, loop):
return [
# loop.create_task(self.fetch_orderbook(
# "btsbots_cny", "CNY",
# self.exchanges.orderbook_btsbots, "CNY", "BTS")),
# loop.create_task(self.fetch_orderbook(
# "btsbots_usd", "USD",
# self.exchanges.orderbook_btsbots, "USD", "BTS")),
# loop.create_task(self.fetch_orderbook(
# "btsbots_open.btc", "BTC",
# self.exchanges.orderbook_btsbots, "OPEN.BTC", "BTS")),
loop.create_task(self.fetch_orderbook(
"aex_btc", "BTC",
self.exchanges.orderbook_aex, "btc", "bts")),
loop.create_task(self.fetch_orderbook(
"aex_bitcny", "CNY",
self.exchanges.orderbook_aex, "bitcny", "bts")),
loop.create_task(self.fetch_orderbook(
"aex_bitusd", "USD",
self.exchanges.orderbook_aex, "bitusd", "bts")),
loop.create_task(self.fetch_orderbook(
"zb_btc", "BTC",
self.exchanges.orderbook_zb, "btc", "bts")),
loop.create_task(self.fetch_orderbook(
"zb_usdt", "USD",
self.exchanges.orderbook_zb, "usdt", "bts")),
loop.create_task(self.fetch_orderbook(
"lbank_btc", "BTC",
self.exchanges.orderbook_lbank, "btc", "bts")),
loop.create_task(self.fetch_orderbook(
"binance_btc", "BTC",
self.exchanges.orderbook_binance, "btc", "bts")),
loop.create_task(self.fetch_orderbook(
"poloniex_btc", "BTC",
self.exchanges.orderbook_poloniex, "btc", "bts"))
# loop.create_task(self.fetch_orderbook(
# "yunbi_cny", "CNY",
# self.exchanges.orderbook_yunbi, "cny", "bts")),
# loop.create_task(self.fetch_orderbook(
# "jubi_cny", "CNY",
# self.exchanges.orderbook_jubi, "cny", "bts")),
# loop.create_task(self.fetch_orderbook(
# "19800_cny", "CNY",
# self.exchanges.orderbook_19800, "cny", "bts")),
# loop.create_task(self.fetch_orderbook(
# "bittrex_btc", "BTC",
# self.exchanges.orderbook_bittrex, "btc", "bts")),
]
def run_tasks(self, loop):
loop.run_until_complete(self.exchanges.init_session())
return [
loop.create_task(self.fetch_yahoo_rate()),
loop.create_task(self.fetch_sina_rate()),
] + \
self.run_tasks_orderbook(loop) + \
self.run_tasks_ticker(loop)
