def edit_api_key(canTrade, canWithdraw, email):
config = get_config()
url = "https://" + config.get("host") + "/v1/user/api-keys/update"
token = login(email)
headers = get_headers(token)
secretKey = check_bind_ga(email)
if secretKey is None:
ga = ""
print("GA为空，请先绑定GA~~")
else:
print(secretKey)
ga = get_totp_token(secretKey)
userid = get_user_id_by_email(email)
apikey = get_api_key_by_userid(userid)
print(apikey)
description = "test_add_api_key_edited_" + time.strftime(
"%Y-%m-%d_%H_%M_%S")
payload = {
"description": description,
"ga": ga,
"apiKey": apikey,
"canRead": True,
"canTrade": canTrade,
"canWithdraw": canWithdraw,
"ipRestriction": "*"
}
response = requests.request("POST",
url,
data=json.dumps(payload),
headers=headers)
if "success" != response.json()["status"]:
print("edit api key fail, response= " + response.text)
else:
print("edit api key success, response= " + response.text)
return response
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.url = "http://" + config.get(
"host") + "/v1/json/user/change-password"
self.my_token = login(self.email)
self.headers = get_headers(self.my_token)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.ua_url = "https://" + config.get("host") + "/v1/user/accounts"
self.my_token = login(self.email)
print(self.my_token)
self.ua_headers = get_headers(self.my_token)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.url = "https://" + config.get(
"host") + "/v1/trade/getFeeRate?symbol=BTC_USDT"
self.my_token = login(self.email)
self.headers = get_headers(self.my_token)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.url = "https://" + config.get(
"host") + "/v1/google/user/security/getAuth"
self.my_token = login(self.email)
self.headers = get_headers(self.my_token)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.url = "https://" + config.get(
"host") + "/v1/feeRefunds/commissions"
self.my_token = login(self.email)
self.headers = get_headers(self.my_token)
def setUp(self):
self.config = get_config()
self.email = "*****@*****.**"
self.create_order_url = "https://" + self.config.get(
"host") + "/v1/trade/orders"
self.my_token = login(self.email)
self.headers = get_headers(self.my_token)
time.sleep(2)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.profile_url = "https://" + config.get(
"host") + "/v1/json/user/profile"
# self.my_token = login.login("*****@*****.**")
self.my_token = login(self.email)
self.profile_headers = get_headers(self.my_token)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.url = "https://" + config.get(
"host"
) + "/v1/trade/orders-by-pagination?pageNumber=1&pageSize=20&start=2019-06-18&end=2019-07-18&isFinalStatus=false&status=SUBMITTED%2CSEQUENCED%2CPARTIAL_FILLED&orderType=BUY_LIMIT%2CBUY_MARKET%2CSELL_LIMIT%2CSELL_MARKET"
self.my_token = login(self.email)
self.headers = get_headers(self.my_token)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.url = "https://" + config.get(
"host"
) + "/v1/trade/matchDetails-by-pagination?pageNumber=1&pageSize=60&start=2019-06-18&end=2019-07-18&orderType=BUY_LIMIT%2CBUY_MARKET%2CSELL_LIMIT%2CSELL_MARKET"
self.my_token = login(self.email)
print(self.my_token)
self.headers = get_headers(self.my_token)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.url = "https://" + config.get("host") + "/v1/user/phone/add"
self.my_token = login(self.email)
self.headers = {
'contentType': 'application/json',
'Authorization': self.my_token,
'accept': "application/json",
}
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.internal_url = "https://" + config.get(
"host") + "/v1/user/withdraw/send-phone-code?type=internal"
self.my_token = login(self.email)
self.headers = {
"Content-Type": "application/json",
"Authorization": self.my_token,
}
def get_api_keys(email):
config = get_config()
url = "https://" + config.get("host") + "/v1/user/api-keys"
token = login(email)
headers = get_headers(token)
response = requests.request("GET", url, headers=headers)
if "success" != response.json()["status"]:
print("get api keys error, response= " + response.text)
else:
print("get api keys success, response= " + response.text)
return response
def get_email(email):
config = get_config()
url = "https://" + config.get("host") + "/v1/json/user/email"
# my_token = login("*****@*****.**")
my_token = login(email)
headers = get_headers(my_token)
response = requests.request("GET", url, headers=headers)
if "success" != response.json()['status']:
print("change password get email error! error_code:" + response.json()['code'])
else:
print("change password get email ok !")
def logout(email):
config = get_config()
url = "https://" + config.get("host") + "/v1/sign-out"
token = login(email)
headers = get_headers(token)
response = requests.request("POST", url, headers=headers)
if "success" != response.json()["status"]:
print("logout fail ,error info " + response.text)
else:
print("logout success " + response.text)
return response
def check_referrer(referrerCode):
"""check 邀请码在系统是否存在"""
config = get_config()
# referrerCode = "qK0TW62N"
url = "https://" + config.get(
"host") + "/v1/check_referrer?referrerCode=" + referrerCode
headers = login.get_headers()
response = requests.request("GET", url, headers=headers)
if "success" != response.json()["status"]:
print("check referrer error, response: " + response.text)
else:
print("check referrer success, response: " + response.text)
def get_deposit_record(currency, email):
"""获取充值记录"""
config = get_config()
url = "https://" + config.get(
"host"
) + "/v1/user/deposit/record?currency=" + currency + "&pageSize=10&pageNumber=1&startTime=&endTime="
token = login(email)
headers = get_headers(token)
response = requests.request("GET", url, headers=headers)
if "success" != response.json()['status']:
print("get deposit record error, response:" + response.text)
else:
print("get deposit record success, response:" + response.text)
return response
def test_phone(self):
config = get_config()
url = "https://" + config.get("host") + "/v1/user/phone/sendMessage"
token = login()
headers = {
'Content-Type': 'application/json',
'Authorization': token,
}
payload = {"phoneNumber": "13190000001", "country": "86"}
response = requests.request("POST",
url,
data=json.dumps(payload),
headers=headers)
print(response.text)
def get_deposit_address_and_record(currency, email):
"""获取用户单个币种的基本充值信息"""
config = get_config()
print(config)
url = "https://" + config.get(
"host"
) + "/v1/user/deposit/address-and-record?currency=" + currency + "&limit=100"
token = login(email)
headers = get_headers(token)
response = requests.request("GET", url, headers=headers)
if "success" != response.json()['status']:
print("get deposit info error, response:" + response.text)
else:
print("get deposit info success, response:" + response.text)
return response
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.url = "https://" + config.get("host") + "/v1/trade/orders"
self.my_token = login(self.email)
time.sleep(2)
self.headers = get_headers(self.my_token)
self.amount = round(random.uniform(10, 12), 5)
self.price = round(random.uniform(2, 3), 2)
self.symbol = random.choice(['ETH_USDT', 'LTC_USDT'])
self.trade_type = random.choice(['BUY_LIMIT', 'SELL_LIMIT'])
if self.trade_type == "BUY_LIMIT":
self.price = self.price + 1
else:
self.price = self.price - 1
def findback_password_reset(email, vcode):
"""重置密码操作"""
config = get_config()
url = "https://" + config.get("host") + "/v1/json/findbackpassword/reset"
headers = get_headers()
# newPassword = "******"
newPassword = hmac_password(email, b"Password01")
payload = {"email": email, "vcode": vcode, "newPassword": newPassword}
response = requests.request("POST",
url,
data=json.dumps(payload),
headers=headers)
if "success" != response.json()["status"]:
print("reset password error , response: " + response.text)
else:
print("reset password success, response: " + response.text)
def find_password_re_send(email):
"""重新发送验证码"""
config = get_config()
url = "https://" + config.get("host") + "/v1/json/finapassword_re_send"
headers = get_headers()
payload = {"email": email}
response = requests.request("POST",
url,
data=json.dumps(payload),
headers=headers)
if "success" != response.json()["status"]:
print("findback password resend email error, response: " +
response.text)
else:
print("findback password resend email success, response: " +
response.text)
def login(email):
"""根据email登录并拿到token"""
config = get_config()
base_url = "https://" + config.get("host") + "/v1/sign-in"
headers = get_headers()
# if email is None:
# email = "*****@*****.**"
# secretKey = "OFCSSHNSK47IL4TJJ4X4GYBUY3V7BMJS"
# ga = ga_auth.get_totp_token(secretKey)
# else:
# email = email
# secretKey = check_bind_ga(email)
# if secretKey is None:
# ga = ""
# else:
# print(secretKey)
# # ga = ga_auth.get_totp_token(secretKey)
# userid = get_user_id_by_email(email)
# secretKey = get_user_GA_secret(userid).get("secretKey")
# ga = get_GA_code.get_google_code(secretKey)
# print(type(ga))
email = email
secretKey = check_bind_ga(email)
if secretKey is None:
ga = ""
else:
print(secretKey)
# ga = ga_auth.get_totp_token(secretKey)
userid = get_user_id_by_email(email)
secretKey = get_user_GA_secret(userid).get("secretKey")
ga = get_GA_code.get_google_code(secretKey)
print(type(ga))
password = hmac_password(email, b"Password01")
payload = {"email": email, "password": password, "ga": ga}
print("ga= " + str(ga))
response = requests.request("POST", base_url, headers=headers, data=json.dumps(payload))
status = response.json()['status']
if status != "success":
print("login fail！ response: " + response.text)
else:
print("login success, response: " + response.text)
my_token = response.json()['data']['token']
print("my_token=" + my_token)
return my_token
def register_action(email, refId=None):
"""注册时，有邀请码，可填写邀请码，无邀请码时，refId可不填"""
config = get_config()
vcode = register_vcode(email, refId)
payload = {"email": str(email), "zone": "+08:00", "code": vcode}
headers = {
'Content-Type': "application/json",
}
url = "https://" + config.get("host") + "/v1/register"
response = requests.request("POST",
url,
data=json.dumps(payload),
headers=headers)
if "success" != response.json()["status"]:
print("register error, response: " + response.text)
else:
print("register success, response: " + response.text)
return response
def delete_api_key(email):
config = get_config()
url = "https://" + config.get("host") + "/v1/user/api-keys/delete"
token = login(email)
headers = get_headers(token)
userid = get_user_id_by_email(email)
apikey = get_api_key_by_userid(userid)
print(apikey)
payload = {"apiKey": apikey}
response = requests.request("POST",
url,
data=json.dumps(payload),
headers=headers)
if "success" != response.json()["status"]:
print("delete api key fail, response= " + response.text)
else:
print("delete api key success, response= " + response.text)
return response
def register_vcode(email, refId=None):
"""获取注册时邮件vcode， 无邀请码时，可不填写refId"""
config = get_config()
password = hmac_password(email) # 调用方法获取加密后的password
headers = login.get_headers()
if refId is None:
payload = {
"email": email,
"password": password,
"enhancedValidateRequest": {
"challenge": "62234f6e396b3471802d33b7ef1a74f16f",
"validate": "1e8986dd53879b284c4b16b53f8cd19f",
"seccode": "1e8986dd53879b284c4b16b53f8cd19f|jordan",
"id": "43de7e0f-f16f-4ddc-bc5a-7bbc770e52dc"
}
}
else:
payload = {
"email": email,
"password": password,
"enhancedValidateRequest": {
"challenge": "62234f6e396b3471802d33b7ef1a74f16f",
"validate": "1e8986dd53879b284c4b16b53f8cd19f",
"seccode": "1e8986dd53879b284c4b16b53f8cd19f|jordan",
"id": "43de7e0f-f16f-4ddc-bc5a-7bbc770e52dc"
},
"refId": refId
}
base_url = "https://" + config.get("host") + "/v1/vcode/register"
response = requests.request("POST",
base_url,
headers=headers,
data=json.dumps(payload))
status = response.json()['status']
if status != "success":
print("get register code fail！ error_code:" + response.json()['code'])
else:
print("get register code success")
reg_vcode = get_vcode(email)
return reg_vcode
def login_with_param(email):
"""根据email登录并拿到token"""
config = get_config()
# email = "*****@*****.**"
password = hmac_password(email, b"Password01")
headers = {
'Accept': "application/json",
'Content-Type': "application/json",
'cache-control': "no-cache"
}
payload = {"email": email, "password": password, "ga": ""}
base_url = "https://" + config.get("host") + "/v1/sign-in"
response = requests.request("POST", base_url, headers=headers, data=json.dumps(payload))
status = response.json()['status']
if status != "success":
print("login fail！ error_code:" + response.json()['code'])
else:
print("login success")
my_token = response.json()['data']['token']
print("my_token=" + my_token)
return my_token
def findback_password(email):
"""找回密码发送验证码"""
config = get_config()
url = "https://" + config.get("host") + "/v1/json/findbackpassword"
headers = get_headers()
payload = {
"email": email,
"enhancedValidateRequest": {
"challenge": "014c66cc91d20941d389e1969d31e5ea4q",
"validate": "ccd019664487cb7c9078576e15fe7fed",
"seccode": "ccd019664487cb7c9078576e15fe7fed|jordan",
"id": "3d28c569-3904-4bb0-b4a2-20ccf1d53b96"
}
}
response = requests.request("POST",
url,
data=json.dumps(payload),
headers=headers)
if "success" != response.json()["status"]:
print("findback password error, response: " + response.text)
else:
print("findback password success, response: " + response.text)
def setUp(self):
config = get_config()
self.email = "*****@*****.**"
self.mt_url = "https://" + config.get("host") + "/v1/market/trades"
self.my_token = login(self.email)
self.mt_headers = get_headers(self.my_token)
def setUp(self):
config = get_config()
self.url = "https://" + config.get("host") + "/v1/sign-in"
self.headers = get_headers()
