Files
2026-06-02 17:46:38 +08:00

97 lines
3.3 KiB
Python

"""
接受OA请求,操作数字城管的工单批转接口
"""
import logging
from typing import Optional
from apps.api import dcm
from apps.app_handler import AppHandler
from dock.dcm import dcm_push_dispose
from models.dcm_dispose import DcmDispose
from models.dcm_task import DcmTask
from paste.core import aio_pool
from paste.core.logging import echo_log
from paste.web.decorators import route
@route(f'{dcm.ApiPrefix}/transfer')
class DisposeHandler(AppHandler):
"""
批转接口。
对接数字城管系统的批转接口,请求后本接口先将数据保存本地,然后响应客户端,然后开始后台启动推送。
"""
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.dcm_task: Optional[DcmTask] = None
self.dcm_dispose: Optional[DcmDispose] = None
def _params_for_db(self, **kwargs: dict) -> dict:
"""
提取数据库所需参数。
"""
return {
DcmDispose.flow_token.key: kwargs.get('flowToken', ''),
DcmDispose.dcm_task_id.key: kwargs.get('gdId', ''),
DcmDispose.act_id.key: self.dcm_task.act_id,
DcmDispose.task_number.key: kwargs.get('taskNumber', ''),
DcmDispose.opinion.key: kwargs.get('opinion', ''),
DcmDispose.attachments.key: kwargs.get('attachments', ''),
DcmDispose.send_message.key: kwargs.get('sendMessage', '1'),
DcmDispose.trans_info.key: '52,254,0,0',
DcmDispose.add_num.key: kwargs.get('addNum', '0'),
DcmDispose.task_list_id.key: kwargs.get('taskListId', '600058'),
DcmDispose.undertake_user_name.key: kwargs.get('undertakeUserName', ''),
DcmDispose.undertake_phone.key: kwargs.get('undertakePhone', '')
}
async def dispose(self, **kwargs) -> dict:
# 必填参数校验
required_keys = ['gdId', 'taskNumber', 'opinion', 'attachments', 'flowToken']
missing = [
k for k in required_keys
if k not in kwargs or kwargs[k] is None
]
if missing:
raise ValueError(f"缺少必要参数: {missing}")
# 读取待办任务对象
dcm_task_id = kwargs.get('gdId', '')
self.dcm_task = await DcmTask.async_find_by_id(dcm_task_id)
# 保存请求数据
params = self._params_for_db(**kwargs)
self.dcm_dispose = DcmDispose().copy_from_dict(params)
self.dcm_dispose.status = 0
await self.dcm_dispose.async_save()
# 后台执行提交批转请求到数字城管
await aio_pool.run_background_task(
dcm_push_dispose.push_dispose(self.dcm_dispose, self.dcm_task)
)
return {
'msg': '批转成功.'
}
# @auth_token
async def post(self):
"""
处理 POST 请求。
---
tags:
- D3I API
summary: 批转接口
"""
try:
echo_log(self.request.body.decode())
_, params = self.get_request_params()
_result = await self.dispose(**params)
self.response_ok(code=0, data=_result)
except Exception as e:
self.response_error(e, status_code=200, api_status_code=500)
self.log(msg=e, level=logging.ERROR, is_log_exc=True)