This commit is contained in:
falsycat 2022-07-15 19:17:55 +09:00
parent 37840311e0
commit 3f61bf8509
11 changed files with 695 additions and 0 deletions

2
.gitignore vendored Normal file
View File

@ -0,0 +1,2 @@
__pycache__/
/secret.json

347
bitbank.py Normal file
View File

@ -0,0 +1,347 @@
import asyncio
import datetime
import logging
import pybotters
import logic
import util.candlestick
import util.depth
import util.pair
import util.ticker
PUBLIC_API_ENTRYPOINT = "https://public.bitbank.cc"
PRIVATE_API_ENTRYPOINT = "https://api.bitbank.cc/v1"
logger = logging.getLogger(__name__)
async def init(cli):
btc_store = pybotters.bitbankDataStore()
asyncio.create_task(cli.ws_connect(
'wss://stream.bitbank.cc/socket.io/?EIO=3&transport=websocket',
send_str=[
'42["join-room","ticker_btc_jpy"]',
'42["join-room","transactions_btc_jpy"]',
'42["join-room","depth_whole_btc_jpy"]',
'42["join-room","depth_diff_btc_jpy"]',
],
hdlr_str = btc_store.onmessage,
))
btc = Pair(cli, btc_store, "btc_jpy")
logic.MA_Cross(
logging.getLogger("bitbank/btc/MA_Cross/1m"),
btc, btc.candlestick_1m, 1000, 0.002, 0.001)
eth_store = pybotters.bitbankDataStore()
asyncio.create_task(cli.ws_connect(
'wss://stream.bitbank.cc/socket.io/?EIO=3&transport=websocket',
send_str=[
'42["join-room","ticker_eth_jpy"]',
'42["join-room","transactions_eth_jpy"]',
'42["join-room","depth_whole_eth_jpy"]',
'42["join-room","depth_diff_eth_jpy"]',
],
hdlr_str = eth_store.onmessage,
))
eth = Pair(cli, eth_store, "eth_jpy")
logic.MM(
logging.getLogger("bitbank/eth/MM"),
eth, 1000, 1)
logic.MA_Cross(
logging.getLogger("bitbank/eth/MA_Cross/1m"),
eth, eth.candlestick_1m, 1000, 0.002, 0.001)
class Pair(util.pair.Pair):
def __init__(self, cli, store, name):
super().__init__(name)
self._cli = cli
self.ticker = Ticker(store, name)
self.candlestick_1s = Candlestick(cli, self.ticker, None, 1)
self.candlestick_1m = Candlestick(cli, self.ticker, "1min", 60)
self.candlestick_1h = Candlestick(cli, self.ticker, "1hour", 60*60)
self.depth = Depth(store)
self._cancelling_orders = []
self._cancelling = False
async def _order(self, amount, side, type, trigger = None, price = None, post_only = None):
data = {
"pair" : self.name,
"amount": str(amount),
"side" : side,
"type" : type,
}
if price is not None: data["price"] = price
if trigger is not None: data["trigger_price"] = trigger
if post_only is not None: data["post_only"] = post_only
res = await self._cli.post(PRIVATE_API_ENTRYPOINT+"/user/spot/order", data=data)
res = await res.json()
_check_response(res)
return Order(self._cli, self, res["data"]["order_id"])
async def _cancel_order(self, id):
self._cancelling_orders.append(id)
async def cancel(self):
while True:
res = await self._cli.post(PRIVATE_API_ENTRYPOINT+"/user/spot/cancel_orders", data={
"pair": self.name,
"order_ids": self._cancelling_orders,
})
res = await res.json()
res["success"] == 1 or res["data"]["code"] != 20001:
break
await asyncio.sleep(1)
_check_response(res)
self._cancelling_orders = []
self._cancelling = False
if not self._cancelling:
self._cancelling = True
asyncio.create_task(cancel(self))
def buy_market(self, amount):
return self._order(amount, "buy", "market")
def sell_market(self, amount):
return self._order(amount, "sell", "market")
def buy_limit(self, amount, price, post_only = False):
return self._order(amount, "buy", "limit", None, price, post_only)
def sell_limit(self, amount, price, post_only = False):
return self._order(amount, "sell", "limit", None, price, post_only)
def buy_stop(self, amount, trigger):
return self._order(amount, "buy", "stop", trigger)
def sell_stop(self, amount, trigger):
return self._order(amount, "sell", "stop", trigger)
def buy_stop_limit(self, amount, trigger, price):
return self._order(amount, "buy", "stop", trigger, price)
def sell_stop_limit(self, amount, trigger, price):
return self._order(amount, "sell", "stop", trigger, price)
class Order(util.pair.Order):
def __init__(self, cli, pair, id):
super().__init__()
self._cli = cli
self._pair = pair
self._id = id
async def update(self):
res = await self._cli.get(PRIVATE_API_ENTRYPOINT+"/user/spot/order", params={
"pair": self._pair.name,
"order_id": self._id,
})
res = await res.json()
_check_response(res)
res = res["data"]
self.amount = float(res["start_amount"])
self.remain = float(res["remaining_amount"])
self.price = float(res["average_price"])
self.done = \
res["status"] == "FULLY_FILLED" or \
res["status"] == "CANCELLED_UNFILLED" or \
res["status"] == "CANCELLED_PARTIALLY_FILLED"
async def cancel(self):
await self._pair._cancel_order(self._id)
class Ticker(util.ticker.Ticker):
def __init__(self, store, pair):
super().__init__(pair)
self._store = store
asyncio.create_task(self._main())
async def _main(self):
while True:
await self._store.ticker.wait()
ticker = self._store.ticker.find()
for i in range(len(ticker)):
if ticker[i]["pair"] == self.pair:
self.price = float(ticker[i]["last"])
self.sell = float(ticker[i]["sell"])
self.buy = float(ticker[i]["buy"])
self.volume = float(ticker[i]["vol"])
self.ts = int(ticker[i]["timestamp"])
self._event.set()
self._event.clear()
break
class Candlestick(util.candlestick.Candlestick):
def __init__(self, cli, ticker, unit, interval):
super().__init__(ticker, interval)
self._cli = cli
self._unit = unit
async def _fetch(self, date, depth = 0, rem = util.candlestick.MAX_CANDLES):
if self._unit is None:
return []
long_term = \
self._unit == "4hour" or \
self._unit == "8hour" or \
self._unit == "12hour" or \
self._unit == "1day" or \
self._unit == "1week" or \
self._unit == "1month"
if long_term:
strdate = date.strftime("%Y")
else:
strdate = date.strftime("%Y%m%d")
pair = self._ticker.pair
data = await self._cli.get(PUBLIC_API_ENTRYPOINT+f"/{pair}/candlestick/{self._unit}/{strdate}")
data = await data.json()
_check_response(data)
data = data["data"]["candlestick"][0]["ohlcv"]
first_index = 0
if depth == 0: first_index = 1
ret = []
for i in range(first_index, len(data)):
ret.append([
float(data[i][0]),
float(data[i][1]),
float(data[i][2]),
float(data[i][3]),
])
if len(ret) < rem:
back = datetime.timedelta(days=1)
if long_term:
back = datetime.timedelta(years=1)
ret = [*ret, *await self._fetch(date-back, depth+1, rem-len(ret))]
return ret
class Depth(util.depth.Depth):
def __init__(self, store):
super().__init__()
self._store = store
asyncio.create_task(self._main())
async def _main(self):
while True:
await self._store.depth.wait()
items = self._store.depth.sorted()
asks = items["asks"]
bids = items["bids"]
self.asks = []
for i in range(len(asks)):
self.asks.append([float(asks[i][0]), float(asks[i][1])])
self.bids = []
for i in range(len(bids)):
self.bids.append([float(bids[i][0]), float(bids[i][1])])
self._event.set()
self._event.clear()
def _check_response(res):
if res["success"] == 1:
return
code = str(res["data"]["code"])
if code in ERROR_CODES:
raise Exception(ERROR_CODES[code])
raise Exception("unknown error")
ERROR_CODES = {
'10000': 'URLが存在しません',
'10001': 'システムエラーが発生しました。サポートにお問い合わせ下さい',
'10002': '不正なJSON形式です。送信内容をご確認下さい',
'10003': 'システムエラーが発生しました。サポートにお問い合わせ下さい',
'10005': 'タイムアウトエラーが発生しました。しばらく間をおいて再度実行して下さい',
'20001': 'API認証に失敗しました',
'20002': 'APIキーが不正です',
'20003': 'APIキーが存在しません',
'20004': 'API Nonceが存在しません',
'20005': 'APIシグネチャが存在しません',
'20011': '2段階認証に失敗しました',
'20014': 'SMS認証に失敗しました',
'20023': '2段階認証コードを入力して下さい',
'20024': 'SMS認証コードを入力して下さい',
'20025': '段階認証コードとSMS認証コードを入力して下さい',
'20026': '一定回数以上段階認証に失敗したためロックしました。60秒待ってから再度お試しください',
'30001': '注文数量を指定して下さい',
'30006': '注文IDを指定して下さい',
'30007': '注文ID配列を指定して下さい',
'30009': '銘柄を指定して下さい',
'30012': '注文価格を指定して下さい',
'30013': '売買どちらかを指定して下さい',
'30015': '注文タイプを指定して下さい',
'30016': 'アセット名を指定して下さい',
'30019': 'uuidを指定して下さい',
'30039': '出金額を指定して下さい',
'30101': 'トリガー価格を指定してください',
'40001': '注文数量が不正です',
'40006': 'count値が不正です',
'40007': '終了時期が不正です',
'40008': 'end_id値が不正です',
'40009': 'from_id値が不正です',
'40013': '注文IDが不正です',
'40014': '注文ID配列が不正です',
'40015': '指定された注文が多すぎます',
'40017': '銘柄名が不正です',
'40020': '注文価格が不正です',
'40021': '売買区分が不正です',
'40022': '開始時期が不正です',
'40024': '注文タイプが不正です',
'40025': 'アセット名が不正です',
'40028': 'uuidが不正です',
'40048': '出金額が不正です',
'40112': 'トリガー価格が不正です',
'40113': 'post_only値が不正です',
'40114': 'Post Onlyはご指定の注文タイプでは指定できません',
'50003': '現在、このアカウントはご指定の操作を実行できない状態となっております。サポートにお問い合わせ下さい',
'50004': '現在、このアカウントは仮登録の状態となっております。アカウント登録完了後、再度お試し下さい',
'50005': '現在、このアカウントはロックされております。サポートにお問い合わせ下さい',
'50006': '現在、このアカウントはロックされております。サポートにお問い合わせ下さい',
'50008': 'ユーザの本人確認が完了していません',
'50009': 'ご指定の注文は存在しません',
'50010': 'ご指定の注文はキャンセルできません',
'50011': 'APIが見つかりません',
'50026': 'ご指定の注文は既にキャンセル済みです',
'50027': 'ご指定の注文は既に約定済みです',
'60001': '保有数量が不足しています',
'60002': '成行買い注文の数量上限を上回っています',
'60003': '指定した数量が制限を超えています',
'60004': '指定した数量がしきい値を下回っています',
'60005': '指定した価格が上限を上回っています',
'60006': '指定した価格が下限を下回っています',
'60011': '同時発注制限件数(30件)を上回っています',
'60016': '指定したトリガー価格が上限を上回っています',
'70001': 'システムエラーが発生しました。サポートにお問い合わせ下さい',
'70002': 'システムエラーが発生しました。サポートにお問い合わせ下さい',
'70003': 'システムエラーが発生しました。サポートにお問い合わせ下さい',
'70004': '現在取引停止中のため、注文を承ることができません',
'70005': '現在買い注文停止中のため、注文を承ることができません',
'70006': '現在売り注文停止中のため、注文を承ることができません',
'70009': 'ただいま成行注文を一時的に制限しています。指値注文をご利用ください',
'70010': 'ただいまシステム負荷が高まっているため、最小注文数量を一時的に引き上げています',
'70011': 'ただいまリクエストが混雑してます。しばらく時間を空けてから再度リクエストをお願いします',
'70012': 'システムエラーが発生しました。サポートにお問い合わせ下さい',
'70013': 'ただいまシステム負荷が高まっているため、注文および注文キャンセルを一時的に制限しています',
'70014': 'ただいまシステム負荷が高まっているため、出金申請および出金申請キャンセルを一時的に制限しています',
'70015': 'ただいまシステム負荷が高まっているため、貸出申請および貸出申請キャンセルを一時的に制限しています',
'70016': '貸出申請および貸出申請キャンセル停止中のため、リクエストを承ることができません',
'70017': '指定された銘柄は注文停止中のため、リクエストを承ることができません',
'70018': '指定された銘柄は注文およびキャンセル停止中のため、リクエストを承ることができません',
'70019': '注文はキャンセル中です',
'70020': '現在成行注文停止中のため、注文を承ることができません',
'70021': '指値注文価格が乖離率を超過しています',
'70022': '現在逆指値指値注文停止中のため、注文を承ることができません',
'70023': '現在逆指値成行注文停止中のため、注文を承ることができません'
}

