diff --git a/libraries/wrappers/earn_okx.py b/libraries/wrappers/earn_okx.py index 38d917b..7370cd4 100644 --- a/libraries/wrappers/earn_okx.py +++ b/libraries/wrappers/earn_okx.py @@ -76,8 +76,12 @@ class okx_earn: response: {'code': '0', 'data': [{'amt': '10', 'ccy': 'USDT', 'clientId': '', 'from': '18', 'to': '6', 'transId': '1064005710'}], 'msg': ''} ''' - transfer = self.funding_api.funds_transfer(coin,amount,"18","6","0") - return transfer + response = self.funding_api.funds_transfer(coin,amount,"18","6","0") + if response["data"]!=[]: + return {"Success": "", + "transId": response["data"][0]["transId"]} + else: + return {"Error": response} def transfer_to_trading(self,coin,amount): @@ -102,8 +106,13 @@ class okx_earn: response: {'code': '0', 'data': [{'amt': '20', 'ccy': 'USDT', 'clientId': '', 'from': '6', 'to': '18', 'transId': '1064008141'}], 'msg': ''} ''' - transfer = self.funding_api.funds_transfer(coin,amount,"6","18","0") - return transfer + response = self.funding_api.funds_transfer(coin,amount,"6","18","0") + if response["data"]!=[]: + return {"Success": "", + "amount": response["data"][0]["amt"], + "transId": response["data"][0]["transId"]} + else: + return {"Error": response} def get_transfer_state(self, transaction_id): @@ -134,8 +143,13 @@ class okx_earn: response {'code': '0', 'data': [{'amt': '10', 'ccy': 'USDT', 'clientId': '', 'from': '18', 'instId': '', 'state': 'success', 'subAcct': '', 'to': '6', 'toInstId': '', 'transId': '1064005710', 'type': '0'}], 'msg': ''} ''' - transfer = self.funding_api.transfer_state(transaction_id) - return transfer + response = self.funding_api.transfer_state(transaction_id) + if response["data"][0]["state"]=="success": + return {"Success": "", + "amount": response["data"][0]["amt"], + "transId": response["data"][0]["transId"]} + else: + return {"Error": response} def get_available_products(self, coin): diff --git a/main.py b/main.py index 95c7030..05677d1 100644 --- a/main.py +++ b/main.py @@ -10,6 +10,7 @@ binance = earn_binance.binance_earn() okx = earn_okx.okx_earn() gateio = earn_gateio.gateio_earn() + if __name__=="__main__": '''