From 90e655d242c2ad468e181ea4bdd49a55f36a76c1 Mon Sep 17 00:00:00 2001 From: VakarisZ Date: Wed, 8 Apr 2020 16:06:02 +0300 Subject: [PATCH] Improved stix2 core import statement --- .../cc/services/attack/mitre_api_interface.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/monkey/monkey_island/cc/services/attack/mitre_api_interface.py b/monkey/monkey_island/cc/services/attack/mitre_api_interface.py index af169ec5f..cbe0d911d 100644 --- a/monkey/monkey_island/cc/services/attack/mitre_api_interface.py +++ b/monkey/monkey_island/cc/services/attack/mitre_api_interface.py @@ -1,7 +1,6 @@ from typing import List, Dict -from stix2 import FileSystemSource, Filter, CourseOfAction, AttackPattern -from stix2.core import STIXDomainObject +from stix2 import FileSystemSource, Filter, CourseOfAction, AttackPattern, core class MitreApiInterface: @@ -33,14 +32,14 @@ class MitreApiInterface: return all_techniques @staticmethod - def get_stix2_external_reference_id(stix2_data: STIXDomainObject) -> str: + def get_stix2_external_reference_id(stix2_data: core.STIXDomainObject) -> str: for reference in stix2_data['external_references']: if reference['source_name'] == "mitre-attack" and 'external_id' in reference: return reference['external_id'] return '' @staticmethod - def get_stix2_external_reference_url(stix2_data: STIXDomainObject) -> str: + def get_stix2_external_reference_url(stix2_data: core.STIXDomainObject) -> str: for reference in stix2_data['external_references']: if 'url' in reference: return reference['url']