Delete token range

adamchainz · Feb 11, 2024 · 5c0b991 · 5c0b991
1 parent 0f9ba24
commit 5c0b991
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/src/django_upgrade/fixers/request_user_attributes.py b/src/django_upgrade/fixers/request_user_attributes.py
@@ -64,5 +64,4 @@ def rewrite_user_attribute(tokens: list[Token], i: int, *, attr: str) -> None:
     j = find(tokens, i, name=NAME, src=attr)
     y = find(tokens, j, name=OP, src="(")
     z = find(tokens, y, name=OP, src=")")
-    del tokens[z]
-    del tokens[y]
+    del tokens[y : z + 1]
diff --git a/tests/fixers/test_request_user_attributes.py b/tests/fixers/test_request_user_attributes.py
@@ -149,15 +149,28 @@ def test_if_self_request_user_is_authenticated():
 
 def test_spaces_between_noop():
     check_noop(
-        "request . user . is_authenticated  ",
+        "request . user . is_authenticated ",
         settings,
     )
 
 
 def test_spaces_between():
     check_transformed(
         "request . user . is_authenticated ( )",
-        "request . user . is_authenticated  ",
+        "request . user . is_authenticated ",
+        settings,
+    )
+
+
+def test_comment_between():
+    check_transformed(
+        """\
+        request.user.is_anonymous(  # something
+        )
+        """,
+        """\
+        request.user.is_anonymous
+        """,
         settings,
     )