OSDN Git Service

fix bug
[bytom/bytom-kit.git] / app / api / resources.py
1 from flask_restful import Resource
2 from flask_restful import reqparse
3 from app.model.hello import get_hello_result
4 from app.model.signature import sign
5 from app.model.signature import verify
6 from app.model.key import create_entropy
7 from app.model.key import entropy_to_mnemonic
8 from app.model.key import mnemonic_to_seed
9 from app.model.key import seed_to_root_xprv
10
11 parser = reqparse.RequestParser()
12 parser.add_argument('private_key_str', type=str)
13 parser.add_argument('message_str', type=str)
14 parser.add_argument('public_key_str', type=str)
15 parser.add_argument('signature_str', type=str)
16 parser.add_argument('entropy_str', type=str)
17 parser.add_argument('mnemonic_str', type=str)
18 parser.add_argument('seed_str', type=str)
19
20 class Hello(Resource):
21
22     def get(self, content):
23         return get_hello_result(content)
24
25 class Sign(Resource):
26
27     def post(self):
28         args = parser.parse_args()
29         private_key = args.get('private_key_str')
30         message = args.get('message_str')
31         sig = sign(private_key, message)
32         return sig
33
34 class Verify(Resource):
35     
36     def post(self):
37         args = parser.parse_args()
38         public_key = args.get('public_key_str')
39         signature = args.get('signature_str')
40         message = args.get('message_str')
41         result = verify(public_key, signature, message)
42         return result
43
44 class Create_Entropy(Resource):
45
46     def post(self):
47         entropy = create_entropy()
48         return entropy
49
50 class Entropy_To_Mnemonic(Resource):
51
52     def post(self):
53         args = parser.parse_args()
54         entropy_str = args.get('entropy_str')
55         mnemonic_str = entropy_to_mnemonic(entropy_str)
56         return mnemonic_str
57
58 class Mnemonic_To_Seed(Resource):
59
60     def post(self):
61         args = parser.parse_args()
62         mnemonic_str = args.get('mnemonic_str')
63         seed_str = mnemonic_to_seed(mnemonic_str)
64         return seed_str
65
66 class Seed_To_Root_Xprv(Resource):
67
68     def post(self):
69         args = parser.parse_args()
70         seed_str = args.get('seed_str')
71         root_xprv_str = seed_to_root_xprv(seed_str)
72         return root_xprv_str