72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
import os
|
|
|
|
import requests
|
|
|
|
from common.config import Config
|
|
|
|
|
|
class LoginHelper:
|
|
"""
|
|
登录辅助类
|
|
"""
|
|
|
|
def __init__(self, base_url: str = Config.frontend_url) -> None:
|
|
self.base_url = base_url
|
|
self.session = requests.Session()
|
|
|
|
def login(self, username: str = 'admin', password: str | None = None, max_retries: int = 3) -> str | None:
|
|
"""
|
|
执行登录操作（在测试环境中，验证码已禁用）
|
|
"""
|
|
if password is None:
|
|
password = get_test_password(username)
|
|
|
|
for _attempt in range(max_retries):
|
|
# 在测试环境中，验证码已禁用，所以直接登录
|
|
login_data = {'username': username, 'password': password}
|
|
|
|
headers = {'Content-Type': 'application/x-www-form-urlencoded', 'Referer': f'{self.base_url}/login'}
|
|
|
|
response = self.session.post(f'{Config.backend_url}/login', data=login_data, headers=headers)
|
|
|
|
http_ok = 200
|
|
if response.status_code == http_ok:
|
|
result = response.json()
|
|
# 检查登录是否成功 - 可能是 code=200 或 success=true
|
|
if result.get('code') == http_ok or result.get('success'):
|
|
# 尝试从不同可能的位置获取token
|
|
token = result.get('token') # 直接在根级别
|
|
if not token:
|
|
token = result.get('data', {}).get('token') # 在data对象内
|
|
|
|
if token:
|
|
print('登录成功')
|
|
return token
|
|
print(f'登录失败: 响应中未找到token - {result}')
|
|
else:
|
|
print(f'登录失败: {result.get("msg", "未知错误")}')
|
|
else:
|
|
print(f'登录请求失败: 状态码 {response.status_code}')
|
|
|
|
print('登录失败，已达到最大重试次数')
|
|
return None
|
|
|
|
|
|
def get_test_password(username: str) -> str:
|
|
"""读取测试账号密码，避免把凭据写入测试代码或日志。"""
|
|
env_name = f'SHIERFENG_TEST_{username.upper()}_PASSWORD'
|
|
password = os.getenv(env_name)
|
|
if not password:
|
|
raise RuntimeError(f'未设置测试账号密码环境变量: {env_name}')
|
|
return password
|
|
|
|
|
|
# 使用示例
|
|
if __name__ == '__main__':
|
|
helper = LoginHelper()
|
|
token = helper.login()
|
|
if token:
|
|
print(f'获取到token: {token}')
|
|
else:
|
|
print('登录失败')
|