2
logic/__init__.py Normal file
View File

@ -0,0 +1,2 @@
from logic.ma_cross import MA_Cross
from logic.mm import MM

84
logic/ma_cross.py Normal file
View File

@ -0,0 +1,84 @@
import asyncio
SHORT_PERIOD = 13
LONG_PERIOD = 200
EXPECT_CYCLE_DUR_IN_CANDLE_INTERVALS = 5
class MA_Cross:
def __init__(self, logger, pair, candlestick, lot, limit_rate, stop_rate):
self._logger = logger
self._pair = pair
self._candlestick = candlestick
self._lot = lot
self._limit_rate = limit_rate
self._stop_rate = stop_rate
asyncio.create_task(self._main())
async def _main(self):
self._logger.info("started")
while True:
await self._candlestick.wait()
if len(self._candlestick.values) < LONG_PERIOD:
continue
short = self._candlestick.SMA(SHORT_PERIOD)
long = self._candlestick.SMA(LONG_PERIOD)
plong = self._candlestick.SMA(LONG_PERIOD, 1)
expect_sell = self._pair.ticker.buy * (1+self._limit_rate)
ask_volume = self._pair.depth.askVolumeUntil(expect_sell)
interval_volume = self._pair.ticker.volume / 24 / 3600 * self._candlestick.interval
if plong < long and short > long:
if ask_volume < interval_volume * EXPECT_CYCLE_DUR_IN_CANDLE_INTERVALS:
await self._buy()
async def _buy(self):
self._logger.info("buy signal received")
# buy (market) and sell (limit)
amount = self._lot/self._pair.ticker.buy
try:
buy = await self._pair.buy_market(amount)
except Exception as e:
self._logger.warn("failed to buy", e)
return
while not buy.done:
await asyncio.sleep(0.1)
await buy.update()
self._logger.info(f"BUY confirmed (+{amount} / -{buy.price*amount})")
limit_price = buy.price*(1+self._limit_rate)
stop_price = buy.price*(1-self._stop_rate)
try:
stop = await self._pair.sell_stop(amount, stop_price)
limit = await self._pair.sell_limit(amount, limit_price)
except Exception as e:
print("failed to order", e.message)
self._logger.error("FAILED TO ORDER for STOPPING and LIMITATION!!")
return
self._logger.debug("waiting for STOP or LIMIT")
while (not stop.done) and (not limit.done):
await asyncio.sleep(2)
try:
await asyncio.gather(limit.update(), stop.update())
except Exception:
continue
if stop.done:
self._logger.info(f"STOP confirmed (-{amount} / +{stop.price*amount})")
# cancel limit and cooldown
await asyncio.gather(asycio.sleep(5), limit.cancel())
return
if limit.done:
self._logger.info(f"LIMIT confirmed (-{amount} / +{limit.price*amount})")
await stop.cancel()
return

