def destroy(self, request, pk=None, **kwargs):
try:
vid = int(request.query_params.get('id', None))
context = Vulnerability.destory(vid=vid)
except Exception as E:
logger.error(E)
context = data_return(500, CODE_MSG.get(500), {})
return Response(context)
def list(self, request, **kwargs):
try:
ipaddress = request.query_params.get('ipaddress', None)
context = Vulnerability.list(ipaddress=ipaddress)
except Exception as E:
logger.error(E)
context = data_return(500, CODE_MSG.get(500), [])
return Response(context)
def list(self, request, **kwargs):
try:
hid = int(request.query_params.get('hid', -1))
context = Vulnerability.list(hid=hid)
except Exception as E:
logger.error(E)
context = data_return(500, CODE_MSG.get(500), [])
return Response(context)
def add_vulnerability(self,
hid_or_ipaddress=None,
extra_data=None,
desc=''):
if extra_data is None:
extra_data = {}
if isinstance(extra_data, dict) is not True:
logger.warning('数据类型检查错误,数据 {}'.format(extra_data))
extra_data = {}
if isinstance(hid_or_ipaddress, int):
result = Vulnerability.add_or_update(hid_or_ipaddress,
self.loadpath, extra_data,
desc)
return result
elif isinstance(hid_or_ipaddress, str):
result = Vulnerability.add_or_update(
Host.get_by_ipaddress(hid_or_ipaddress).get('id'),
self.loadpath, extra_data, desc)
return result
def add_vulnerability(self, ipaddress=None, extra_data=None, desc=''):
"""增加一个漏洞信息"""
if extra_data is None:
extra_data = {}
if isinstance(extra_data, dict) is not True:
logger.warning('数据类型检查错误,数据 {}'.format(extra_data))
extra_data = {}
result = Vulnerability.add_or_update(ipaddress, self.loadpath,
extra_data, desc)
return result