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.
33 lines
925 B
33 lines
925 B
import httpx
|
|
from httpx import BasicAuth
|
|
|
|
def login(credentials_file='brain_credentials.txt'):
|
|
"""登录WorldQuant Brain API"""
|
|
# 读取本地账号密码
|
|
with open(credentials_file) as f:
|
|
credentials = eval(f.read())
|
|
username, password = credentials[0], credentials[1]
|
|
|
|
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
|
|
|
|
# 使用示例
|
|
if __name__ == "__main__":
|
|
client = login()
|
|
if client:
|
|
client.close() |