82
logic/mm.py Normal file
View File

@ -0,0 +1,82 @@
import asyncio
class MM:
def __init__(self, logger, pair, lot, epsilon):
self._logger = logger
self._pair = pair
self._lot = lot
self._remain = 0
self._epsilon = epsilon
asyncio.create_task(self._main())
async def _main(self):
self._logger.info("started")
while True:
depth = self._pair.depth
await depth.wait()
if len(depth.bids) == 0 or len(depth.asks) == 0:
continue
spread = depth.asks[0][0] - depth.bids[0][0]
if spread > self._epsilon*2:
sell_price = depth.asks[0][0]-self._epsilon
buy_price = depth.bids[0][0]+self._epsilon
await self._make(sell_price, buy_price)
async def _make(self, sell_price, buy_price):
if self._remain < self._epsilon/buy_price:
amount = self._lot/buy_price
try:
buy = await self._pair.buy_limit(amount, buy_price, True)
except Exception:
buy = None
try:
sell = await self._pair.sell_limit(amount, sell_price, True)
except Exception:
sell = None
else:
buy = None
try:
sell = await self._pair.sell_limit(self._remain, sell_price, True)
except Exception:
sell = None
while (sell is not None) or (buy is not None):
await asyncio.sleep(1)
try:
coro = []
if buy is not None: coro.append(buy .update())
if sell is not None: coro.append(sell.update())
await asyncio.gather(*coro)
except Exception as e:
continue
depth = self._pair.depth
if sell is not None:
if sell.done:
self._remain -= sell.amount-sell.remain
sell = None
elif sell_price > depth.asks[0][0]:
break
if buy is not None:
if buy.done:
self._remain += buy.amount-buy.remain
buy = None
elif buy_price < depth.bids[0][0]:
break
async def cancel(self, order, flag):
await asyncio.sleep(1)
await order.cancel()
await order.update()
self._remain += flag*(order.amount-order.remain)
coro = []
if sell is not None:
coro.append(cancel(self, sell, -1))
if buy is not None:
coro.append(cancel(self, buy, 1))
await asyncio.gather(*coro)

