safeheron-api-sdk-python 1.1.13__tar.gz → 1.1.15__tar.gz

{safeheron_api_sdk_python-1.1.13 → safeheron_api_sdk_python-1.1.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: safeheron_api_sdk_python
-Version: 1.1.13
+Version: 1.1.15
 Summary: Python for Safeheron API
 Home-page: https://github.com/Safeheron/safeheron-api-sdk-python
 Author: safeheron

{safeheron_api_sdk_python-1.1.13 → safeheron_api_sdk_python-1.1.15}/safeheron_api_sdk_python/api/account_api.py

@@ -31,7 +31,7 @@ class OneAccountRequest:
 
 class CreateAccountRequest:
     def __init__(self):
-        # Account name, within 30 characters
+        # Account name, 50 characters max
         self.accountName = None
         # Merchant unique business ID (100 characters max)
         # The customerRefId uniquely represents a wallet. In the case of duplicate customerRefId values (for example, when resubmitting due to request timeouts or other errors), the data returned by the interface will remain consistent
@@ -49,7 +49,7 @@ class CreateAccountRequest:
 
 class BatchCreateAccountRequest:
     def __init__(self):
-        # The prefix of wallet account name, 30 characters max
+        # The prefix of wallet account name, 50 characters max
         self.accountName = None
         # Display status in Safeheron App
         # True: not display
@@ -132,6 +132,11 @@ class InfoAccountCoinAddressRequest:
         # Coin receiving address
         self.address = None
 
+class AccountCoinBalanceRequest:
+    def __init__(self):
+        # Coin Keys, max 10
+        self.coinKeyList = None
+
 
 class RenameAccountCoinAddressRequest:
     def __init__(self):
@@ -160,9 +165,9 @@ class BatchCreateAccountCoinUTXORequest:
         self.coinKey = None
         # Account key
         self.accountKey = None
-        # Address group name, 30 characters max
-        self.count = None
         # The number, max is 100
+        self.count = None
+        # Address group name, 30 characters max
         self.addressGroupName = None
 
 
@@ -240,6 +245,12 @@ class AccountApi:
     def info_account_coin_address(self, request: InfoAccountCoinAddressRequest):
         return self.api_client.send_request(request, '/v1/account/coin/address/info')
 
+    # Retrieve Coin Balance
+    # Get the coin balance of all asset wallets under the team.
+    def account_coin_balance(self, request: AccountCoinBalanceRequest):
+        return self.api_client.send_request(request, '/v1/account/coin/balance')
+
+
     # Rename Coin Address Group of a Wallet Account
     # Rename a coin address group of a wallet account.
     def rename_account_coin_address(self, request: RenameAccountCoinAddressRequest):

safeheron_api_sdk_python-1.1.15/safeheron_api_sdk_python/api/tools_api.py

@@ -0,0 +1,32 @@
+from safeheron_api_sdk_python.client import *
+
+
+class AmlCheckerRequestRequest:
+    def __init__(self):
+        # Blockchain network, supports:
+        # Bitcoin
+        # Ethereum
+        # Tron
+        self.network = None
+        # Address
+        self.address = None
+
+
+class AmlCheckerRetrievesRequest:
+    def __init__(self):
+        # Risk assessment request ID, which can be created through the Create AML Risk Assessment Request interface.
+        self.requestId = None
+
+
+class ToolsApi:
+
+    def __init__(self, config):
+        self.api_client = Client(config)
+
+    # Create AML Risk Assessment Request
+    def aml_checker_request(self, request: AmlCheckerRequestRequest):
+        return self.api_client.send_request(request, '/v1/tools/aml-checker/request')
+
+    # Retrieve AML Risk Assessment Result
+    def aml_checker_retrieves(self, request: AmlCheckerRetrievesRequest):
+        return self.api_client.send_request(request, '/v1/tools/aml-checker/retrieves')

{safeheron_api_sdk_python-1.1.13 → safeheron_api_sdk_python-1.1.15}/safeheron_api_sdk_python/api/web3_api.py

@@ -3,7 +3,7 @@ from safeheron_api_sdk_python.client import *
 
 class CreateWeb3AccountRequest:
     def __init__(self):
-        # Account name, within 30 characters
+        # Account name, 50 characters max
         self.accountName = None
         # 	Whether display in Safeheron Console
         # True: not display
@@ -17,10 +17,15 @@ class CreateWeb3AccountRequest:
 
 class BatchCreateWeb3AccountRequest:
     def __init__(self):
-        # Transaction key
-        self.txKey = None
-        # Merchant unique business ID (100 characters max)
-        self.customerRefId = None
+        # The prefix of wallet account name, 50 characters max
+        self.accountName = None
+        # 	Whether display in Safeheron Console
+        # True: not display
+        # False: display
+        # Default: false
+        self.hiddenOnUI = None
+        # Number of wallets to be created, greater than 0, less than 100
+        self.count = None
 
 
 class ListWeb3AccountRequest:

{safeheron_api_sdk_python-1.1.13 → safeheron_api_sdk_python-1.1.15}/safeheron_api_sdk_python/cosigner/co_signer_converter.py

@@ -9,18 +9,33 @@ class CoSignerResponse:
         self.txKey = None
 
 
+class CoSignerResponseV3:
+    def __init__(self):
+        # action
+        self.action = None
+        # approvalId
+        self.approvalId = None
+
+
 class CoSignerConverter:
 
     def __init__(self, config):
-        self.api_pub_key = config['apiPubKey']
-        if config.get('bizPrivKey'):
-            self.biz_privKey = PEM_PRIVATE_HEAD + config['bizPrivKey'] + PEM_PRIVATE_END
-        if config.get('bizPrivKeyPemFile'):
-            self.biz_privKey = load_rsa_private_key(config['bizPrivKeyPemFile'])
+        # Supports both coSignerPubKey and apiPublKey
+        self.co_signer_pub_key = config.get('coSignerPubKey') or config.get('apiPubKey')
+
+        # Supports both approvalCallbackServicePrivateKey and bizPrivKey
+        private_key = config.get('approvalCallbackServicePrivateKey') or config.get('bizPrivKey')
+        if private_key:
+            self.approval_callback_service_private_key = PEM_PRIVATE_HEAD + private_key + PEM_PRIVATE_END
+
+        # Supports both approvalCallbackServicePrivateKeyPemFile and bizPrivKeyPemFile
+        pem_file = config.get('approvalCallbackServicePrivateKeyPemFile') or config.get('bizPrivKeyPemFile')
+        if pem_file:
+            self.approval_callback_service_private_key = load_rsa_private_key(pem_file)
 
     def request_convert(self, co_signer_call_back):
-        platform_rsa_pk = get_rsa_key(PEM_PUBLIC_HEAD + self.api_pub_key + PEM_PUBLIC_END)
-        api_user_rsa_sk = get_rsa_key(self.biz_privKey)
+        platform_rsa_pk = get_rsa_key(PEM_PUBLIC_HEAD + self.co_signer_pub_key + PEM_PUBLIC_END)
+        api_user_rsa_sk = get_rsa_key(self.approval_callback_service_private_key)
         required_keys = {
             'key',
             'sig',
@@ -65,10 +80,50 @@ class CoSignerConverter:
 
         return json.loads(r.decode())
 
+    def request_v3_convert(self, co_signer_call_back):
+        platform_rsa_pk = get_rsa_key(PEM_PUBLIC_HEAD + self.co_signer_pub_key + PEM_PUBLIC_END)
+        required_keys = {
+            'version',
+            'sig',
+            'bizContent',
+            'timestamp',
+        }
+
+        missing_keys = required_keys.difference(co_signer_call_back.keys())
+        if missing_keys:
+            raise Exception(co_signer_call_back)
+        co_signer_call_back['version'] = 'v3'
+        sig = co_signer_call_back.pop('sig')
+
+        need_sign_message = sort_request(co_signer_call_back)
+        v = rsa_pass_verify(platform_rsa_pk, need_sign_message, sig)
+        if not v:
+            raise Exception("rsa verify: false")
+        return json.loads(b64decode(co_signer_call_back['bizContent']).decode())
+
+    def response_v3_converter(self, co_signer_response: CoSignerResponseV3):
+        api_user_rsa_sk = get_rsa_key(self.approval_callback_service_private_key)
+        ret = dict()
+        response_data = json.dumps(co_signer_response.__dict__).replace('\'', '\"').replace('\n', '').encode('utf-8')
+
+        if response_data is not None:
+            ret['bizContent'] = b64encode(response_data).decode()
+
+        ret['timestamp'] = str(int(time.time() * 1000))
+        ret['code'] = str('200')
+        ret['version'] = str('v3')
+        ret['message'] = str('SUCCESS')
+
+        # 4 sign request
+        need_sign_message = sort_request(ret)
+        ret['sig'] = rsa_pss_sign(api_user_rsa_sk, need_sign_message)
+        return ret
+
+
     # It has been Deprecated,Please use convertCoSignerResponseWithNewCryptoType
     def response_converter(self, co_signer_response: CoSignerResponse):
-        platform_rsa_pk = get_rsa_key(PEM_PUBLIC_HEAD + self.api_pub_key + PEM_PUBLIC_END)
-        api_user_rsa_sk = get_rsa_key(self.biz_privKey)
+        platform_rsa_pk = get_rsa_key(PEM_PUBLIC_HEAD + self.co_signer_pub_key + PEM_PUBLIC_END)
+        api_user_rsa_sk = get_rsa_key(self.approval_callback_service_private_key)
 
         ret = dict()
 
@@ -98,8 +153,8 @@ class CoSignerConverter:
         return ret
 
     def response_converter_with_new_crypto_type(self, co_signer_response: CoSignerResponse):
-        platform_rsa_pk = get_rsa_key(PEM_PUBLIC_HEAD + self.api_pub_key + PEM_PUBLIC_END)
-        api_user_rsa_sk = get_rsa_key(self.biz_privKey)
+        platform_rsa_pk = get_rsa_key(PEM_PUBLIC_HEAD + self.co_signer_pub_key + PEM_PUBLIC_END)
+        api_user_rsa_sk = get_rsa_key(self.approval_callback_service_private_key)
 
         ret = dict()
 

{safeheron_api_sdk_python-1.1.13 → safeheron_api_sdk_python-1.1.15}/safeheron_api_sdk_python/tools.py

@@ -8,6 +8,7 @@ from Cryptodome.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5
 from Cryptodome.Hash import SHA256, SHA1
 from Cryptodome import Random
 from Crypto.Cipher import PKCS1_OAEP
+from Cryptodome.Signature import pss
 import json
 import time
 import requests
@@ -147,6 +148,16 @@ def rsa_sign(private_key, message_hash: bytes):
     except Exception as e:
         raise Exception("rsa sign error: %s" % e)
 
+def rsa_pss_sign(private_key, message_hash: bytes):
+    try:
+        signer = pss.new(private_key)
+        digest = SHA256.new()
+        digest.update(message_hash)
+        signed = signer.sign(digest)
+        return b64encode(signed).decode()
+    except Exception as e:
+        raise Exception("rsa sign error: %s" % e)
+
 
 def rsa_verify(public_key, message_hash: bytes, signature):
     try:
@@ -157,6 +168,16 @@ def rsa_verify(public_key, message_hash: bytes, signature):
     except Exception as e:
         raise Exception("rsa sign error: %s" % e)
 
+def rsa_pass_verify(public_key, message_hash: bytes, signature):
+    try:
+        verifier = pss.new(public_key)
+        digest = SHA256.new()
+        digest.update(message_hash)
+        verifier.verify(digest, b64decode(signature))
+        return True
+    except Exception as e:
+        raise Exception("rsa sign error: %s" % e)
+
 
 def sort_request(r: dict):
     sortData = json.dumps(r, sort_keys=True).replace(' ', '')

{safeheron_api_sdk_python-1.1.13 → safeheron_api_sdk_python-1.1.15}/safeheron_api_sdk_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: safeheron_api_sdk_python
-Version: 1.1.13
+Version: 1.1.15
 Summary: Python for Safeheron API
 Home-page: https://github.com/Safeheron/safeheron-api-sdk-python
 Author: safeheron

{safeheron_api_sdk_python-1.1.13 → safeheron_api_sdk_python-1.1.15}/safeheron_api_sdk_python.egg-info/SOURCES.txt

@@ -11,6 +11,7 @@ safeheron_api_sdk_python.egg-info/top_level.txt
 safeheron_api_sdk_python/api/account_api.py
 safeheron_api_sdk_python/api/coin_api.py
 safeheron_api_sdk_python/api/mpc_sign_api.py
+safeheron_api_sdk_python/api/tools_api.py
 safeheron_api_sdk_python/api/transaction_api.py
 safeheron_api_sdk_python/api/web3_api.py
 safeheron_api_sdk_python/api/webhook_api.py

{safeheron_api_sdk_python-1.1.13 → safeheron_api_sdk_python-1.1.15}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_packages
 with open("README.rst", "r") as f:
     long_description = f.read()
 setup(name='safeheron_api_sdk_python',
-      version='1.1.13',
+      version='1.1.15',
       description='Python for Safeheron API',
       long_description=long_description,
       author='safeheron',