def get(self, id=None):
queryData = request.args.to_dict()
if id:
task = TaskModel.findById(id)
if task: return task.json(), 201
else: return {'message': 'task not found'}, 404
tasks = TaskModel.find(**queryData)
return {'tasks': list(map(lambda x: x.json(), tasks))}, 200
def JobManagerProcess(element_id, status):
task = TaskModel.findById(element_id)
notification_producer.publish({"args":[],
"kwargs":{"event_id":element_id,
"event_type": "task",
"column": "status",
"value": status},
"task":"notification",
"id":str(uuid4().hex)})
job_manager = JobManager(task, status)
job_manager.process_status(status)
def RuleTask(self, task_id, event_type, column, value, rule):
result = 0
if rule["element"] == "task":
rule_value = rule["value"]
if rule.get("device") and rule.get("device") != 'None':
rule_element = "device"
if rule.get("deviceClass") and rule.get("deviceClass") != 'None':
rule_element = "deviceClass"
if rule.get("group") and rule.get("group") != 'None':
rule_element = "group "
if rule.get("location") and rule.get("location") != 'None':
rule_element = "location"
get_task = TaskModel.findById(task_id)
task = get_task.__dict__
get_device = DeviceModel.find_by_id(get_task.device_id)
device = get_device.__dict__
if device[rule_element] == rule.get(
rule_element) and rule_value == value:
result = 1
return result, device[rule_element], "status", rule_value
if rule["element"] == "device":
rule_column = rule["column"]
rule_operation = rule["operation"]
rule_value = rule["value"]
get_task = TaskModel.findById(task_id)
task = get_task.__dict__
get_device = DeviceModel.find_by_id(get_task.device_id)
device_name = get_device.name
device = get_device.__dict__
try:
element_column = device[rule_column]
except:
return result, None, None, None
result = self.comparison(element_column, rule_value,
rule_operation)
return result, device_name, rule_column, rule_value
return result, None, None, None
def ProcessCompliance(self, status):
compliance_execution = ComplianceExecutionModel.find_by_id(self.job.compliance_execution_id)
check_list = CheckModel.find(**{"compliance_id": compliance_execution.compliance_id})
device_name = DeviceModel.find_by_id(TaskModel.findById(self.task.id).device_id).name
if status == "SUCCESSFUL":
total = 0
for check in check_list:
check_result = CheckResultModel(check.id, check.name, device_name, self.task.id, self.job.compliance_execution_id, status = 'NEW', parameters = {})
check_result.save_to_db(commit=False)
total, result_lines = self.ApplyChecks(self.task, check)
if total == len(check.rules):
check_result.status = "SUCCESSFUL"
check_result.parameters["result"] = result_lines
else:
check_result.status = "FAILED"
check_result.parameters["result"] = result_lines
check_result.commit()
elif status == "FAILED":
for check in check_list:
check_result = CheckResultModel(check.id, check.name, device_name, self.task.id, self.job.compliance_execution_id, status = 'NEW', parameters = {})
check_result.save_to_db(commit=False)
check_result.status = "FAILED"
check_result.commit()
return