32
main.py Normal file
View File

@ -0,0 +1,32 @@
import asyncio
import logging
import pybotters
import signal
import bitbank
async def main():
alive = True
logging.basicConfig(level=logging.DEBUG)
def teardown():
nonlocal alive
alive = False
for t in asyncio.all_tasks():
t.cancel()
loop = asyncio.get_event_loop()
loop.add_signal_handler(signal.SIGTERM, teardown)
loop.add_signal_handler(signal.SIGINT, teardown)
async with pybotters.Client(apis="secret.json") as cli:
await asyncio.gather(bitbank.init(cli))
while alive: await asyncio.sleep(0.5)
if __name__ == "__main__":
try:
loop = asyncio.new_event_loop()
loop.run_until_complete(main())
finally:
loop.close()

1
requirements.txt Normal file
View File

@ -0,0 +1 @@
pybotters

62
util/candlestick.py Normal file
View File

@ -0,0 +1,62 @@
import asyncio
import datetime
MAX_CANDLES = 200
OPEN = 0
HIGH = 0
LOW = 0
CLOSE = 0
class Candlestick:
def __init__(self, ticker, interval):
self.values = []
self.latest = None
self.interval = interval
self._ticker = ticker
self._event = asyncio.Event()
asyncio.create_task(self._refresh_and_start())
async def _refresh_and_start(self):
self.values = await self._fetch(datetime.datetime.now(tz=datetime.timezone.utc))
asyncio.create_task(self._ticker_watcher())
asyncio.create_task(self._main())
async def _fetch(self, data, depth = 0, rem = MAX_CANDLES):
assert(False)
async def _ticker_watcher(self):
while True:
await self._ticker.wait()
v = self._ticker.price
if self.latest is None:
self.latest = [v, -1e100, 1e100, None]
self.latest[1] = max(self.latest[1], v)
self.latest[2] = min(self.latest[2], v)
self.latest[3] = v
async def _main(self):
while True:
await asyncio.sleep(self.interval)
if self.latest is None:
if len(self.values) == 0:
continue
self.latest = self.values[0]
n = min(MAX_CANDLES-1, len(self.values))
self.values = [self.latest, *self.values[0:n]]
self.latest = None
self._event.set()
self._event.clear()
async def wait(self):
await self._event.wait()
def SMA(self, period, offset=0, type = CLOSE):
sum = 0
end = min(period+offset, len(self.values))
for i in range(offset, end):
sum += self.values[i][type]
return sum / (end - offset)

