Small modifications: bug in ZT report resource and unused imports removed

This commit is contained in:
VakarisZ 2020-09-08 14:08:36 +03:00
parent 4eebb1a088
commit d9ba4dd3a4
2 changed files with 2 additions and 5 deletions

View File

@ -5,6 +5,7 @@ from flask import jsonify
from monkey_island.cc.resources.auth.auth import jwt_required from monkey_island.cc.resources.auth.auth import jwt_required
from monkey_island.cc.services.reporting.report import ReportService from monkey_island.cc.services.reporting.report import ReportService
from monkey_island.cc.services.zero_trust.monkey_finding_service import MonkeyFindingService
from monkey_island.cc.services.zero_trust.zero_trust_service import \ from monkey_island.cc.services.zero_trust.zero_trust_service import \
ZeroTrustService ZeroTrustService
@ -35,6 +36,6 @@ class Report(flask_restful.Resource):
elif report_data == REPORT_DATA_PRINCIPLES_STATUS: elif report_data == REPORT_DATA_PRINCIPLES_STATUS:
return jsonify(ZeroTrustService.get_principles_status()) return jsonify(ZeroTrustService.get_principles_status())
elif report_data == REPORT_DATA_FINDINGS: elif report_data == REPORT_DATA_FINDINGS:
return jsonify(ZeroTrustService.get_all_monkey_findings()) return jsonify(MonkeyFindingService.get_all_monkey_findings())
flask_restful.abort(http.client.NOT_FOUND) flask_restful.abort(http.client.NOT_FOUND)

View File

@ -1,7 +1,3 @@
from typing import List
from bson.objectid import ObjectId
import common.common_consts.zero_trust_consts as zero_trust_consts import common.common_consts.zero_trust_consts as zero_trust_consts
from monkey_island.cc.models.zero_trust.finding import Finding from monkey_island.cc.models.zero_trust.finding import Finding