diff --git a/bot split/order.py b/bot split/order.py index 43caa36..988e8c4 100644 --- a/bot split/order.py +++ b/bot split/order.py @@ -39,11 +39,11 @@ class OrderManager: self._send_timestamps.append(now) return True - def sell(self, symbol: str, dir: Dir, price: int, size: int): - return self._attempt_send("add", order_id=self._next_order(), symbol=symbol, dir=dir, price=price, size=size) + def sell(self, symbol: str, price: int, size: int): + return self._attempt_send("add", order_id=self._next_order(), symbol=symbol, dir=Dir.SELL, price=price, size=size) - def buy(self, symbol: str, dir: Dir, price: int, size: int): - return self._attempt_send("add", order_id=self._next_order(), symbol=symbol, dir=dir, price=price, size=size) + def buy(self, symbol: str, price: int, size: int): + return self._attempt_send("add", order_id=self._next_order(), symbol=symbol, dir=Dir.BUY, price=price, size=size) def convert(self, symbol: str, dir: Dir, size: int): return self._attempt_send("convert", order_id=self._next_order(), symbol=symbol, dir=dir, size=size)