483 lines
16 KiB
Python
483 lines
16 KiB
Python
# +======================================================================
|
|
# | DjangoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
|
# +----------------------------------------------------------------------
|
|
# | Copyright (c) 2021-2026 小蚂蚁云团队 版权所有
|
|
# +----------------------------------------------------------------------
|
|
# | 开源协议: Apache-2.0
|
|
# | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
|
# +----------------------------------------------------------------------
|
|
# | 官方网站: https://www.djangoadmin.cn
|
|
# +----------------------------------------------------------------------
|
|
# | 研发团队: 小蚂蚁云团队 荣誉出品
|
|
# +======================================================================
|
|
# | 版权和免责声明:
|
|
# | 一、知识产权声明
|
|
# | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
|
# | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
|
# | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
|
# | 二、使用限制与禁止条款
|
|
# | 您理解并同意，禁止将本产品用于以下任何行为：
|
|
# | 1. 任何违反中华人民共和国法律法规的用途；
|
|
# | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
|
# | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
|
# | 三、责任豁免
|
|
# | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
|
# | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
|
# | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
|
# | 四、违约与法律责任
|
|
# | 如发现任何违反本声明的行为，本团队有权：
|
|
# | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
|
# | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
|
# | 3. 保留进一步采取刑事控告等法律措施的权利。
|
|
# | 五、许可范围说明
|
|
# | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
|
# | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
|
# | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
|
# | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
|
# | 成的一切经济损失及商誉损害。
|
|
# | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
|
# | 许可协议》附件为准。
|
|
# +======================================================================
|
|
|
|
"""
|
|
定时任务管理视图模块
|
|
=============================================================
|
|
|
|
该模块定义了定时任务管理功能中使用的所有视图类。
|
|
用于处理定时任务信息的HTTP请求，包括分页查询、详情查询、添加、更新、删除等操作。
|
|
"""
|
|
|
|
import json
|
|
import logging
|
|
|
|
from django.conf import settings
|
|
from django.utils.decorators import method_decorator
|
|
from django.views import View
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
|
|
from application.job import services
|
|
from application.operation_log.constants import LogType
|
|
from application.operation_log.decorators import operation_log
|
|
from config.env import DJANGO_DEMO
|
|
from middleware.login_middleware import check_login
|
|
from middleware.permission_middleware import PermissionRequired
|
|
from utils import R
|
|
|
|
|
|
# =============================================================
|
|
# 分页查询视图
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="get")
|
|
class JobPageView(PermissionRequired, View):
|
|
"""
|
|
定时任务分页数据查询视图
|
|
|
|
URL: /job/page
|
|
请求方法: GET
|
|
权限要求：sys:job:page
|
|
"""
|
|
|
|
permission_required = ("sys:job:page",)
|
|
|
|
@operation_log(
|
|
title="定时任务-查询分页列表",
|
|
log_type=LogType.QUERY
|
|
)
|
|
def get(self, request):
|
|
result = services.get_job_page(request)
|
|
return result
|
|
|
|
|
|
# =============================================================
|
|
# 详情查询视图
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="get")
|
|
class JobDetailView(PermissionRequired, View):
|
|
"""
|
|
定时任务详情查询视图
|
|
|
|
URL: /job/detail/<int:job_id>
|
|
请求方法: GET
|
|
权限要求：sys:job:detail
|
|
"""
|
|
|
|
permission_required = ("sys:job:detail",)
|
|
|
|
@operation_log(
|
|
title="定时任务-查询记录详情",
|
|
log_type=LogType.QUERY
|
|
)
|
|
def get(self, request, job_id):
|
|
data = services.get_job_detail(job_id)
|
|
if data:
|
|
return R.ok(data=data)
|
|
return R.failed("任务不存在")
|
|
|
|
|
|
# =============================================================
|
|
# 添加视图
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="post")
|
|
class JobAddView(PermissionRequired, View):
|
|
"""
|
|
添加定时任务视图
|
|
|
|
URL: /job/add
|
|
请求方法: POST
|
|
权限要求：sys:job:add
|
|
"""
|
|
|
|
permission_required = ("sys:job:add",)
|
|
|
|
@operation_log(
|
|
title="定时任务-添加记录",
|
|
log_type=LogType.ADD
|
|
)
|
|
def post(self, request):
|
|
if DJANGO_DEMO:
|
|
return R.failed("演示环境，暂无操作权限")
|
|
|
|
result = services.add_job(request)
|
|
return result
|
|
|
|
|
|
# =============================================================
|
|
# 更新视图
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="put")
|
|
class JobUpdateView(PermissionRequired, View):
|
|
"""
|
|
更新定时任务视图
|
|
|
|
URL: /job/update
|
|
请求方法: PUT
|
|
权限要求：sys:job:update
|
|
"""
|
|
|
|
permission_required = ('sys:job:update',)
|
|
|
|
@operation_log(
|
|
title="定时任务-更新记录",
|
|
log_type=LogType.UPDATE
|
|
)
|
|
def put(self, request):
|
|
if DJANGO_DEMO:
|
|
return R.failed("演示环境，暂无操作权限")
|
|
|
|
result = services.update_job(request)
|
|
return result
|
|
|
|
|
|
# =============================================================
|
|
# 删除视图（支持批量）
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="delete")
|
|
class JobDeleteView(PermissionRequired, View):
|
|
"""
|
|
删除定时任务视图（支持批量）
|
|
|
|
URL: /job/delete/<str:job_id>
|
|
请求方法: DELETE
|
|
权限要求：sys:job:delete
|
|
"""
|
|
|
|
permission_required = ('sys:job:delete',)
|
|
|
|
@operation_log(
|
|
title="定时任务-删除记录",
|
|
log_type=LogType.DELETE
|
|
)
|
|
def delete(self, request, job_id):
|
|
if DJANGO_DEMO:
|
|
return R.failed("演示环境，暂无操作权限")
|
|
|
|
result = services.delete_jobs(job_id)
|
|
return result
|
|
|
|
|
|
# =============================================================
|
|
# 状态设置视图
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="post")
|
|
class JobStatusView(PermissionRequired, View):
|
|
"""
|
|
设置任务状态视图
|
|
|
|
URL: /job/status
|
|
请求方法: POST
|
|
权限要求：sys:job:status
|
|
"""
|
|
|
|
permission_required = ('sys:job:status',)
|
|
|
|
@operation_log(
|
|
title="定时任务-设置状态",
|
|
log_type=LogType.STATUS
|
|
)
|
|
def post(self, request):
|
|
if DJANGO_DEMO:
|
|
return R.failed("演示环境，暂无操作权限")
|
|
|
|
result = services.status_job(request)
|
|
return result
|
|
|
|
|
|
# =============================================================
|
|
# 暂停任务视图
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="get")
|
|
class JobPauseView(PermissionRequired, View):
|
|
"""
|
|
暂停任务视图
|
|
|
|
URL: /job/pause/<int:job_id>
|
|
请求方法: GET
|
|
权限要求：sys:job:pause
|
|
"""
|
|
|
|
permission_required = ('sys:job:pause',)
|
|
|
|
@operation_log(
|
|
title="定时任务-暂停任务",
|
|
log_type=LogType.OTHER
|
|
)
|
|
def get(self, request, job_id):
|
|
if DJANGO_DEMO:
|
|
return R.failed("演示环境，暂无操作权限")
|
|
|
|
result = services.pause_job(job_id)
|
|
return result
|
|
|
|
|
|
# =============================================================
|
|
# 恢复任务视图
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="get")
|
|
class JobResumeView(PermissionRequired, View):
|
|
"""
|
|
恢复任务视图
|
|
|
|
URL: /job/resume/<int:job_id>
|
|
请求方法: GET
|
|
权限要求：sys:job:resume
|
|
"""
|
|
|
|
permission_required = ('sys:job:resume',)
|
|
|
|
@operation_log(
|
|
title="定时任务-恢复任务",
|
|
log_type=LogType.OTHER
|
|
)
|
|
def get(self, request, job_id):
|
|
if DJANGO_DEMO:
|
|
return R.failed("演示环境，暂无操作权限")
|
|
|
|
result = services.resume_job(job_id)
|
|
return result
|
|
|
|
|
|
# =============================================================
|
|
# 立即执行视图
|
|
# =============================================================
|
|
|
|
@method_decorator(check_login, name="get")
|
|
class JobRunOnceView(PermissionRequired, View):
|
|
"""
|
|
立即执行一次任务视图
|
|
|
|
URL: /job/runOnce/<int:job_id>
|
|
请求方法: GET
|
|
权限要求：sys:job:runOnce
|
|
"""
|
|
|
|
permission_required = ('sys:job:runOnce',)
|
|
|
|
@operation_log(
|
|
title="定时任务-立即执行",
|
|
log_type=LogType.OTHER
|
|
)
|
|
def get(self, request, job_id):
|
|
if DJANGO_DEMO:
|
|
return R.failed("演示环境，暂无操作权限")
|
|
|
|
result = services.run_job_once_service(job_id)
|
|
return result
|
|
|
|
|
|
# =============================================================
|
|
# 短信测试视图
|
|
# =============================================================
|
|
|
|
class JobSendSmsView(View):
|
|
"""
|
|
短信测试视图
|
|
|
|
用于测试短信发送功能。
|
|
"""
|
|
|
|
def post(self, request):
|
|
"""
|
|
发送测试短信
|
|
|
|
Args:
|
|
request: HttpRequest对象
|
|
请求参数：
|
|
- phone: 手机号（必填）
|
|
- content: 短信内容（可选，默认为"这是一条测试短信"）
|
|
- template_id: 模板ID（可选）
|
|
- template_params: 模板参数（可选）
|
|
|
|
Returns:
|
|
JsonResponse: 包含发送结果的JSON响应
|
|
"""
|
|
try:
|
|
data = json.loads(request.body.decode('utf-8'))
|
|
|
|
phone = data.get('phone')
|
|
content = data.get('content', '这是一条测试短信')
|
|
template_id = data.get('template_id')
|
|
template_params = data.get('template_params', {})
|
|
|
|
if not phone:
|
|
return R.failed(msg="手机号不能为空")
|
|
|
|
logging.info(f"[短信测试] 手机号: {phone}, 内容: {content}, 模板ID: {template_id}")
|
|
|
|
# 根据配置选择短信服务商
|
|
sms_provider = getattr(settings, 'SMS_PROVIDER', 'mock')
|
|
|
|
if sms_provider == 'aliyun':
|
|
result = self.send_aliyun_sms(phone, content, template_id, template_params)
|
|
elif sms_provider == 'tencent':
|
|
result = self.send_tencent_sms(phone, content, template_id, template_params)
|
|
else:
|
|
result = self.send_mock_sms(phone, content)
|
|
|
|
if result.get('success'):
|
|
return R.ok(data=result, msg="短信发送成功")
|
|
else:
|
|
return R.failed(msg=result.get('message', '短信发送失败'))
|
|
|
|
except json.JSONDecodeError:
|
|
return R.failed(msg="请求数据格式错误")
|
|
except Exception as e:
|
|
logging.error(f"短信发送异常: {str(e)}", exc_info=True)
|
|
return R.failed(msg=f"短信发送失败: {str(e)}")
|
|
|
|
def send_mock_sms(self, phone, content):
|
|
"""模拟短信发送（用于测试）"""
|
|
logging.info(f"[模拟短信] 发送至 {phone}: {content}")
|
|
return {
|
|
'success': True,
|
|
'provider': 'mock',
|
|
'phone': phone,
|
|
'content': content,
|
|
'message_id': f'mock_{phone}_{id}'
|
|
}
|
|
|
|
def send_aliyun_sms(self, phone, content, template_id, template_params):
|
|
"""阿里云短信发送"""
|
|
try:
|
|
from aliyunsdkcore.client import AcsClient
|
|
from aliyunsdkcore.request import CommonRequest
|
|
|
|
access_key_id = getattr(settings, 'ALIYUN_ACCESS_KEY_ID', '')
|
|
access_secret = getattr(settings, 'ALIYUN_ACCESS_SECRET', '')
|
|
sign_name = getattr(settings, 'ALIYUN_SMS_SIGN', '')
|
|
|
|
if not all([access_key_id, access_secret, sign_name, template_id]):
|
|
return {'success': False, 'message': '阿里云短信配置不完整'}
|
|
|
|
client = AcsClient(access_key_id, access_secret, 'cn-hangzhou')
|
|
request = CommonRequest()
|
|
request.set_accept_format('json')
|
|
request.set_domain('dysmsapi.aliyuncs.com')
|
|
request.set_method('POST')
|
|
request.set_protocol_type('https')
|
|
request.set_version('2017-05-25')
|
|
request.set_action_name('SendSms')
|
|
|
|
request.add_query_param('PhoneNumbers', phone)
|
|
request.add_query_param('SignName', sign_name)
|
|
request.add_query_param('TemplateCode', template_id)
|
|
|
|
if template_params:
|
|
import json
|
|
request.add_query_param('TemplateParam', json.dumps(template_params))
|
|
|
|
response = client.do_action_with_exception(request)
|
|
result = json.loads(response)
|
|
|
|
if result.get('Code') == 'OK':
|
|
return {
|
|
'success': True,
|
|
'provider': 'aliyun',
|
|
'phone': phone,
|
|
'request_id': result.get('RequestId'),
|
|
'biz_id': result.get('BizId')
|
|
}
|
|
else:
|
|
return {
|
|
'success': False,
|
|
'provider': 'aliyun',
|
|
'message': result.get('Message', '发送失败')
|
|
}
|
|
|
|
except Exception as e:
|
|
logging.error(f"阿里云短信发送异常: {str(e)}")
|
|
return {'success': False, 'message': str(e)}
|
|
|
|
def send_tencent_sms(self, phone, content, template_id, template_params):
|
|
"""腾讯云短信发送"""
|
|
try:
|
|
from tencentcloud.common import credential
|
|
from tencentcloud.sms.v20210111 import sms_client, models
|
|
|
|
secret_id = getattr(settings, 'TENCENT_SECRET_ID', '')
|
|
secret_key = getattr(settings, 'TENCENT_SECRET_KEY', '')
|
|
sdk_app_id = getattr(settings, 'TENCENT_SMS_APP_ID', '')
|
|
sign_name = getattr(settings, 'TENCENT_SMS_SIGN', '')
|
|
|
|
if not all([secret_id, secret_key, sdk_app_id, sign_name, template_id]):
|
|
return {'success': False, 'message': '腾讯云短信配置不完整'}
|
|
|
|
cred = credential.Credential(secret_id, secret_key)
|
|
client = sms_client.SmsClient(cred, "ap-guangzhou")
|
|
|
|
req = models.SendSmsRequest()
|
|
req.SmsSdkAppId = sdk_app_id
|
|
req.SignName = sign_name
|
|
req.TemplateId = template_id
|
|
req.PhoneNumberSet = [phone]
|
|
|
|
if template_params:
|
|
req.TemplateParamSet = list(template_params.values())
|
|
|
|
resp = client.SendSms(req)
|
|
|
|
if resp.SendStatusSet and resp.SendStatusSet[0].Code == 'Ok':
|
|
return {
|
|
'success': True,
|
|
'provider': 'tencent',
|
|
'phone': phone,
|
|
'serial_no': resp.SendStatusSet[0].SerialNo
|
|
}
|
|
else:
|
|
return {
|
|
'success': False,
|
|
'provider': 'tencent',
|
|
'message': resp.SendStatusSet[0].Message if resp.SendStatusSet else '发送失败'
|
|
}
|
|
|
|
except Exception as e:
|
|
logging.error(f"腾讯云短信发送异常: {str(e)}")
|
|
return {'success': False, 'message': str(e)}
|