zkyuan/src/dashboard/views_backstage.py

# -*-coding:utf-8-*-


import common.common_functions as ccf
import common.error_info as ce
from common import core_views as cv
from dashboard import control_dashboard as ctd


class WebsiteStatisticsView(cv.AuthView):
    def get(self, request):
        """
        #态势网站监测基本统计
        参数：无
        """
        user = request.user
        uid = user.id
        try:
            ret = ctd.get_nssa_website_statistics(uid)
            return cv.to_suc(ret)
        except Exception as e:
            return cv.to_fail(e)


class WebsiteAvailableAlertView(cv.AuthView):
    def get(self, request):
        """
        #态势网站可用性告警
        参数：无
        """
        user = request.user
        uid = user.id
        try:
            ret = ctd.get_nssa_website_available(uid)
            return cv.to_suc(ret)
        except Exception as e:
            return cv.to_fail(e)


class WebsiteRiskView(cv.AuthView):
    def get(self, request):
        """
        #态势风险网站TOP10
        参数：无
        """
        user = request.user
        uid = user.id
        try:
            ret = ctd.get_nssa_website_risk(uid)
            return cv.to_suc(ret)
        except Exception as e:
            return cv.to_fail(e)


class WebsiteRiskLatestView(cv.AuthView):
    def get(self, request):
        """
        #态势最新威胁信息
        参数：无
        """
        user = request.user
        uid = user.id
        try:
            ret = ctd.get_nssa_website_risk(uid)
            return cv.to_suc(ret)
        except Exception as e:
            return cv.to_fail(e)


class WebsiteBugView(cv.AuthView):
    def get(self, request):
        """
        #态势网站漏洞信息
        参数：无
        """
        user = request.user
        uid = user.id
        try:
            ret = ctd.get_nssa_website_bug(uid)
            return cv.to_suc(ret)
        except Exception as e:
            return cv.to_fail(e)


class WebsiteTendencyView(cv.AuthView):
    def get(self, request):
        """
        #态势风险趋势
        参数：无
        """
        user = request.user
        uid = user.id
        try:
            ret = ctd.get_nssa_website_tendency(uid)
            return cv.to_suc(ret)
        except Exception as e:
            return cv.to_fail(e)