zwf
52 lines
2.3 KiB
Python
52 lines
2.3 KiB
Python
import asyncio
|
|
import json
|
|
import pandas as pd
|
|
import models
|
|
|
|
from tornado.httpclient import HTTPResponse, HTTPRequest
|
|
|
|
from dock.dcm import dcm_api
|
|
from paste.util import udict
|
|
from paste.core.logging import echo_log
|
|
from models.dcm_task_attachment import DcmTaskAttachment
|
|
|
|
|
|
async def get_attachment_request(relation_id: int, relation_type_id: int = 1):
|
|
"""
|
|
获取 DCM 企业待办的附件列表。
|
|
|
|
向 DCM 的附件查询接口发送 GET 请求，获取与指定关系 ID 和类型 ID 关联的附件信息。
|
|
自动注入有效的 Cookie（如 JSESSIONID）至请求头，并解析返回的 JSON 数据。
|
|
|
|
Args:
|
|
relation_id (int): 关联记录的 ID，例如任务 ID。
|
|
relation_type_id (int): 关联类型 ID，默认为 1（任务类型）。
|
|
"""
|
|
api_url = f"/home/mis/attachrec/getattach"
|
|
request_body = {
|
|
"relationID": relation_id,
|
|
"relationTypeID": relation_type_id,
|
|
}
|
|
# 构造 API 请求
|
|
return await dcm_api.new_api_request(api_url, request_body, 'GET')
|
|
|
|
|
|
async def after_attachment_request(response: HTTPResponse, retry_queue: asyncio.Queue[HTTPRequest]):
|
|
response_body = response.body.decode()
|
|
response_data = json.loads(response_body)
|
|
list_data = udict.get_by_path(response_data, 'resultInfo.data.mediaList')
|
|
attachment_df = pd.DataFrame(list_data)
|
|
# 更换映射方向，用于将源数据列名改为与数据库表对应
|
|
forward_mapping = {dict_f: table_f for table_f, dict_f in DcmTaskAttachment.FieldMapping.items()}
|
|
mapped_df = attachment_df.rename(columns=forward_mapping)
|
|
# 这里把空数据都换成 None，以便存入数据库时是 null
|
|
mapped_df.replace(models.EmptyInDF + models.EmptyDatetimeInDF, None, inplace=True)
|
|
dcm_task_id = getattr(response.request, 'dcm_task_id')
|
|
rec_id = getattr(response.request, 'rec_id')
|
|
mapped_df[DcmTaskAttachment.dcm_task_id.key] = dcm_task_id
|
|
mapped_df[DcmTaskAttachment.rec_id.key] = rec_id
|
|
# 筛选数据状态
|
|
_created, _updated = await DcmTaskAttachment.save_batch(mapped_df)
|
|
echo_log(f"成功创建企业待办 {rec_id} 的附件：{_created}条，更新：{_updated}条.")
|
|
if retry_queue:
|
|
echo_log(f"企业待办附件重试队列中有：{retry_queue.qsize()} 个请求在等待.") |