Correctly handle references in __all__ definitions when renaming symbols in autofixes (#10527)

2024-03-22 20:06:35 +00:00
parent 61b7982422
commit 9feb9b0aa8
16 changed files with 173 additions and 20 deletions
--- a/crates/ruff_python_semantic/src/reference.rs
+++ b/crates/ruff_python_semantic/src/reference.rs
@@ -93,6 +93,12 @@ impl ResolvedReference {
        self.flags
            .intersects(SemanticModelFlags::TYPE_CHECKING_BLOCK)
    }
+
+    /// Return `true` if the context is in the r.h.s. of an `__all__` definition.
+    pub const fn in_dunder_all_definition(&self) -> bool {
+        self.flags
+            .intersects(SemanticModelFlags::DUNDER_ALL_DEFINITION)
+    }
 }

 impl Ranged for ResolvedReference {