21
util/depth.py Normal file
View File

@ -0,0 +1,21 @@
import asyncio
MAX_ITEMS = 10
class Depth:
def __init__(self):
self.bids = []
self.asks = []
self._event = asyncio.Event()
async def wait(self):
await self._event.wait()
def askVolumeUntil(self, price):
ret = 0
for i in range(len(self.asks)):
if self.asks[i][0] > price: break
ret += self.asks[i][1]
return ret

48
util/pair.py Normal file
View File

@ -0,0 +1,48 @@
import asyncio
class Pair:
def __init__(self, name):
self.name = name
self.ticker = None
self.candlestick_1s = None
self.candlestick_1m = None
self.candlestick_1h = None
def buy_market(amount):
assert(False)
def sell_market(amount):
assert(False)
def buy_limit(amount, price):
assert(False)
def sell_limit(amount, price):
assert(False)
def buy_stop(amount, price):
assert(False)
def sell_stop(amount, price):
assert(False)
def buy_stop_limit(amount, trigger, price):
assert(False)
def sell_stop_limit(amount, trigger, price):
assert(False)
class Order:
def __init__(self):
self.price = None
self.amount = None
self.remain = None
self.done = False
async def update(self):
assert(False)
async def cancel(self):
assert(False)

14
util/ticker.py Normal file
View File

@ -0,0 +1,14 @@
import asyncio
class Ticker:
def __init__(self, pair):
self.pair = pair
self.price = None
self.sell = None
self.buy = None
self.volume = None
self.ts = None
self._event = asyncio.Event()
async def wait(self):
await self._event.wait()