island: Create RansomwareReportService and add `get_exploitation_stats()` to it

This commit is contained in:
Shreya 2021-07-08 15:44:43 +05:30
parent 0db85ae407
commit 79d042b471
2 changed files with 15 additions and 1 deletions

View File

@ -2,9 +2,10 @@ import flask_restful
from flask import jsonify 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.ransomware_report import RansomwareReportService
class RansomwareReport(flask_restful.Resource): class RansomwareReport(flask_restful.Resource):
@jwt_required @jwt_required
def get(self): def get(self):
return jsonify({"report": None}) return jsonify({"report": None, "stats": RansomwareReportService.get_exploitation_stats()})

View File

@ -0,0 +1,13 @@
from monkey_island.cc.services.reporting.report import ReportService
class RansomwareReportService:
def __init__(self):
pass
@staticmethod
def get_exploitation_stats():
scanned = ReportService.get_scanned()
exploited = ReportService.get_exploited()
return {"scanned": scanned, "exploited": exploited}