78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
|
# No copyright
|
||
|
|
import logging
|
||
|
|
|
||
|
|
class Player:
|
||
|
|
def __init__(self, pb):
|
||
|
|
self._pb = pb
|
||
|
|
self.assets = {}
|
||
|
|
self.orders = {}
|
||
|
|
|
||
|
|
async def cancel(self, pair, orders):
|
||
|
|
res = await self._post("user/spot/cancel_orders", data={
|
||
|
|
"pair": f"{pair[0]}_{pair[1]}",
|
||
|
|
"order_ids": orders
|
||
|
|
})
|
||
|
|
self._check(res)
|
||
|
|
|
||
|
|
async def orderLimitSell(self, pair, amount, price):
|
||
|
|
res = await self._post("user/spot/order", data={
|
||
|
|
"pair": f"{pair[0]}_{pair[1]}",
|
||
|
|
"amount": str(amount),
|
||
|
|
"price": str(price),
|
||
|
|
"side": "sell",
|
||
|
|
"type": "limit",
|
||
|
|
})
|
||
|
|
return self._check(res)["data"]["order_id"]
|
||
|
|
|
||
|
|
async def orderLimitBuy(self, pair, amount, price):
|
||
|
|
res = await self._post("user/spot/order", data={
|
||
|
|
"pair": f"{pair[0]}_{pair[1]}",
|
||
|
|
"amount": str(amount),
|
||
|
|
"price": str(price),
|
||
|
|
"side": "buy",
|
||
|
|
"type": "limit",
|
||
|
|
})
|
||
|
|
return self._check(res)["data"]["order_id"]
|
||
|
|
|
||
|
|
async def update(self):
|
||
|
|
self.assets = {}
|
||
|
|
self.orders = {}
|
||
|
|
|
||
|
|
assets = (await self._get("user/assets"))["data"]["assets"]
|
||
|
|
for asset in assets:
|
||
|
|
self.assets[asset["asset"]] = Asset(asset)
|
||
|
|
|
||
|
|
orders = (await self._get("user/spot/active_orders"))["data"]["orders"]
|
||
|
|
for order in orders:
|
||
|
|
self.orders[order["order_id"]] = Order(order)
|
||
|
|
|
||
|
|
async def _post(self, suffix, data):
|
||
|
|
res = await self._pb.post(f"https://api.bitbank.cc/v1/{suffix}", data=data)
|
||
|
|
return self._check(await res.json())
|
||
|
|
|
||
|
|
async def _get(self, suffix):
|
||
|
|
res = await self._pb.get(f"https://api.bitbank.cc/v1/{suffix}")
|
||
|
|
return self._check(await res.json())
|
||
|
|
|
||
|
|
def _check(self, json):
|
||
|
|
if "success" not in json or 1 != json["success"]:
|
||
|
|
code = json["data"]["code"]
|
||
|
|
raise Exception(f"API error: {code}")
|
||
|
|
return json
|
||
|
|
|
||
|
|
|
||
|
|
class Asset:
|
||
|
|
def __init__(self, json):
|
||
|
|
self.name = json["asset"]
|
||
|
|
self.amount = float(json["onhand_amount"])
|
||
|
|
self.locked = float(json["locked_amount"])
|
||
|
|
self.precision = int(json["amount_precision"])
|
||
|
|
|
||
|
|
class Order:
|
||
|
|
def __init__(self, json):
|
||
|
|
self.id = json["order_id"]
|
||
|
|
self.pair = json["pair"]
|
||
|
|
self.amount = float(json["start_amount"])
|
||
|
|
self.remain = float(json["remaining_amount"])
|
||
|
|
self.price = float(json["average_price"])
|