Remove patterns with outdegree 0

WithSecureLabs · May 26, 2021 · a43e511 · a43e511
1 parent 00ce4ce
commit a43e511
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/aws/attacks.py b/lib/aws/attacks.py
@@ -1382,7 +1382,10 @@ def compute(self, max_iterations=5):
                     pruned += db.run(
                         "MATCH (:Pattern)-[attack:ATTACK|CREATE]->(:Generic) "
                         "DELETE attack "
-                        "RETURN COUNT(attack) AS pruned"
+                        "WITH COUNT(attack) AS pruned "
+                        "OPTIONAL MATCH (p:Pattern) WHERE NOT EXISTS((p)-[:ATTACK|CREATE]->()) "
+                        "DETACH DELETE p "
+                        "RETURN pruned"
                     )[0]["pruned"]
 
         db.close()