You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.0 KiB
71 lines
2.0 KiB
import httpx
|
|
from httpx import BasicAuth, Timeout
|
|
|
|
|
|
def login():
|
|
"""登录WorldQuant Brain API"""
|
|
try:
|
|
# 从nacos获取账号密码
|
|
with httpx.Client(timeout=10.0) as temp_client:
|
|
nacos_resp = temp_client.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.get('user_name')
|
|
password = config.get('password')
|
|
|
|
if not username or not password:
|
|
print('账号密码不完整')
|
|
return False
|
|
|
|
print(f"正在登录账户: {username}")
|
|
|
|
# 创建客户端并设置超时（关键修复）
|
|
# 设置更长的超时时间：连接30秒，读取60秒
|
|
timeout = Timeout(connect=30.0, read=60.0, write=30.0, pool=30.0)
|
|
client = httpx.Client(
|
|
auth=BasicAuth(username, password),
|
|
timeout=timeout
|
|
)
|
|
|
|
# 发送登录请求
|
|
response = client.post(
|
|
'https://api.worldquantbrain.com/authentication')
|
|
|
|
if response.status_code == 201:
|
|
print("登录成功!")
|
|
# 登录成功后将client对象返回出来
|
|
return client
|
|
else:
|
|
print(
|
|
f"登录失败: {response.status_code} - {response.text}")
|
|
client.close()
|
|
return False
|
|
|
|
except Exception as e:
|
|
print(f"登录异常: {e}")
|
|
return False
|
|
|
|
|
|
def fetch_alpha_by_id(alpha_id: str, client: httpx.Client):
|
|
"""根据Alpha ID获取Alpha详情"""
|
|
url = f"https://api.worldquantbrain.com/alphas/{alpha_id}"
|
|
response = client.get(url)
|
|
if response.status_code == 200:
|
|
return response.json()
|
|
else:
|
|
return {}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
client = login()
|
|
if not client:
|
|
exit(1)
|
|
|
|
alpha = fetch_alpha_by_id('O0mnm8zd', client)
|
|
print(alpha)
|
|
|