初始化项目

2026-06-02 17:46:38 +08:00
commit 646a4d02c0
240 changed files with 33662 additions and 0 deletions
@@ -0,0 +1,56 @@
+import asyncio
+import json
+
+import pandas as pd
+from tornado.httpclient import HTTPResponse, HTTPRequest
+
+from dock.govc import govc_api
+import models
+from models.govc_task_requester import GovcTaskRequester
+from paste.util import udict
+from paste.core.logging import echo_log
+
+
+async def get_requster_request(cguid: str):
+    """
+    获取市12345的工单的诉求人信息，请求响应是单条工单的数据
+
+    :param cguid: 工单列表请求返回的cguid
+    """
+    api_url = '/rest/sztaskworkordercommonrest/getInformation'
+    headers = {
+        'Referer': f'{govc_api.ApiUrl}/rest/sztaskworkordercommonrest/getInformation'
+    }
+    request_body = {
+        'caseguid': cguid, 'secret': 0
+    }
+    # 构造 API 请求
+    return await govc_api.new_api_request(api_url, request_body, headers=headers, method='GET')
+
+
+async def after_requester_request(response: HTTPResponse, retry_queue: asyncio.Queue[HTTPRequest]):
+    """
+    任务请求响应后的处理程序。
+
+    :param response: 响应对象
+    :param retry_queue: 重试队列
+    """
+    response_body = response.body.decode()
+    response_data = json.loads(response_body)
+    requester_info = udict.get_by_path(response_data, 'params')
+    if requester_info:
+        mapped_df = pd.DataFrame([requester_info])
+        # 更换映射方向，用于将源数据列名改为与数据库表对应
+        forward_mapping = {dict_f: table_f for table_f, dict_f in GovcTaskRequester.FieldMapping.items()}
+        mapped_df = mapped_df.rename(columns=forward_mapping)
+        # 这里把空数据都换成 None，以便存入数据库时是 null
+        mapped_df.replace(models.EmptyInDF + models.EmptyDatetimeInDF, None, inplace=True)
+        task_id = getattr(response.request, GovcTaskRequester.task_id.key)
+        mapped_df[GovcTaskRequester.task_id.key] = task_id
+        # 筛选数据状态
+        _created, _updated = await GovcTaskRequester.save_batch(mapped_df)
+        echo_log(f"成功创建诉求人信息：{_created}条，更新：{_updated}条.")
+    else:
+        echo_log('未获取到诉求人信息')
+    if retry_queue:
+        echo_log(f"诉求人信息重试队列中有：{retry_queue.qsize()} 个请求在等待.")