Skip to content

Commit c2e7c56

Browse files
committed
Small refactoring
1 parent a477396 commit c2e7c56

File tree

2 files changed

+6
-9
lines changed

2 files changed

+6
-9
lines changed

src/codemodder/codemods/test/utils.py

-2
Original file line numberDiff line numberDiff line change
@@ -118,8 +118,6 @@ def assert_num_changes(
118118
assert len(changes[0].changes) == expected_num_changes
119119
actual_num = len(changes[0].changes)
120120

121-
# actual_num = len(changes)
122-
123121
if min_num_changes is not None:
124122
assert (
125123
actual_num >= min_num_changes

src/core_codemods/semgrep/semgrep_no_csrf_exempt.py

+6-7
Original file line numberDiff line numberDiff line change
@@ -27,13 +27,12 @@ def leave_Decorator(
2727
):
2828
return updated_node
2929
# Due to semgrep's odd way of reporting the position for this (decorators + functiondef), we match by line only
30-
if self.node_is_selected_by_line_only(original_node):
31-
if (
32-
self.find_base_name(original_node.decorator)
33-
== "django.views.decorators.csrf.csrf_exempt"
34-
):
35-
self.report_change(original_node)
36-
return cst.RemovalSentinel.REMOVE
30+
if self.node_is_selected_by_line_only(original_node) and (
31+
self.find_base_name(original_node.decorator)
32+
== "django.views.decorators.csrf.csrf_exempt"
33+
):
34+
self.report_change(original_node)
35+
return cst.RemovalSentinel.REMOVE
3736
return updated_node
3837

3938

0 commit comments

Comments
 (0)