Add utils for decimal, functions related to sending and cancelling orders.

Author: Qiu-INJ

src/injective/chain_client/_transaction.py

@@ -74,6 +74,84 @@ def add_exchange_msg_deposit(
         }
         self._msgs.append(msg)
 
+    def add_exchange_msg_cancel_spot_order(self, subaccount: str, market_id: int, order_hash: str) -> None:
+        msg = {
+            "type": "exchange/MsgCancelSpotOrder",
+            "value": {
+                "sender": privkey_to_address(self._privkey, hrp=self._hrp),
+                "subaccount_id": subaccount,
+                "market_id": market_id,
+                "order_hash": order_hash,
+            },
+        }
+        self._msgs.append(msg)
+
+    def add_exchange_msg_batch_cancel_spot_order(self, subaccount_id_list, market_id_list, order_hash_list) -> None:
+        msg = {
+            #INFO: ATTENTION, Order`s`
+            "type": "exchange/MsgBatchCancelSpotOrders",
+            "value": {
+                "sender": privkey_to_address(self._privkey, hrp=self._hrp),
+                "data": []
+            },
+        }
+
+        for i in range(len(subaccount_id_list)):
+            msg["value"]["data"].append(
+                {
+                    "subaccount_id": subaccount_id_list[i],
+                    "market_id": market_id_list[i],
+                    "order_hash": order_hash_list[i],
+                }
+            )
+
+        self._msgs.append(msg)
+
+    def add_exchange_msg_create_spot_limit_order(self, subaccount_id: str,  market_id: str, fee_recipient: str, price, quantity, order_type, trigger_price) -> None:
+        msg = {
+            "type": "exchange/MsgCreateSpotLimitOrder",
+            "value": {
+                "sender": privkey_to_address(self._privkey, hrp=self._hrp),
+                "order": {
+                    'market_id': market_id,
+                    'order_info': {
+                        'subaccount_id': subaccount_id,
+                        'fee_recipient': fee_recipient,
+                        'price': price,
+                        'quantity': quantity,
+                    },
+                    'order_type': order_type,
+                    "trigger_price": trigger_price,
+                },
+            }
+        }
+        self._msgs.append(msg)
+
+    def add_exchange_msg_batch_create_spot_limit_orders(self, subaccount_id_list,  market_id_list, fee_recipient_list, price_list, quantity_list, order_type_list, trigger_price_list) -> None:
+        msg = {
+            "type": "exchange/MsgBatchCreateSpotLimitOrders",
+            "value": {
+                "sender": privkey_to_address(self._privkey, hrp=self._hrp),
+                "orders": []
+            }
+        }
+
+        for i in range(len(subaccount_id_list)):
+            msg["value"]["orders"].append({
+                'market_id': market_id_list[i],
+                'order_info': {
+                    'subaccount_id': subaccount_id_list[i],
+                    'fee_recipient': fee_recipient_list[i],
+                    'price': price_list[i],
+                    'quantity': quantity_list[i],
+                },
+                'order_type': order_type_list[i],
+                "trigger_price": trigger_price_list[i],
+            }
+            )
+        self._msgs.append(msg)
+
+
     def get_signed(self) -> str:
         pubkey = privkey_to_pubkey(self._privkey)
         base64_pubkey = base64.b64encode(pubkey).decode("utf-8")

src/injective/constant.py

@@ -0,0 +1,16 @@
+# from etherscan.io
+DECIMALS_DICT = {
+    "INJ": 18,  # uint8
+    "USDT": 6,  # uint256
+    "BNB": 18,  # uint8
+    "LINK": 18,  # uint8
+    "UNI": 18,  # uint8
+    "AAVE": 18,  # uint8 from https://etherscan.io/token/0x7fc66500c84a76ad7e9c93437bfc5ac33e2ddae9#readProxyContract
+    "MATIC": 18,  # uint8
+    "ZRX": 18,  # uint8
+    "USDC": 6,  # uint8 from https://etherscan.io/token/0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48#readProxyContract
+}
+
+
+ORDERTYPE_DICT = {"UNSPECIFIED": 0, "BUY": 1, "SELL": 2,
+                  "STOP_BUY": 3, "STOP_SELL": 4, "TAKE_BUY": 5, "TAKE_SELL": 6}

src/injective/utils.py

@@ -0,0 +1,42 @@
+from decimal import Decimal
+
+
+def price_float_to_string(price, base_decimals, quote_decimals, precision=18) -> str:
+    scale = Decimal(quote_decimals - base_decimals)
+    exchange_price = Decimal(price) * pow(10, scale)
+    price_string = ("{:."+str(precision)+"f}").format(exchange_price)
+    print("price string :{}".format(price_string))
+    return price_string
+
+
+def quantity_float_to_string(quantity, base_decimals, precision=18) -> str:
+    scale = Decimal(base_decimals)
+    exchange_quantity = Decimal(quantity) * pow(10, scale)
+    quantity_string = ("{:."+str(precision)+"f}").format(exchange_quantity)
+    print("quantity string:{}".format(quantity_string))
+    return quantity_string
+
+
+def price_string_to_float(price_string, base_decimals, quote_decimals) -> float:
+    scale = float(base_decimals - quote_decimals)
+    return float(price_string) * pow(10, scale)
+
+
+def quantity_string_to_float(quantity_string, base_decimals) -> float:
+    scale = float(0 - base_decimals)
+    return float(quantity_string) * pow(10, scale)
+
+
+# test
+if __name__ == "__main__":
+    assert "0.000000005000000000" == price_float_to_string(
+        5000, 18, 6, 18), "result of get_price is wrong."
+
+    assert "1222000000000000000000.000000000000000000" == quantity_float_to_string(
+        1222, 18, 18), "result of get_quantity is wrong."
+
+    assert 5000 == price_string_to_float(price_float_to_string(
+        5000, 18, 6, 18), 18, 6), "something is wrong."
+
+    assert 1222 == quantity_string_to_float(
+        quantity_float_to_string(1222, 18, 18), 18), "something is wrong."