all code examples work

mitre-attack · Sep 6, 2023 · 3227bbf · 3227bbf
1 parent d891711
commit 3227bbf
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/examples/get_campaign_by_alias.py b/examples/get_campaign_by_alias.py
@@ -4,9 +4,9 @@
 def main():
     mitre_attack_data = MitreAttackData("enterprise-attack.json")
 
-    C0001 = mitre_attack_data.get_campaign_by_alias("Frankenstein")
+    C0001 = mitre_attack_data.get_campaigns_by_alias("Frankenstein")
 
-    print(C0001.serialize(pretty=True))
+    print(C0001[0].serialize(pretty=True))
 
 
 if __name__ == "__main__":

diff --git a/examples/get_group_by_alias.py b/examples/get_group_by_alias.py
@@ -4,9 +4,9 @@
 def main():
     mitre_attack_data = MitreAttackData("enterprise-attack.json")
 
-    G0016 = mitre_attack_data.get_group_by_alias("Cozy Bear")
+    G0016 = mitre_attack_data.get_groups_by_alias("Cozy Bear")
 
-    print(G0016.serialize(pretty=True))
+    print(G0016[0].serialize(pretty=True))
 
 
 if __name__ == "__main__":

diff --git a/examples/get_object_by_name.py b/examples/get_object_by_name.py
@@ -4,9 +4,9 @@
 def main():
     mitre_attack_data = MitreAttackData("enterprise-attack.json")
 
-    T1082 = mitre_attack_data.get_object_by_name("System Information Discovery", "attack-pattern")
+    T1082 = mitre_attack_data.get_objects_by_name("System Information Discovery", "attack-pattern")
 
-    print(T1082.serialize(pretty=True))
+    print(T1082[0].serialize(pretty=True))
 
 
 if __name__ == "__main__":

diff --git a/examples/get_software_by_alias.py b/examples/get_software_by_alias.py
@@ -6,7 +6,7 @@ def main():
 
     S0196 = mitre_attack_data.get_software_by_alias("ShellTea")
 
-    print(S0196.serialize(pretty=True))
+    print(S0196[0].serialize(pretty=True))
 
 
 if __name__ == "__main__":