add get_gm_P2WPKH_program

diff --git a/app/api/__init__.py b/app/api/__init__.py
index c539256..60171b3 100644 (file)
--- a/app/api/__init__.py
+++ b/app/api/__init__.py
@@ -31,6 +31,7 @@ from app.api.resources import Get_Gm_Child_Xprv
 from app.api.resources import Get_Gm_Child_Xpub
 from app.api.resources import Gm_Xpub_Verify
 from app.api.resources import Gm_Xprv_Sign
+from app.api.resources import Get_Gm_P2WPKH_Program
 
 
 blueprint = Blueprint('api', __name__, url_prefix='/api/v1')
@@ -65,4 +66,5 @@ api.add_resource(Get_Gm_Public_Key, '/get_gm_public_key')
 api.add_resource(Get_Gm_Child_Xprv, '/get_gm_child_xprv')
 api.add_resource(Get_Gm_Child_Xpub, '/get_gm_child_xpub')
 api.add_resource(Gm_Xpub_Verify, '/gm_xpub_verify')
-api.add_resource(Gm_Xprv_Sign, '/gm_xprv_sign')
\ No newline at end of file
+api.add_resource(Gm_Xprv_Sign, '/gm_xprv_sign')
+api.add_resource(Get_Gm_P2WPKH_Program, '/get_gm_P2WPKH_program')
\ No newline at end of file

diff --git a/app/api/resources.py b/app/api/resources.py
index d08e6b6..2a0f531 100644 (file)
--- a/app/api/resources.py
+++ b/app/api/resources.py
@@ -31,6 +31,7 @@ from app.model.key_gm import get_gm_child_xprv
 from app.model.key_gm import get_gm_child_xpub
 from app.model.key_gm import gm_xpub_verify
 from app.model.key_gm import gm_xprv_sign
+from app.model.receiver import get_gm_P2WPKH_program
 
 
 parser = reqparse.RequestParser()
@@ -309,4 +310,15 @@ class Gm_Xprv_Sign(Resource):
         xprv = args.get('xprv_str')
         message = args.get('message_str')
         signature = gm_xprv_sign(xprv, message)
-        return signature
\ No newline at end of file
+        return signature
+
+class Get_Gm_P2WPKH_Program(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        account_index = args.get('account_index_int')
+        address_index = args.get('address_index_int')
+        change = args.get('change_bool')
+        xpub = args.get('xpub_str')
+        control_program = get_gm_P2WPKH_program(account_index, address_index, change, xpub)
+        return control_program

diff --git a/app/model/receiver.py b/app/model/receiver.py
index 9c63ae4..18b22e9 100644 (file)
--- a/app/model/receiver.py
+++ b/app/model/receiver.py
@@ -1,5 +1,6 @@
 import hashlib
 from app.model import key
+from app.model import key_gm
 from app.model import segwit_addr
 import qrcode
 import pybase64
@@ -93,6 +94,51 @@ def create_P2WPKH_program(account_index_int, address_index_int, change_bool, xpu
     }
 
 
+# get_gm_P2WPKH_program create control program
+# You can get more test data from: https://gist.github.com/zcc0721/5b74b89cae9d75f752a3d2aa9086b9e6
+# Please attention:
+#   account_index_int >= 1
+#   address_index_int >= 1
+#   change_bool: true or false
+# test data 1:
+#   account_index_int: 1
+#   address_index_int: 1
+#   change_bool: false
+#   xpub_str: 03c74f3a946940d43e0f8c6da40680c0078e6e1008ca6ea869d57536c31b7ede20adc168c3698fa538fa587c4e519d1eb7a2593f178bfe0c93890a0f09e1634607
+#   control_program_str: 001434ca9c0cd289ecb8e899f1bdc0e9470442eae367
+# test data 2:
+#   account_index_int: 1
+#   address_index_int: 1
+#   change_bool: true
+#   xpub_str: 03c74f3a946940d43e0f8c6da40680c0078e6e1008ca6ea869d57536c31b7ede20adc168c3698fa538fa587c4e519d1eb7a2593f178bfe0c93890a0f09e1634607
+#   control_program: 0014ee20c5dfb6730d9abd8bf9c5516b8710bc118271
+# test data 3:
+#   account_index_int: 1
+#   address_index_int: 17
+#   change_bool: true
+#   xpub_str: 02914d51fcc3b90a87ffe3424995a9db8757a9d67812edd85207c47edc9f7f34e368684ae4d706f68c710fe1dbd20d73a8faaaf34966678a5d58486ac193a851ca
+#   control_program: 00149441d8a2e415c63c579d7998563472c1a7c4df2f
+# test data 4:
+#   account_index_int: 33
+#   address_index_int: 44
+#   change_bool: true
+#   xpub_str: 03603b2eb079257513d253a92ad45ce5ef12cc285dd8c13fc77c95844468f5eb4482f33c577c3a71ac733136b17d68b65a184b225431ab658555735e436fdb13e6
+#   control_program: 0014ac49fa87604998fb0ff9d5f83c6c8e803ca84de2
+def get_gm_P2WPKH_program(account_index_int, address_index_int, change_bool, xpub_str):
+    path_list = get_path_from_index(account_index_int, address_index_int, change_bool)['path']
+    child_xpub_str = key_gm.get_gm_child_xpub(xpub_str, path_list)['child_xpub']
+    child_public_key_str = key_gm.get_gm_public_key(child_xpub_str)['public_key']
+    child_public_key_byte = bytes.fromhex(child_public_key_str)
+
+    ripemd160 = hashlib.new('ripemd160')
+    ripemd160.update(child_public_key_byte)
+    public_key_hash_str = ripemd160.hexdigest()
+    control_program_str = '0014' + public_key_hash_str
+    return {
+        "control_program": control_program_str
+    }
+
+
 # create_address create address
 # You can get more test data from: https://gist.github.com/zcc0721/8f52d0a80a0a9f964e9d9d9a50e940c5
 # Please attention: