add include/exclude check to codemods that did not have it

src/core_codemods/django_receiver_on_top.py

@@ -23,6 +23,8 @@ def leave_FunctionDef(
     ) -> Union[
         cst.BaseStatement, cst.FlattenSentinel[cst.BaseStatement], cst.RemovalSentinel
     ]:
+        # TODO: add filter by include or exclude that works for nodes
+        # that that have different start/end numbers.
         maybe_receiver_with_index = None
         for i, decorator in enumerate(original_node.decorators):
             true_name = self.find_base_name(decorator.decorator)

src/core_codemods/exception_without_raise.py

@@ -27,6 +27,11 @@ def leave_SimpleStatementLine(
     ) -> Union[
         cst.BaseStatement, cst.FlattenSentinel[cst.BaseStatement], cst.RemovalSentinel
     ]:
+        if not self.filter_by_path_includes_or_excludes(
+            self.node_position(original_node)
+        ):
+            return original_node
+
         match original_node:
             case cst.SimpleStatementLine(
                 body=[cst.Expr(cst.Name() | cst.Attribute() as name)]

src/core_codemods/fix_deprecated_abstractproperty.py

@@ -19,6 +19,11 @@ class FixDeprecatedAbstractproperty(BaseCodemod, NameResolutionMixin):
     def leave_Decorator(
         self, original_node: cst.Decorator, updated_node: cst.Decorator
     ):
+        if not self.filter_by_path_includes_or_excludes(
+            self.node_position(original_node)
+        ):
+            return original_node
+
         if (
             base_name := self.find_base_name(original_node.decorator)
         ) and base_name == "abc.abstractproperty":

src/core_codemods/fix_mutable_params.py

@@ -154,6 +154,8 @@ def leave_FunctionDef(
         updated_node: cst.FunctionDef,
     ):
         """Transforms function definitions with mutable default parameters"""
+        # TODO: add filter by include or exclude that works for nodes
+        # that that have different start/end numbers.
         (
             updated_params,
             new_var_decls,

src/core_codemods/remove_unnecessary_f_str.py

@@ -34,6 +34,11 @@ def _check_formatted_string(
         _original_node: cst.FormattedString,
         updated_node: cst.FormattedString,
     ):
+        if not self.filter_by_path_includes_or_excludes(
+            self.node_position(_original_node)
+        ):
+            return _original_node
+
         transformed_node = super()._check_formatted_string(_original_node, updated_node)
         if not _original_node.deep_equals(transformed_node):
             self.report_change(_original_node)

src/core_codemods/use_generator.py

@@ -25,6 +25,11 @@ class UseGenerator(BaseCodemod, NameResolutionMixin):
     ]
 
     def leave_Call(self, original_node: cst.Call, updated_node: cst.Call):
+        if not self.filter_by_path_includes_or_excludes(
+            self.node_position(original_node)
+        ):
+            return original_node
+
         match original_node.func:
             # NOTE: could also support things like `list` and `tuple`
             # but it's a less compelling use case

src/core_codemods/use_walrus_if.py

@@ -118,6 +118,8 @@ def visit_If(self, node: cst.If):
         )
 
     def leave_If(self, original_node, updated_node):
+        # TODO: add filter by include or exclude that works for nodes
+        # that that have different start/end numbers.
         if (result := self._if_stack.pop()) is not None:
             position, named_expr = result
             is_name = m.matches(updated_node.test, m.Name())

tests/codemods/base_codemod_test.py

@@ -33,6 +33,32 @@ def run_and_assert(self, tmpdir, input_code, expected):
         tmp_file_path = Path(tmpdir / "code.py")
         self.run_and_assert_filepath(tmpdir, tmp_file_path, input_code, expected)
 
+    def assert_no_change_line_excluded(
+        self, tmpdir, input_code, expected, lines_to_exclude
+    ):
+        tmp_file_path = Path(tmpdir / "code.py")
+        input_tree = cst.parse_module(dedent(input_code))
+        self.execution_context = CodemodExecutionContext(
+            directory=tmpdir,
+            dry_run=True,
+            verbose=False,
+            registry=mock.MagicMock(),
+            repo_manager=mock.MagicMock(),
+        )
+
+        self.file_context = FileContext(
+            tmpdir,
+            tmp_file_path,
+            lines_to_exclude,
+            [],
+            [],
+        )
+        codemod_instance = self.initialize_codemod(input_tree)
+        output_tree = codemod_instance.transform_module(input_tree)
+
+        assert output_tree.code == dedent(expected)
+        assert len(self.file_context.codemod_changes) == 0
+
     def run_and_assert_filepath(self, root, file_path, input_code, expected):
         input_tree = cst.parse_module(dedent(input_code))
         self.execution_context = CodemodExecutionContext(

tests/codemods/test_exception_without_raise.py

@@ -54,3 +54,13 @@ def test_raised_exception(self, tmpdir):
         """
         self.run_and_assert(tmpdir, dedent(input_code), dedent(input_code))
         assert len(self.file_context.codemod_changes) == 0
+
+    def test_exclude_line(self, tmpdir):
+        input_code = expected = """\
+        print(1)
+        ValueError("Bad value!")
+        """
+        lines_to_exclude = [2]
+        self.assert_no_change_line_excluded(
+            tmpdir, input_code, expected, lines_to_exclude
+        )

tests/codemods/test_fix_deprecated_abstractproperty.py

@@ -121,3 +121,17 @@ def foo(self):
                 pass
         """
         self.run_and_assert(tmpdir, original_code, new_code)
+
+    def test_exclude_line(self, tmpdir):
+        input_code = expected = """\
+        import abc
+
+        class A:
+            @abc.abstractproperty
+            def foo(self):
+                pass
+        """
+        lines_to_exclude = [4]
+        self.assert_no_change_line_excluded(
+            tmpdir, input_code, expected, lines_to_exclude
+        )

tests/codemods/test_remove_unnecessary_f_str.py

@@ -32,3 +32,12 @@ def test_change(self, tmpdir):
 """
         self.run_and_assert(tmpdir, before, after)
         assert len(self.file_context.codemod_changes) == 3
+
+    def test_exclude_line(self, tmpdir):
+        input_code = expected = """\
+        bad: str = f"bad" + "bad"
+        """
+        lines_to_exclude = [1]
+        self.assert_no_change_line_excluded(
+            tmpdir, input_code, expected, lines_to_exclude
+        )

tests/codemods/test_use_generator.py

@@ -29,3 +29,12 @@ def test_not_global_function(self, tmpdir):
         x = any([i for i in range(10)])
         """
         self.run_and_assert(tmpdir, original_code, expected)
+
+    def test_exclude_line(self, tmpdir):
+        input_code = expected = """\
+        x = any([i for i in range(10)])
+        """
+        lines_to_exclude = [1]
+        self.assert_no_change_line_excluded(
+            tmpdir, input_code, expected, lines_to_exclude
+        )