Revert extensions change

astral-sh · Nov 9, 2023 · 42665cb · 42665cb
1 parent 545bd30
commit 42665cb
Show file tree

Hide file tree

Showing 2 changed files with 71 additions and 2 deletions.
diff --git a/crates/ruff_python_semantic/src/model.rs b/crates/ruff_python_semantic/src/model.rs
@@ -8,6 +8,7 @@ use ruff_python_ast::call_path::{collect_call_path, from_unqualified_name, CallP
 use ruff_python_ast::helpers::from_relative_import;
 use ruff_python_ast::{self as ast, Expr, Operator, Stmt};
 use ruff_python_stdlib::path::is_python_stub_file;
+use ruff_python_stdlib::typing::is_typing_extension;
 use ruff_text_size::{Ranged, TextRange, TextSize};
 
 use crate::binding::{
@@ -174,8 +175,16 @@ impl<'a> SemanticModel<'a> {
 
     /// Return `true` if the call path is a reference to `typing.${target}`.
     pub fn match_typing_call_path(&self, call_path: &CallPath, target: &str) -> bool {
-        if let ["typing" | "_typeshed" | "typing_extensions", name] = call_path.as_slice() {
-            if *name == target {
+        if call_path.as_slice() == ["typing", target] {
+            return true;
+        }
+
+        if call_path.as_slice() == ["_typeshed", target] {
+            return true;
+        }
+
+        if is_typing_extension(target) {
+            if call_path.as_slice() == ["typing_extensions", target] {
                 return true;
             }
         }

diff --git a/crates/ruff_python_stdlib/src/typing.rs b/crates/ruff_python_stdlib/src/typing.rs
@@ -1,3 +1,63 @@
+/// Returns `true` if a name is a member of Python's `typing_extensions` module.
+///
+/// See: <https://pypi.org/project/typing-extensions/>
+pub fn is_typing_extension(member: &str) -> bool {
+    matches!(
+        member,
+        "Annotated"
+            | "Any"
+            | "AsyncContextManager"
+            | "AsyncGenerator"
+            | "AsyncIterable"
+            | "AsyncIterator"
+            | "Awaitable"
+            | "ChainMap"
+            | "ClassVar"
+            | "Concatenate"
+            | "ContextManager"
+            | "Coroutine"
+            | "Counter"
+            | "DefaultDict"
+            | "Deque"
+            | "Final"
+            | "Literal"
+            | "LiteralString"
+            | "NamedTuple"
+            | "Never"
+            | "NewType"
+            | "NotRequired"
+            | "OrderedDict"
+            | "ParamSpec"
+            | "ParamSpecArgs"
+            | "ParamSpecKwargs"
+            | "Protocol"
+            | "Required"
+            | "Self"
+            | "TYPE_CHECKING"
+            | "Text"
+            | "Type"
+            | "TypeAlias"
+            | "TypeGuard"
+            | "TypeVar"
+            | "TypeVarTuple"
+            | "TypedDict"
+            | "Unpack"
+            | "assert_never"
+            | "assert_type"
+            | "clear_overloads"
+            | "final"
+            | "get_type_hints"
+            | "get_args"
+            | "get_origin"
+            | "get_overloads"
+            | "is_typeddict"
+            | "overload"
+            | "override"
+            | "reveal_type"
+            | "runtime_checkable"
+    )
+}
+
 /// Returns `true` if a call path is a generic from the Python standard library (e.g. `list`, which
 /// can be used as `list[int]`).
 ///