def count_vulns(self, workspace_name):
"""
---
get:
tags: ["Hosts"]
summary: Counts Vulnerabilities per host
responses:
200:
description: Ok
content:
application/json:
schema: HostCountSchema
"""
host_ids = flask.request.args.get('hosts', None)
if host_ids:
host_id_list = host_ids.split(',')
else:
host_id_list = None
res_dict = {'hosts': {}}
host_count_schema = HostCountSchema()
host_count = Host.query_with_count(None, host_id_list, workspace_name)
for host in host_count.all():
res_dict["hosts"][host.id] = host_count_schema.dump(host)
# return counts.data
return res_dict
def count_vulns(self, workspace_name):
host_ids = flask.request.args.get('hosts', None)
if host_ids:
host_id_list = host_ids.split(',')
else:
host_id_list = None
res_dict = {'hosts': {}}
host_count_schema = HostCountSchema()
host_count = Host.query_with_count(None, host_id_list, workspace_name)
for host in host_count.all():
res_dict["hosts"][host.id] = host_count_schema.dump(host).data
# return counts.data
return res_dict
def _get_base_query(self, workspace_name):
return Host.query_with_count(None, None, workspace_name)
def create_host(self, host_name="pepito", os="linux"):
host = Host(host_name, os)
self.model_controller.addHostSYNC(host)
return host