Add filtering logic to use-defusedxml transformation

pixee · Feb 29, 2024 · b115358 · b115358
1 parent 2d1d4e3
commit b115358
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/core_codemods/use_defused_xml.py b/src/core_codemods/use_defused_xml.py
@@ -83,6 +83,9 @@ def matching_functions(self) -> dict[str, str]:
         return _matching_functions
 
     def transform_module_impl(self, tree: cst.Module) -> cst.Module:
+        if not self.node_is_selected(tree):
+            return tree
+
         visitor = DefusedXmlModifier(
             self.context,
             self.file_context,