OSDN Git Service

add path_list test data
[bytom/bytom-kit.git] / app / api / resources.py
index 6d904a3..8eff581 100644 (file)
@@ -1,16 +1,41 @@
+from flask_restful import inputs
 from flask_restful import Resource
 from flask_restful import reqparse
 from app.model.hello import get_hello_result
 from app.model.signature import sign
 from app.model.signature import verify
 from app.model.key import create_entropy
+from app.model.key import entropy_to_mnemonic
+from app.model.key import mnemonic_to_seed
+from app.model.key import seed_to_root_xprv
+from app.model.key import xprv_to_expanded_private_key
+from app.model.key import xpub_to_public_key
+from app.model.key import xpub_verify
+from app.model.key import xprv_sign
+from app.model.key import xprv_to_xpub
+from app.model.key import xprv_sign
+from app.model.key import xprv_to_child_xprv
+from app.model.key import xpub_to_child_xpub
+from app.model.receiver import create_P2WPKH_program
+from app.model.receiver import create_address
+from app.model.receiver import get_path_from_index
 
 parser = reqparse.RequestParser()
-parser.add_argument('private_key', type=str)
-parser.add_argument('message', type=str)
-parser.add_argument('public_key', type=str)
-parser.add_argument('signature', type=str)
-parser.add_argument('message', type=str)
+parser.add_argument('private_key_str', type=str)
+parser.add_argument('message_str', type=str)
+parser.add_argument('public_key_str', type=str)
+parser.add_argument('signature_str', type=str)
+parser.add_argument('entropy_str', type=str)
+parser.add_argument('mnemonic_str', type=str)
+parser.add_argument('seed_str', type=str)
+parser.add_argument('xprv_str', type=str)
+parser.add_argument('xpub_str', type=str)
+parser.add_argument('path_list', type=str, action='append')
+parser.add_argument('account_index_int', type=int)
+parser.add_argument('address_index_int', type=int)
+parser.add_argument('change_bool', type=inputs.boolean)
+parser.add_argument('control_program_str', type=str)
+parser.add_argument('network_str', type=str)
 
 class Hello(Resource):
 
@@ -21,8 +46,8 @@ class Sign(Resource):
 
     def post(self):
         args = parser.parse_args()
-        private_key = args.get('private_key')
-        message = args.get('message')
+        private_key = args.get('private_key_str')
+        message = args.get('message_str')
         sig = sign(private_key, message)
         return sig
 
@@ -30,9 +55,9 @@ class Verify(Resource):
     
     def post(self):
         args = parser.parse_args()
-        public_key = args.get('public_key')
-        signature = args.get('signature')
-        message = args.get('message')
+        public_key = args.get('public_key_str')
+        signature = args.get('signature_str')
+        message = args.get('message_str')
         result = verify(public_key, signature, message)
         return result
 
@@ -40,4 +65,120 @@ class Create_Entropy(Resource):
 
     def post(self):
         entropy = create_entropy()
-        return entropy
\ No newline at end of file
+        return entropy
+
+class Entropy_To_Mnemonic(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        entropy_str = args.get('entropy_str')
+        mnemonic_str = entropy_to_mnemonic(entropy_str)
+        return mnemonic_str
+
+class Mnemonic_To_Seed(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        mnemonic_str = args.get('mnemonic_str')
+        seed_str = mnemonic_to_seed(mnemonic_str)
+        return seed_str
+
+class Seed_To_Root_Xprv(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        seed_str = args.get('seed_str')
+        root_xprv_str = seed_to_root_xprv(seed_str)
+        return root_xprv_str
+
+class Xprv_To_Expanded_Private_Key(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xprv_str = args.get('xprv_str')
+        expanded_private_key_str = xprv_to_expanded_private_key(xprv_str)
+        return expanded_private_key_str
+
+class Xpub_To_Public_Key(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xpub_str = args.get('xpub_str')
+        public_key_str = xpub_to_public_key(xpub_str)
+        return public_key_str
+
+class Xpub_Verify(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xpub = args.get('xpub_str')
+        message = args.get('message_str')
+        signature = args.get('signature_str')
+        result = xpub_verify(xpub, message, signature)
+        return result
+
+class Xprv_Sign(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xprv = args.get('xprv_str')
+        message = args.get('message_str')
+        signature = xprv_sign(xprv, message)
+        return signature
+
+class Xprv_To_Xpub(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xprv = args.get('xprv_str')
+        xpub = xprv_to_xpub(xprv)
+        return xpub
+
+class Xprv_To_Child_Xprv(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xprv = args.get('xprv_str')
+        path = args.get('path_list')
+        child_xprv = xprv_to_child_xprv(xprv, path)
+        return child_xprv
+
+class Xpub_To_Child_Xpub(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xpub = args.get('xpub_str')
+        path = args.get('path_list')
+        child_xpub = xpub_to_child_xpub(xpub, path)
+        return child_xpub
+
+class Get_Path_From_Index(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')
+        path_list = get_path_from_index(account_index, address_index, change)
+        return path_list
+
+
+class Create_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 = create_P2WPKH_program(account_index, address_index, change, xpub)
+        return control_program
+
+class Create_Address(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        control_program = args.get('control_program_str')
+        network = args.get('network_str')
+        address = create_address(control_program, network)
+        return address
\ No newline at end of file