Add orderhash example

Author: nhannamsiu

examples/sync/chain_client/Orderhashes.py

@@ -0,0 +1,149 @@
+# Copyright 2021 Injective Labs
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Injective Chain Tx/Query client for Python. Example only."""
+
+import sys
+sys.path.insert(0, '/Users/nam/desktop/injective/sdk-python/')
+
+
+import asyncio
+import logging
+
+from pyinjective.composer import Composer as ProtoMsgComposer
+from pyinjective.client import Client
+from pyinjective.transaction import Transaction
+from pyinjective.constant import Network
+from pyinjective.wallet import PrivateKey, PublicKey, Address
+from pyinjective.orderhash import compute_order_hashes
+
+async def main() -> None:
+    # select network: local, testnet, mainnet
+    network = Network.testnet()
+    composer = ProtoMsgComposer(network=network.string())
+
+    # initialize grpc client
+    client = Client(network, insecure=False)
+
+    # load account
+    priv_key = PrivateKey.from_hex("5d386fbdbf11f1141010f81a46b40f94887367562bd33b452bbaa6ce1cd1381e")
+    pub_key = priv_key.to_public_key()
+    address = pub_key.to_address().init_num_seq(network.lcd_endpoint)
+    subaccount_id = address.get_subaccount_id(index=0)
+
+    # prepare trade info
+    spot_market_id = "0xa508cb32923323679f29a032c70342c147c17d0145625922b0ef22e955c844c0"
+    deriv_market_id = "0x4ca0f92fc28be0c9761326016b5a1a2177dd6375558365116b5bdda9abc229ce"
+    fee_recipient = "inj1hkhdaj2a2clmq5jq6mspsggqs32vynpk228q3r"
+
+    spot_orders = [
+        composer.SpotOrder(
+            market_id=spot_market_id,
+            subaccount_id=subaccount_id,
+            fee_recipient=fee_recipient,
+            price=7.523,
+            quantity=0.01,
+            is_buy=True
+        ),
+        composer.SpotOrder(
+            market_id=spot_market_id,
+            subaccount_id=subaccount_id,
+            fee_recipient=fee_recipient,
+            price=27.92,
+            quantity=0.01,
+            is_buy=False
+        ),
+    ]
+
+    deriv_orders = [
+        composer.DerivativeOrder(
+            market_id=deriv_market_id,
+            subaccount_id=subaccount_id,
+            fee_recipient=fee_recipient,
+            price=41027,
+            quantity=0.01,
+            is_buy=True,
+            leverage=0.7,
+        ),
+        composer.DerivativeOrder(
+            market_id=deriv_market_id,
+            subaccount_id=subaccount_id,
+            fee_recipient=fee_recipient,
+            price=62140,
+            quantity=0.01,
+            is_buy=False,
+            is_reduce_only=True
+        ),
+    ]
+
+    # prepare tx msg
+    spot_msg = composer.MsgBatchCreateSpotLimitOrders(
+        sender=address.to_acc_bech32(),
+        orders=spot_orders
+    )
+
+    deriv_msg = composer.MsgBatchCreateDerivativeLimitOrders(
+        sender=address.to_acc_bech32(),
+        orders=deriv_orders
+    )
+
+    # compute order hashes
+    order_hashes = compute_order_hashes(network, spot_orders + deriv_orders)
+    print(order_hashes)
+
+    # build sim tx
+    tx = (
+        Transaction()
+        .with_messages(spot_msg,deriv_msg)
+        .with_sequence(address.get_sequence())
+        .with_account_num(address.get_number())
+        .with_chain_id(network.chain_id)
+    )
+    sim_sign_doc = tx.get_sign_doc(pub_key)
+    sim_sig = priv_key.sign(sim_sign_doc.SerializeToString())
+    sim_tx_raw_bytes = tx.get_tx_data(sim_sig, pub_key)
+
+    # simulate tx
+    (simRes, success) = client.simulate_tx(sim_tx_raw_bytes)
+    if not success:
+        print(simRes)
+        return
+
+    sim_res_msg = ProtoMsgComposer.MsgResponses(simRes.result.data, simulation=True)
+    print("simulation msg response")
+    print(sim_res_msg)
+
+    # # build tx
+    # gas_price = 500000000
+    # gas_limit = simRes.gas_info.gas_used + 20000 # add 20k for gas, fee computation
+    # fee = [composer.Coin(
+    #     amount=gas_price * gas_limit,
+    #     denom=network.fee_denom,
+    # )]
+    # current_height = client.get_latest_block().block.header.height
+    # tx = tx.with_gas(gas_limit).with_fee(fee).with_memo("").with_timeout_height(current_height+50)
+    # sign_doc = tx.get_sign_doc(pub_key)
+    # sig = priv_key.sign(sign_doc.SerializeToString())
+    # tx_raw_bytes = tx.get_tx_data(sig, pub_key)
+    #
+    # # broadcast tx: send_tx_async_mode, send_tx_sync_mode, send_tx_block_mode
+    # res = client.send_tx_block_mode(tx_raw_bytes)
+    # res_msg = ProtoMsgComposer.MsgResponses(res.data)
+    # print("tx response")
+    # print(res)
+    # print("tx msg response")
+    # print(res_msg)
+
+if __name__ == "__main__":
+    logging.basicConfig(level=logging.INFO)
+    asyncio.get_event_loop().run_until_complete(main())