示例#1
0
  def get(self, name):
    session_id = get_session_id(session, request)

    dao = ThreatDAO(session_id)
    threat = dao.get_threat_by_name(name=name)
    dao.close()

    resp = make_response(json_serialize(threat, session_id=session_id), OK)
    resp.headers['Content-type'] = 'application/json'
    return resp
示例#2
0
    def get(self, name):
        session_id = get_session_id(session, request)

        dao = ThreatDAO(session_id)
        threat = dao.get_threat_by_name(name=name)
        dao.close()

        resp = make_response(json_serialize(threat, session_id=session_id), OK)
        resp.headers['Content-type'] = 'application/json'
        return resp
示例#3
0
    def get_threatened_assets(self, threat_name, environment_name):
        """
    :type threat_name: str
    :type environment_name: str
    :rtype: list[Asset]
    """
        threat_dao = ThreatDAO(self.session_id)

        try:
            found_threat = threat_dao.get_threat_by_name(threat_name)
            threat_id = found_threat.theId
        except ObjectNotFoundHTTPError as ex:
            self.close()
            raise ex
        except ARMHTTPError as ex:
            self.close()
            raise ex

        environment_dao = EnvironmentDAO(self.session_id)
        try:
            found_environment = environment_dao.get_environment_by_name(
                environment_name)
            environment_id = found_environment.theId
        except ObjectNotFoundHTTPError as ex:
            self.close()
            raise ex
        except ARMHTTPError as ex:
            self.close()
            raise ex

        try:
            threatened_assets = self.db_proxy.threatenedAssets(
                threat_id, environment_id)
        except DatabaseProxyException as ex:
            self.close()
            raise ARMHTTPError(ex)
        except ARMException as ex:
            self.close()
            raise ARMHTTPError(ex)

        return threatened_assets
示例#4
0
    def get_threatened_assets(self, threat_name, environment_name):
        """
        :type threat_name: str
        :type environment_name: str
        :rtype: list[Asset]
        """
        threat_dao = ThreatDAO(self.session_id)

        try:
            found_threat = threat_dao.get_threat_by_name(threat_name)
            threat_id = found_threat.theId
        except ObjectNotFoundHTTPError as ex:
            self.close()
            raise ex
        except ARMHTTPError as ex:
            self.close()
            raise ex

        environment_dao = EnvironmentDAO(self.session_id)
        try:
            found_environment = environment_dao.get_environment_by_name(environment_name)
            environment_id = found_environment.theId
        except ObjectNotFoundHTTPError as ex:
            self.close()
            raise ex
        except ARMHTTPError as ex:
            self.close()
            raise ex

        try:
            threatened_assets = self.db_proxy.threatenedAssets(threat_id, environment_id)
        except DatabaseProxyException as ex:
            self.close()
            raise ARMHTTPError(ex)
        except ARMException as ex:
            self.close()
            raise ARMHTTPError(ex)

        return threatened_assets