fix an issue that the MM-algorithm dies after losscut

algo/mm.py

@@ -7,58 +7,99 @@ class MM:
     self._player = player
     self._pair   = pair
 
-    self._lastBuy  = ""
-    self._lastSell = ""
+    self._lastBuy   = ""
+    self._lastSell  = ""
+    self._lastPrice = None
 
-    self._delta = float(config["delta"])
+    self._delta        = float(config["delta"])
+    self._positiveCut  = float(config["+cut"])
+    self._negativeCut  = float(config["-cut"])
 
   async def update(self):
+    sales = self._player.assets[self._pair.names[0]]
+    crncy = self._player.assets[self._pair.names[1]]
+    price = self._pair.ticker.price
+
+    lot = (sales.amount + crncy.amount / price) / 2
+    pos = sales.amount / lot - 1
+
+    # evaluate if cutting should be executed
+    negCut = False
+    posCut = False
+    if self._lastPrice is not None:
+      change  = (price - self._lastPrice) / self._lastPrice
+      negCut  = pos >  0.5 and change < -abs(self._negativeCut)
+      posCut  = pos < -0.5 and change >  abs(self._positiveCut)
+    cutting = negCut or posCut
+
+    # check if the past orders are active
     lastBuyAgreed  = (self._lastBuy is not None and
                       self._lastBuy not in self._player.orders)
     lastSellAgreed = (self._lastSell is not None and
                       self._lastSell not in self._player.orders)
 
-    if not lastBuyAgreed and not lastSellAgreed:
+    if not lastBuyAgreed and not lastSellAgreed and not cutting:
       return
 
+    # cancel active orders
     cancel_orders = []
     if not lastBuyAgreed and self._lastBuy is not None:
+      logging.info(f"cancel buy")
       cancel_orders.append(self._lastBuy)
-      self._lastBuy = None
+    elif lastBuyAgreed:
+      logging.info(f"agree buy")
+
     if not lastSellAgreed and self._lastSell is not None:
+      logging.info(f"cancel sell")
       cancel_orders.append(self._lastSell)
-      self._lastSell = None
+    elif lastSellAgreed:
+      logging.info(f"agree sell")
+
     if len(cancel_orders) != 0:
       await self._player.cancel(self._pair.names, cancel_orders)
+    self._lastBuy  = None
+    self._lastSell = None
 
-    sales = self._player.assets[self._pair.names[0]]
-    crncy = self._player.assets[self._pair.names[1]]
-    price = self._pair.ticker.price
-
-    lot = (sales.amount + crncy.amount / price) / 4
-    pos = sales.amount / lot - 2
-
-    buy_amount  = _quad( pos) * lot
-    sell_amount = _quad(-pos) * lot
+    # execute cutting
+    if cutting:
+      if negCut:
+        logging.info(f"-cut (change: {change}, pos: {pos} -> 0)")
+      if posCut:
+        logging.info(f"+cut (change: {change}, pos: {pos} -> 0)")
+      self._lastBuy   = ""
+      self._lastSell  = ""
+      sell_amount = pos * lot
+      if sell_amount > 0:
+        await self._player.orderMarketSell(self._pair.names, amount=sell_amount)
+      else:
+        await self._player.orderMarketBuy(self._pair.names, amount=-buy_amount)
+      return
 
+    # execute new cycle
+    self._lastPrice = None
     buy_price  = price * (1 - self._delta);
     sell_price = price * (1 + self._delta);
 
-    order_unit = pow(10, -sales.precision)
+    buy_amount  = min(_quad( pos) * lot, crncy.amount / buy_price)
+    sell_amount = min(_quad(-pos) * lot, sales.amount)
+    order_unit  = pow(10, -sales.precision)
+
+    logging.info(f"new cycle (pos: {pos}, price: {price}, buy: {buy_amount}, sell: {sell_amount})")
 
     async def buy():
-      return (None if order_unit > buy_amount else
+      self._lastBuy = (
+          None if order_unit > buy_amount else
           await self._player.orderLimitBuy(
               self._pair.names, amount = buy_amount, price = buy_price))
 
     async def sell():
-      return (None if order_unit > sell_amount else
+      self._lastSell = (
+          None if order_unit > sell_amount else
           await self._player.orderLimitSell(
               self._pair.names, amount = sell_amount, price = sell_price))
 
-    orders = await asyncio.gather(buy(), sell())
-    self._lastBuy  = orders[0]
-    self._lastSell = orders[1]
+    await asyncio.gather(buy(), sell())
+    self._lastPrice = price
 
 # https://kijitora-2018.hatenablog.com/entry/2018/12/23/102913
 def _quad(x):

main.py

@@ -27,6 +27,7 @@ async def main():
   loop.add_signal_handler(signal.SIGTERM, teardown)
   loop.add_signal_handler(signal.SIGINT,  teardown)
 
+  logging.info("#### TMM ####")
   async with pybotters.Client(apis={"bitbank":config["auth"]}) as pb:
     player = Player(pb)
     pair   = Pair(pb, config["pair"])

util/player.py

@@ -1,4 +1,5 @@
 # No copyright
+import asyncio
 import logging
 
 class Player:
@@ -12,7 +13,24 @@ class Player:
       "pair": f"{pair[0]}_{pair[1]}",
       "order_ids": orders
     })
-    self._check(res)
+
+  async def orderMarketSell(self, pair, amount):
+    res = await self._post("user/spot/order", data={
+      "pair": f"{pair[0]}_{pair[1]}",
+      "amount": str(amount),
+      "side": "sell",
+      "type": "market",
+    })
+    return res["data"]["order_id"]
+
+  async def orderMarketBuy(self, pair, amount):
+    res = await self._post("user/spot/order", data={
+      "pair": f"{pair[0]}_{pair[1]}",
+      "amount": str(amount),
+      "side": "buy",
+      "type": "market",
+    })
+    return res["data"]["order_id"]
 
   async def orderLimitSell(self, pair, amount, price):
     res = await self._post("user/spot/order", data={
@@ -22,7 +40,7 @@ class Player:
       "side": "sell",
       "type": "limit",
     })
-    return self._check(res)["data"]["order_id"]
+    return res["data"]["order_id"]
 
   async def orderLimitBuy(self, pair, amount, price):
     res = await self._post("user/spot/order", data={
@@ -32,7 +50,7 @@ class Player:
       "side": "buy",
       "type": "limit",
     })
-    return self._check(res)["data"]["order_id"]
+    return res["data"]["order_id"]
 
   async def update(self):
     self.assets = {}
@@ -47,12 +65,24 @@ class Player:
       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())
+    for i in range(10):
+      try:
+        res = await self._pb.post(f"https://api.bitbank.cc/v1/{suffix}", data=data)
+        return self._check(await res.json())
+      except Exception as e:
+        err = e
+        await asyncio.sleep(1)
+    raise Exception(f"API error: {suffix} ({err})")
 
   async def _get(self, suffix):
-    res = await self._pb.get(f"https://api.bitbank.cc/v1/{suffix}")
-    return self._check(await res.json())
+    for i in range(10):
+      try:
+        res = await self._pb.get(f"https://api.bitbank.cc/v1/{suffix}")
+        return self._check(await res.json())
+      except Exception as e:
+        err = e
+        await asyncio.sleep(1)
+    raise Exception(f"API error: {suffix} ({err})")
 
   def _check(self, json):
     if "success" not in json or 1 != json["success"]: