版本回退

This commit is contained in:
whm
2026-02-03 14:54:34 +08:00
parent 28cddd7409
commit e130134791
2 changed files with 60 additions and 60 deletions

View File

@@ -124,68 +124,68 @@ def delete_account(request: AccountDeleteRequest, db: Session = Depends(get_db))
message="账号删除成功",
data=None
)
# 获取今日上传的数据接口
@router.post("/get_uplaod_data", response_model=AccountListResponse)
def get_account(request: AccountGetRequest, db: Session = Depends(get_db)):
"""根据多种条件查询账号,并合并外部接口的 is_ok 字段"""
# 1. 从数据库查询账号列表
accounts = AccountService.search_accounts(
db,
account_id=request.account_id,
username=request.username,
project_name=request.project_name,
status=request.status,
today_updated=request.today_updated,
yh_id=request.yh_id,
cl_name=request.cl_name
)
# # 获取今日上传的数据接口
# @router.post("/get_uplaod_data", response_model=AccountListResponse)
# def get_account(request: AccountGetRequest, db: Session = Depends(get_db)):
# """根据多种条件查询账号,并合并外部接口的 is_ok 字段"""
# # 1. 从数据库查询账号列表
# accounts = AccountService.search_accounts(
# db,
# account_id=request.account_id,
# username=request.username,
# project_name=request.project_name,
# status=request.status,
# today_updated=request.today_updated,
# yh_id=request.yh_id,
# cl_name=request.cl_name
# )
# 2. 调用外部接口获取 today_data构建 user_name -> is_ok 映射
user_is_ok_map = {} # 存储映射关系,提升匹配效率
url = "https://engineering.yuxindazhineng.com/index/index/get_over_data"
payload = {"user_id": "68c0dbfdb7cbcd616e7c5ab5"}
# # 2. 调用外部接口获取 today_data构建 user_name -> is_ok 映射
# user_is_ok_map = {} # 存储映射关系,提升匹配效率
# url = "https://engineering.yuxindazhineng.com/index/index/get_over_data"
# payload = {"user_id": "68c0dbfdb7cbcd616e7c5ab5"}
try:
# 发送POST请求并解析JSON响应
response = requests.request("POST", url, data=payload).json()
if response['code'] == 0:
today_data = response['data']
# try:
# # 发送POST请求并解析JSON响应
# response = requests.request("POST", url, data=payload).json()
# if response['code'] == 0:
# today_data = response['data']
# 遍历 today_data构建映射字典
for item in today_data:
# 校验字段是否存在,避免 KeyError 异常
if 'user_name' in item and 'is_ok' in item:
user_name = item['user_name']
is_ok = item['is_ok']
user_is_ok_map[user_name] = is_ok # 键user_nameis_ok
# # 遍历 today_data构建映射字典
# for item in today_data:
# # 校验字段是否存在,避免 KeyError 异常
# if 'user_name' in item and 'is_ok' in item:
# user_name = item['user_name']
# is_ok = item['is_ok']
# user_is_ok_map[user_name] = is_ok # 键user_nameis_ok
except Exception as e:
# 捕获接口调用/解析异常,不阻断核心业务(数据库查询结果正常返回)
print(f"外部接口调用失败或数据解析异常:{str(e)}")
# except Exception as e:
# # 捕获接口调用/解析异常,不阻断核心业务(数据库查询结果正常返回)
# print(f"外部接口调用失败或数据解析异常:{str(e)}")
# 3. 遍历 accounts给每个账号对象添加 is_ok 字段(关键步骤)
for account in accounts:
# 匹配 username数据库和 user_name外部接口
current_username = account.username
# 从映射字典中获取 is_ok无匹配则默认赋值 0可根据业务调整默认值
is_ok_value = user_is_ok_map.get(current_username, 0)
# # 3. 遍历 accounts给每个账号对象添加 is_ok 字段(关键步骤)
# for account in accounts:
# # 匹配 username数据库和 user_name外部接口
# current_username = account.username
# # 从映射字典中获取 is_ok无匹配则默认赋值 0可根据业务调整默认值
# is_ok_value = user_is_ok_map.get(current_username, 0)
# 给 account 对象添加 is_ok 字段(兼容 Pydantic 模型,需满足对应配置)
account.is_ok = is_ok_value
# # 给 account 对象添加 is_ok 字段(兼容 Pydantic 模型,需满足对应配置)
# account.is_ok = is_ok_value
# 4. 处理空结果返回
if not accounts:
return AccountListResponse(
code=ResponseCode.ACCOUNT_NOT_FOUND,
message=ResponseMessage.ACCOUNT_NOT_FOUND,
total=0,
data=[]
)
# # 4. 处理空结果返回
# if not accounts:
# return AccountListResponse(
# code=ResponseCode.ACCOUNT_NOT_FOUND,
# message=ResponseMessage.ACCOUNT_NOT_FOUND,
# total=0,
# data=[]
# )
# 5. 正常返回结果(包含 is_ok 字段的 accounts
return AccountListResponse(
code=ResponseCode.SUCCESS,
message="查询成功",
total=len(accounts),
data=accounts
)
# # 5. 正常返回结果(包含 is_ok 字段的 accounts
# return AccountListResponse(
# code=ResponseCode.SUCCESS,
# message="查询成功",
# total=len(accounts),
# data=accounts
# )