126 lines
5.4 KiB
Python
126 lines
5.4 KiB
Python
from django_celery_beat.models import PeriodicTask, IntervalSchedule, CrontabSchedule
|
||
from rest_framework.request import Request
|
||
|
||
from .filters import ServerFilter, MonitorFilter
|
||
from .models import Server, Monitor
|
||
from .serializers import ServerSerializer, MonitorSerializer
|
||
from ..op_drf.response import SuccessResponse
|
||
from ..op_drf.viewsets import CustomModelViewSet
|
||
from ..permission.permissions import CommonPermission
|
||
from ..system.models import ConfigSettings
|
||
|
||
|
||
class ServerModelViewSet(CustomModelViewSet):
|
||
"""
|
||
服务器信息 模型的CRUD视图
|
||
"""
|
||
queryset = Server.objects.all()
|
||
serializer_class = ServerSerializer
|
||
# extra_filter_backends = [DataLevelPermissionsFilter]
|
||
filter_class = ServerFilter
|
||
update_extra_permission_classes = (CommonPermission,)
|
||
destroy_extra_permission_classes = (CommonPermission,)
|
||
create_extra_permission_classes = (CommonPermission,)
|
||
ordering = '-create_datetime' # 默认排序
|
||
|
||
|
||
class MonitorModelViewSet(CustomModelViewSet):
|
||
"""
|
||
服务器监控信息 模型的CRUD视图
|
||
"""
|
||
queryset = Monitor.objects.all()
|
||
serializer_class = MonitorSerializer
|
||
# extra_filter_backends = [DataLevelPermissionsFilter]
|
||
filter_class = MonitorFilter
|
||
update_extra_permission_classes = (CommonPermission,)
|
||
destroy_extra_permission_classes = (CommonPermission,)
|
||
create_extra_permission_classes = (CommonPermission,)
|
||
ordering = '-create_datetime' # 默认排序
|
||
|
||
def get_monitor_info(self, request: Request, *args, **kwargs):
|
||
# 定时获取系统监控信息
|
||
periodictask_obj = PeriodicTask.objects.filter(task='apps.vadmin.monitor.tasks.get_monitor_info').first()
|
||
if not periodictask_obj:
|
||
intervalschedule_obj, _ = IntervalSchedule.objects.get_or_create(every=5, period="seconds")
|
||
periodictask_obj = PeriodicTask()
|
||
periodictask_obj.task = "apps.vadmin.monitor.tasks.get_monitor_info"
|
||
periodictask_obj.name = "定时获取系统监控信息"
|
||
periodictask_obj.interval = intervalschedule_obj
|
||
periodictask_obj.enabled = False
|
||
periodictask_obj.save()
|
||
|
||
# 定时清理多余 系统监控信息
|
||
clean_task_obj = PeriodicTask.objects.filter(
|
||
task='apps.vadmin.monitor.tasks.clean_surplus_monitor_info').first()
|
||
if not clean_task_obj:
|
||
crontab_obj, _ = CrontabSchedule.objects.get_or_create(day_of_month="*", day_of_week="*", hour="1",
|
||
minute="0", month_of_year="*")
|
||
clean_task_obj = PeriodicTask()
|
||
clean_task_obj.task = "apps.vadmin.monitor.tasks.clean_surplus_monitor_info"
|
||
clean_task_obj.name = "定时清理多余-系统监控信息"
|
||
clean_task_obj.crontab = crontab_obj
|
||
clean_task_obj.enabled = True
|
||
clean_task_obj.save()
|
||
# 配置添加
|
||
config_obj = ConfigSettings.objects.filter(configKey='sys.monitor.info.save_days').first()
|
||
if not config_obj:
|
||
config_obj = ConfigSettings()
|
||
config_obj.configKey = "sys.monitor.info.save_days"
|
||
config_obj.configName = "定时清理多余系统监控信息"
|
||
config_obj.configValue = "30"
|
||
config_obj.configType = "Y"
|
||
config_obj.status = "1"
|
||
config_obj.remark = "定时清理多余-系统监控信息，默认30天"
|
||
config_obj.save()
|
||
|
||
# 获取保留天数
|
||
return SuccessResponse(data={
|
||
"enabled": periodictask_obj.enabled,
|
||
"interval": periodictask_obj.interval.every,
|
||
"save_days": config_obj.configValue if config_obj else "30",
|
||
})
|
||
|
||
def enabled_monitor_info(self, request: Request, *args, **kwargs):
|
||
enabled = request.query_params.get('enabled', None)
|
||
save_days = request.query_params.get('save_days', None)
|
||
interval = request.query_params.get('interval', None)
|
||
|
||
periodictask_obj = PeriodicTask.objects.filter(task='apps.vadmin.monitor.tasks.get_monitor_info').first()
|
||
if enabled:
|
||
# 更新监控状态
|
||
periodictask_obj.enabled = True if enabled == "1" else False
|
||
periodictask_obj.save()
|
||
|
||
# 更新 定时清理多余 系统监控信息 状态
|
||
clean_task_obj = PeriodicTask.objects.filter(
|
||
task='apps.vadmin.monitor.tasks.clean_surplus_monitor_info').first()
|
||
clean_task_obj.enabled = True if enabled == "1" else False
|
||
clean_task_obj.save()
|
||
# 更新保留天数
|
||
config_obj = ConfigSettings.objects.filter(configKey='sys.monitor.info.save_days').first()
|
||
print(33, save_days)
|
||
print(22, config_obj)
|
||
if save_days and config_obj:
|
||
config_obj.configValue = save_days
|
||
config_obj.save()
|
||
# 更新监控获取频率
|
||
if interval:
|
||
periodictask_obj.interval.every = interval
|
||
periodictask_obj.interval.save()
|
||
return SuccessResponse(data={
|
||
"enabled": periodictask_obj.enabled,
|
||
"interval": periodictask_obj.interval.every,
|
||
"save_days": config_obj.configValue if config_obj else "30",
|
||
})
|
||
|
||
def clean_all(self, request: Request, *args, **kwargs):
|
||
"""
|
||
清空监控信息
|
||
:param request:
|
||
:param args:
|
||
:param kwargs:
|
||
:return:
|
||
"""
|
||
self.get_queryset().delete()
|
||
return SuccessResponse(msg="清空成功")
|