初始化项目

2026-06-02 17:46:38 +08:00
commit 646a4d02c0
240 changed files with 33662 additions and 0 deletions
@@ -0,0 +1,98 @@
+"""
+向 OA 平台上报数据已经完整推送。
+
+对应文档接口：11、签收
+"""
+import asyncio
+import json
+from typing import Optional, Union
+
+from sqlalchemy import select, desc
+from tornado.httpclient import HTTPResponse, HTTPRequest
+
+import dock
+from dock.oa import oa_api_request
+from models.dcm_push_status import DcmPushStatus
+from models.dcm_task import DcmTask
+from paste.core.logging import echo_log
+from paste.util import udict
+from paste.web import requests
+
+
+async def after_sign_task_request(response: HTTPResponse, retry_queue: asyncio.Queue[HTTPRequest]):
+    """
+    签收工单请求响应后的处理程序。
+
+    :param response: 响应对象
+    :param retry_queue: 重试队列
+    """
+    body = response.body.decode()
+    echo_log(body)
+    body_data = json.loads(body)
+    code = udict.get_by_path(body_data, 'code')
+    message = udict.get_by_path(body_data, 'msg')
+    if code == 200:
+        echo_log(f"签收工单成功.")
+    else:
+        echo_log(f"签收工单失败：{message}")
+
+    if retry_queue:
+        echo_log(f"签收工单重试队列中有：{retry_queue.qsize()} 个请求在等待.")
+
+
+async def sign_task(fetch_size: int = 50,
+                    task_id: Optional[Union[str, int, list[Union[str, int]]]] = None):
+    """
+    签收指定数量的工单任务
+
+    :param fetch_size: 本次签收的任务数量
+    :param task_id: 待办任务 ID 可选
+    """
+    # 根据条件获取目标任务 ID 列表（支持指定 task_id 或分页获取）
+    task_query = select(DcmTask.id).join(
+        DcmPushStatus, DcmPushStatus.dcm_task_id == DcmTask.id
+    ).where(
+        DcmPushStatus.push_task_status == 1,
+        DcmPushStatus.push_task_detail_status == 1,
+        DcmPushStatus.push_task_attachment_status == 1,
+        DcmPushStatus.push_task_extend_info_status == 1,
+        DcmPushStatus.push_task_file_upload_status == 1,
+        DcmPushStatus.push_task_more_info_status == 1,
+        DcmPushStatus.push_task_process_info_status == 1,
+    ).order_by(
+        desc(DcmTask.act_id)
+    )
+    if task_id:
+        if isinstance(task_id, list):
+            task_query = task_query.where(DcmTask.id.in_(task_id))
+            echo_log(f"本次签收待办列表：{task_id} 的工单...")
+        else:
+            task_query = task_query.where(DcmTask.id == task_id)
+            echo_log(f"本次签收待办：{task_id} 的工单...")
+    else:
+        task_query = task_query.limit(fetch_size)
+        echo_log(f"本次签收前 {fetch_size} 条待办工单...")
+
+    dcm_task_df = await DcmTask.query_as_df(task_query)
+    # 格式化为字符串
+    dcm_task_df[DcmTask.id.key] = dcm_task_df[DcmTask.id.key].astype(str)
+
+    echo_log(f"正在准备请求队列...")
+    # 构建请求队列
+    sign_request_queue = asyncio.Queue()
+    task_id_list = dcm_task_df[DcmTask.id.key].unique().tolist()
+    for task_id in task_id_list:
+        request = await oa_api_request.get_sign_task_request(task_id)
+        await sign_request_queue.put(request)
+    echo_log(f"开始签收工单...")
+    await requests.async_concurrency(
+        sign_request_queue, con_count=dock.CONCURRENCY_COUNT, retry=dock.MAX_RETRY_COUNT,
+        after_request=after_sign_task_request
+    )
+    echo_log(f"签收工单完成...")
+
+
+if __name__ == "__main__":
+    from paste.core import aio_pool
+    _runner = aio_pool.get_aio_runner()
+    _runner(sign_task(task_id=2054174091228876801))