mirror of https://github.com/openspug/spug
97 lines
4.4 KiB
Python
97 lines
4.4 KiB
Python
# Copyright: (c) OpenSpug Organization. https://github.com/openspug/spug
|
|
# Copyright: (c) <spug.dev@gmail.com>
|
|
# Released under the AGPL-3.0 License.
|
|
from django.views.generic import View
|
|
from libs import json_response, JsonParser, Argument, human_datetime
|
|
from apps.monitor.models import Detection
|
|
from django_redis import get_redis_connection
|
|
from django.conf import settings
|
|
from apps.monitor.executors import dispatch
|
|
import json
|
|
|
|
|
|
class DetectionView(View):
|
|
def get(self, request):
|
|
detections = Detection.objects.all()
|
|
groups = [x['group'] for x in detections.order_by('group').values('group').distinct()]
|
|
return json_response({'groups': groups, 'detections': [x.to_view() for x in detections]})
|
|
|
|
def post(self, request):
|
|
form, error = JsonParser(
|
|
Argument('id', type=int, required=False),
|
|
Argument('name', help='请输入任务名称'),
|
|
Argument('group', help='请选择任务分组'),
|
|
Argument('targets', type=list, filter=lambda x: len(x), help='请输入监控地址'),
|
|
Argument('type', filter=lambda x: x in dict(Detection.TYPES), help='请选择监控类型'),
|
|
Argument('extra', required=False),
|
|
Argument('desc', required=False),
|
|
Argument('rate', type=int, default=5),
|
|
Argument('threshold', type=int, default=3),
|
|
Argument('quiet', type=int, default=24 * 60),
|
|
Argument('notify_grp', type=list, help='请选择报警联系组'),
|
|
Argument('notify_mode', type=list, help='请选择报警方式'),
|
|
).parse(request.body)
|
|
if error is None:
|
|
form.targets = json.dumps(form.targets)
|
|
form.notify_grp = json.dumps(form.notify_grp)
|
|
form.notify_mode = json.dumps(form.notify_mode)
|
|
if form.id:
|
|
Detection.objects.filter(pk=form.id).update(
|
|
updated_at=human_datetime(),
|
|
updated_by=request.user,
|
|
**form)
|
|
task = Detection.objects.filter(pk=form.id).first()
|
|
if task and task.is_active:
|
|
form.action = 'modify'
|
|
rds_cli = get_redis_connection()
|
|
rds_cli.lpush(settings.MONITOR_KEY, json.dumps(form))
|
|
else:
|
|
dtt = Detection.objects.create(created_by=request.user, **form)
|
|
form.action = 'add'
|
|
form.id = dtt.id
|
|
rds_cli = get_redis_connection()
|
|
rds_cli.lpush(settings.MONITOR_KEY, json.dumps(form))
|
|
return json_response(error=error)
|
|
|
|
def patch(self, request):
|
|
form, error = JsonParser(
|
|
Argument('id', type=int, help='请指定操作对象'),
|
|
Argument('is_active', type=bool, required=False)
|
|
).parse(request.body, True)
|
|
if error is None:
|
|
Detection.objects.filter(pk=form.id).update(**form)
|
|
if form.get('is_active') is not None:
|
|
if form.is_active:
|
|
task = Detection.objects.filter(pk=form.id).first()
|
|
message = {'id': form.id, 'action': 'add'}
|
|
message.update(task.to_dict(selects=('targets', 'extra', 'rate', 'type', 'threshold', 'quiet')))
|
|
else:
|
|
message = {'id': form.id, 'action': 'remove'}
|
|
rds_cli = get_redis_connection()
|
|
rds_cli.lpush(settings.MONITOR_KEY, json.dumps(message))
|
|
return json_response(error=error)
|
|
|
|
def delete(self, request):
|
|
form, error = JsonParser(
|
|
Argument('id', type=int, help='请指定操作对象')
|
|
).parse(request.GET)
|
|
if error is None:
|
|
task = Detection.objects.filter(pk=form.id).first()
|
|
if task:
|
|
if task.is_active:
|
|
return json_response(error='该监控项正在运行中，请先停止后再尝试删除')
|
|
task.delete()
|
|
return json_response(error=error)
|
|
|
|
|
|
def run_test(request):
|
|
form, error = JsonParser(
|
|
Argument('type', help='请选择监控类型'),
|
|
Argument('targets', type=list, filter=lambda x: len(x), help='请输入监控地址'),
|
|
Argument('extra', required=False)
|
|
).parse(request.body)
|
|
if error is None:
|
|
is_success, message = dispatch(form.type, form.targets[0], form.extra)
|
|
return json_response({'is_success': is_success, 'message': message})
|
|
return json_response(error=error)
|