update

test/wqb-login/login.py

@@ -0,0 +1,40 @@
+import httpx
+from httpx import BasicAuth
+import os
+
+def login(credentials_file='account.txt'):
+    """登录WorldQuant Brain API"""
+    # 读取本地账号密码
+    if not os.path.exists(credentials_file):
+        print("未找到 account.txt 文件")
+        with open(credentials_file, 'w') as f: f.write("")
+        print("account.txt 文件已创建，请填写账号密码, 格式: ['username', 'password]")
+        exit(1)
+
+    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()