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()