331 lines
14 KiB
Python
331 lines
14 KiB
Python
import os
|
|
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
from django.db.models import Q
|
|
from rest_framework.request import Request
|
|
|
|
from .models import LoginInfor, OperationLog, CeleryLog
|
|
from ..op_drf.filters import DataLevelPermissionsFilter
|
|
from ..op_drf.viewsets import CustomModelViewSet
|
|
from ..permission.permissions import CommonPermission
|
|
from ..system.filters import DictDetailsFilter, DictDataFilter, ConfigSettingsFilter, MessagePushFilter, \
|
|
SaveFileFilter, LoginInforFilter, OperationLogFilter, CeleryLogFilter
|
|
from ..system.models import DictData, DictDetails, ConfigSettings, SaveFile, MessagePush
|
|
from ..system.models import MessagePushUser
|
|
from ..system.serializers import DictDataSerializer, DictDataCreateUpdateSerializer, DictDetailsSerializer, \
|
|
DictDetailsCreateUpdateSerializer, ConfigSettingsSerializer, \
|
|
ConfigSettingsCreateUpdateSerializer, SaveFileSerializer, SaveFileCreateUpdateSerializer, \
|
|
ExportConfigSettingsSerializer, ExportDictDataSerializer, ExportDictDetailsSerializer, \
|
|
MessagePushSerializer, MessagePushCreateUpdateSerializer, ExportMessagePushSerializer, LoginInforSerializer, \
|
|
OperationLogSerializer, ExportOperationLogSerializer, ExportLoginInforSerializer, CeleryLogSerializer, \
|
|
ExportCeleryLogSerializer
|
|
from ..utils.export_excel import export_excel_save_model
|
|
from ..utils.file_util import get_all_files, remove_empty_dir, delete_files
|
|
from ..utils.response import SuccessResponse
|
|
|
|
|
|
class DictDataModelViewSet(CustomModelViewSet):
|
|
"""
|
|
字典管理模型的CRUD视图
|
|
"""
|
|
queryset = DictData.objects.all()
|
|
serializer_class = DictDataSerializer
|
|
create_serializer_class = DictDataCreateUpdateSerializer
|
|
update_serializer_class = DictDataCreateUpdateSerializer
|
|
# list_serializer_class = ListRoleSerializer
|
|
# retrieve_serializer_class = DetailRoleSerializer
|
|
extra_filter_backends = [DataLevelPermissionsFilter]
|
|
filter_class = DictDataFilter
|
|
update_extra_permission_classes = (CommonPermission,)
|
|
destroy_extra_permission_classes = (CommonPermission,)
|
|
create_extra_permission_classes = (CommonPermission,)
|
|
search_fields = ('dictName',)
|
|
ordering = 'id' # 默认排序
|
|
export_field_data = ['字典主键', '字典名称', '字典类型', '字典状态', '创建者', '修改者', '备注']
|
|
export_serializer_class = ExportDictDataSerializer
|
|
|
|
|
|
class DictDetailsModelViewSet(CustomModelViewSet):
|
|
"""
|
|
字典详情 模型的CRUD视图
|
|
"""
|
|
queryset = DictDetails.objects.all()
|
|
serializer_class = DictDetailsSerializer
|
|
create_serializer_class = DictDetailsCreateUpdateSerializer
|
|
update_serializer_class = DictDetailsCreateUpdateSerializer
|
|
filter_class = DictDetailsFilter
|
|
extra_filter_backends = [DataLevelPermissionsFilter]
|
|
update_extra_permission_classes = (CommonPermission,)
|
|
destroy_extra_permission_classes = (CommonPermission,)
|
|
create_extra_permission_classes = (CommonPermission,)
|
|
search_fields = ('dictLabel',)
|
|
ordering = 'sort' # 默认排序
|
|
|
|
def dict_details_list(self, request: Request, *args, **kwargs):
|
|
"""
|
|
根据字典类型查询字典数据信息
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
dict_details_dic = cache.get('system_dict_details', {})
|
|
if not dict_details_dic:
|
|
queryset = self.filter_queryset(self.get_queryset())
|
|
queryset_dic = queryset.order_by('sort').values('dict_data__dictType', 'dictLabel', 'dictValue',
|
|
'is_default')
|
|
for ele in queryset_dic:
|
|
dictType = ele.pop('dict_data__dictType')
|
|
if dictType in dict_details_dic:
|
|
dict_details_dic[dictType].append(ele)
|
|
else:
|
|
dict_details_dic[dictType] = [ele]
|
|
cache.set('system_dict_details', dict_details_dic, 84600)
|
|
return SuccessResponse(dict_details_dic.get(kwargs.get('pk'), []))
|
|
|
|
def clearCache(self, request: Request, *args, **kwargs):
|
|
"""
|
|
清理键值缓存
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
cache.delete('system_dict_details')
|
|
return SuccessResponse(msg='清理成功！')
|
|
|
|
def export(self, request: Request, *args, **kwargs):
|
|
"""
|
|
导出字典详情数据
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
dictType = request.query_params.get('dictType')
|
|
field_data = ['字典详情主键', '字典标签', '字典键值', '是否默认', '字典状态', '字典排序', '创建者', '修改者', '备注']
|
|
data = ExportDictDetailsSerializer(DictDetails.objects.filter(dict_data__dictType=dictType), many=True).data
|
|
return SuccessResponse(export_excel_save_model(request, field_data, data, f'导出字典[{dictType}]详情数据.xls'))
|
|
|
|
|
|
class ConfigSettingsModelViewSet(CustomModelViewSet):
|
|
"""
|
|
参数设置 模型的CRUD视图
|
|
"""
|
|
queryset = ConfigSettings.objects.all()
|
|
serializer_class = ConfigSettingsSerializer
|
|
create_serializer_class = ConfigSettingsCreateUpdateSerializer
|
|
update_serializer_class = ConfigSettingsCreateUpdateSerializer
|
|
filter_class = ConfigSettingsFilter
|
|
search_fields = ('configName',)
|
|
ordering = 'id' # 默认排序
|
|
extra_filter_backends = [DataLevelPermissionsFilter]
|
|
update_extra_permission_classes = (CommonPermission,)
|
|
destroy_extra_permission_classes = (CommonPermission,)
|
|
create_extra_permission_classes = (CommonPermission,)
|
|
export_field_data = ['参数主键', '参数名称', '参数键名', '参数键值', '系统内置', '参数状态', '创建者', '修改者', '备注']
|
|
export_serializer_class = ExportConfigSettingsSerializer
|
|
|
|
def get_config_key(self, request: Request, *args, **kwargs):
|
|
"""
|
|
根据 参数键名 查询参数数据信息
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
config_key_dic = cache.get('system_configKey')
|
|
if not config_key_dic:
|
|
queryset = self.filter_queryset(self.get_queryset())
|
|
config_key_dic = {ele.get('configKey'): ele.get('configValue') for ele in
|
|
queryset.values('configValue', 'configKey')}
|
|
cache.set('system_configKey', config_key_dic, 84600)
|
|
return SuccessResponse(msg=config_key_dic.get(kwargs.get('pk'), ''))
|
|
|
|
def clearCache(self, request: Request, *args, **kwargs):
|
|
"""
|
|
清理键值缓存
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
cache.delete('system_configKey')
|
|
return SuccessResponse(msg='清理成功！')
|
|
|
|
|
|
class SaveFileModelViewSet(CustomModelViewSet):
|
|
"""
|
|
文件管理 模型的CRUD视图
|
|
"""
|
|
queryset = SaveFile.objects.all()
|
|
serializer_class = SaveFileSerializer
|
|
create_serializer_class = SaveFileCreateUpdateSerializer
|
|
update_serializer_class = SaveFileCreateUpdateSerializer
|
|
filter_class = SaveFileFilter
|
|
extra_filter_backends = [DataLevelPermissionsFilter]
|
|
update_extra_permission_classes = (CommonPermission,)
|
|
destroy_extra_permission_classes = (CommonPermission,)
|
|
create_extra_permission_classes = (CommonPermission,)
|
|
search_fields = ('configName',)
|
|
ordering = '-create_datetime' # 默认排序
|
|
|
|
def clearsavefile(self, request: Request, *args, **kwargs):
|
|
"""
|
|
清理废弃文件
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
# 获取废弃文件列表
|
|
file_list = get_all_files(os.path.join(settings.MEDIA_ROOT, 'system'))
|
|
queryset_files = [os.path.join(os.path.join(settings.MEDIA_ROOT) + os.sep, ele) for ele in
|
|
list(self.get_queryset().values_list('file', flat=True))]
|
|
|
|
delete_list = list(set(file_list) - set(queryset_files))
|
|
# 进行文件删除操作
|
|
delete_files(delete_list)
|
|
# 递归删除空文件
|
|
remove_empty_dir(os.path.join(settings.MEDIA_ROOT, 'system'))
|
|
return SuccessResponse(msg=f"成功清理废弃文件{len(delete_list)}个")
|
|
|
|
|
|
class MessagePushModelViewSet(CustomModelViewSet):
|
|
"""
|
|
消息推送模型的CRUD视图
|
|
"""
|
|
queryset = MessagePush.objects.all()
|
|
serializer_class = MessagePushSerializer
|
|
create_serializer_class = MessagePushCreateUpdateSerializer
|
|
update_serializer_class = MessagePushCreateUpdateSerializer
|
|
# extra_filter_backends = [DataLevelPermissionsFilter]
|
|
update_extra_permission_classes = (CommonPermission,)
|
|
destroy_extra_permission_classes = (CommonPermission,)
|
|
create_extra_permission_classes = (CommonPermission,)
|
|
filter_class = MessagePushFilter
|
|
ordering = "-update_datetime" # 默认排序
|
|
export_field_data = ['消息序号', '标题', '内容', '消息类型', '是否审核', '消息状态', '通知接收消息用户',
|
|
'创建者', '修改者', '修改时间', '创建时间']
|
|
export_serializer_class = ExportMessagePushSerializer
|
|
|
|
def get_user_messages(self, request: Request, *args, **kwargs):
|
|
"""
|
|
获取用户自己消息列表
|
|
"""
|
|
queryset = self.filter_queryset(self.get_queryset())
|
|
is_read = request.query_params.get('is_read', None)
|
|
if is_read:
|
|
if is_read == 'False':
|
|
queryset = queryset.exclude(Q(messagepushuser_message_push__is_read=True),
|
|
Q(messagepushuser_message_push__user=request.user))
|
|
elif is_read == 'True':
|
|
queryset = queryset.filter(messagepushuser_message_push__is_read=True,
|
|
messagepushuser_message_push__user=request.user)
|
|
queryset = queryset.filter(is_reviewed=True).distinct()
|
|
page = self.paginate_queryset(queryset)
|
|
if hasattr(self, 'handle_logging'):
|
|
self.handle_logging(request, *args, **kwargs)
|
|
if page is not None:
|
|
if getattr(self, 'values_queryset', None):
|
|
return self.get_paginated_response(page)
|
|
serializer = self.get_serializer(page, many=True)
|
|
return self.get_paginated_response(serializer.data)
|
|
if getattr(self, 'values_queryset', None):
|
|
return SuccessResponse(page)
|
|
serializer = self.get_serializer(queryset, many=True)
|
|
return SuccessResponse(serializer.data)
|
|
|
|
def update_is_read(self, request: Request, *args, **kwargs):
|
|
"""
|
|
修改为已读
|
|
"""
|
|
instance, _ = MessagePushUser.objects.get_or_create(message_push_id=kwargs.get('pk'), user=request.user)
|
|
instance.is_read = True
|
|
instance.save()
|
|
return SuccessResponse()
|
|
|
|
|
|
class LoginInforModelViewSet(CustomModelViewSet):
|
|
"""
|
|
登录日志 模型的CRUD视图
|
|
"""
|
|
queryset = LoginInfor.objects.all()
|
|
serializer_class = LoginInforSerializer
|
|
filter_class = LoginInforFilter
|
|
extra_filter_backends = [DataLevelPermissionsFilter]
|
|
update_extra_permission_classes = (CommonPermission,)
|
|
destroy_extra_permission_classes = (CommonPermission,)
|
|
create_extra_permission_classes = (CommonPermission,)
|
|
ordering = '-create_datetime' # 默认排序
|
|
export_field_data = ['访问编号', '用户名称', '登录地址', '登录地点', '浏览器', '操作系统',
|
|
'登录状态', '操作信息', '登录日期']
|
|
export_serializer_class = ExportLoginInforSerializer
|
|
|
|
def clean_all(self, request: Request, *args, **kwargs):
|
|
"""
|
|
清空登录日志
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
self.get_queryset().delete()
|
|
return SuccessResponse(msg="清空成功")
|
|
|
|
|
|
class OperationLogModelViewSet(CustomModelViewSet):
|
|
"""
|
|
操作日志 模型的CRUD视图
|
|
"""
|
|
queryset = OperationLog.objects.all()
|
|
serializer_class = OperationLogSerializer
|
|
filter_class = OperationLogFilter
|
|
extra_filter_backends = [DataLevelPermissionsFilter]
|
|
update_extra_permission_classes = (CommonPermission,)
|
|
destroy_extra_permission_classes = (CommonPermission,)
|
|
create_extra_permission_classes = (CommonPermission,)
|
|
ordering = '-create_datetime' # 默认排序
|
|
export_field_data = ['请求模块', '请求地址', '请求参数', '请求方式', '操作说明', '请求ip地址',
|
|
'请求浏览器', '响应状态码', '操作地点', '操作系统', '返回信息', '响应状态', '操作用户名']
|
|
export_serializer_class = ExportOperationLogSerializer
|
|
|
|
def clean_all(self, request: Request, *args, **kwargs):
|
|
"""
|
|
清空操作日志
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
self.get_queryset().delete()
|
|
return SuccessResponse(msg="清空成功")
|
|
|
|
|
|
class CeleryLogModelViewSet(CustomModelViewSet):
|
|
"""
|
|
定时任务日志 模型的CRUD视图
|
|
"""
|
|
queryset = CeleryLog.objects.all()
|
|
serializer_class = CeleryLogSerializer
|
|
extra_filter_backends = [DataLevelPermissionsFilter]
|
|
update_extra_permission_classes = (CommonPermission,)
|
|
destroy_extra_permission_classes = (CommonPermission,)
|
|
create_extra_permission_classes = (CommonPermission,)
|
|
filter_class = CeleryLogFilter
|
|
ordering = '-create_datetime' # 默认排序
|
|
export_field_data = ['任务名称', '执行参数', '执行时间', '运行状态', '任务结果', '创建时间']
|
|
export_serializer_class = ExportCeleryLogSerializer
|
|
|
|
def clean_all(self, request: Request, *args, **kwargs):
|
|
"""
|
|
清空定时任务日志
|
|
:param request:
|
|
:param args:
|
|
:param kwargs:
|
|
:return:
|
|
"""
|
|
self.get_queryset().delete()
|
|
return SuccessResponse(msg="清空成功")
|