39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
import httpx
|
|
from httpx import BasicAuth
|
|
|
|
|
|
def login(credentials_file='account.txt'):
|
|
"""登录WorldQuant Brain API"""
|
|
# 从nacos获取账号密码
|
|
nacos_resp = httpx.get('http://192.168.31.41:30848/nacos/v1/cs/configs?dataId=wq_account&group=quantify')
|
|
if nacos_resp.status_code != 200:
|
|
print('获取账号密码失败')
|
|
return False
|
|
|
|
config = nacos_resp.json()
|
|
|
|
username = config['user_name']
|
|
password = config['password']
|
|
|
|
print(f"正在登录账户: {username}")
|
|
|
|
# 创建客户端并认证
|
|
client = httpx.Client(auth=BasicAuth(username, password))
|
|
|
|
# 发送登录请求
|
|
response = client.post('https://api.worldquantbrain.com/authentication')
|
|
print(f"登录状态: {response.status_code}")
|
|
|
|
if response.status_code == 201:
|
|
print("登录成功!")
|
|
print(response.json())
|
|
return client
|
|
else:
|
|
print(f"登录失败: {response.json()}")
|
|
client.close()
|
|
return None
|
|
|
|
# 使用示例
|
|
client = login()
|
|
if client:
|
|
client.close() |