zwf
96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
"""
|
|
向 OA 平台上报数据已经完整推送。
|
|
|
|
对应文档接口：8、签收
|
|
"""
|
|
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.govs_push_status import GovsPushStatus
|
|
from models.govs_order_master import GovsOrderMaster
|
|
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):
|
|
"""
|
|
签收指定数量的工单任务，或签收指定id的工单
|
|
|
|
:param fetch_size: 本次签收的任务数量
|
|
:param task_id: 待办任务 ID 可选
|
|
"""
|
|
# 根据条件获取目标任务 ID 列表（支持指定 task_id 或分页获取）
|
|
task_query = select(GovsOrderMaster.id).join(
|
|
GovsPushStatus, GovsPushStatus.master_id == GovsOrderMaster.id
|
|
).where(
|
|
GovsPushStatus.push_order_status == 1,
|
|
GovsPushStatus.push_order_detail_status == 1,
|
|
GovsPushStatus.push_order_process_status == 1
|
|
).order_by(
|
|
desc(GovsOrderMaster.id)
|
|
)
|
|
if task_id is not None:
|
|
if isinstance(task_id, list):
|
|
task_query = task_query.where(GovsOrderMaster.id.in_(task_id))
|
|
echo_log(f"本次签收待办列表：{task_id} 的工单...")
|
|
else:
|
|
task_query = task_query.where(GovsOrderMaster.id == task_id)
|
|
echo_log(f"本次签收待办：{task_id} 的工单...")
|
|
else:
|
|
task_query = task_query.limit(fetch_size)
|
|
echo_log(f"本次签收前 {fetch_size} 条待办工单...")
|
|
|
|
govs_task_df = await GovsOrderMaster.query_as_df(task_query)
|
|
# 格式化为字符串
|
|
govs_task_df[GovsOrderMaster.id.key] = govs_task_df[GovsOrderMaster.id.key].astype(str)
|
|
|
|
echo_log(f"正在准备请求队列...")
|
|
# 构建请求队列
|
|
sign_request_queue = asyncio.Queue()
|
|
task_id_list = govs_task_df[GovsOrderMaster.id.key].unique().tolist()
|
|
for task_id in task_id_list:
|
|
request = await oa_api_request.get_sign_govs_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=2057694366475214849))
|