main commit

venv/lib/python3.12/site-packages/sqlalchemy/ext/mypy/names.py

@@ -1,5 +1,5 @@
 # ext/mypy/names.py
-# Copyright (C) 2021-2025 the SQLAlchemy authors and contributors
+# Copyright (C) 2021 the SQLAlchemy authors and contributors
 # <see AUTHORS file>
 #
 # This module is part of SQLAlchemy and is released under
@@ -58,14 +58,6 @@ NAMED_TYPE_BUILTINS_STR = "builtins.str"
 NAMED_TYPE_BUILTINS_LIST = "builtins.list"
 NAMED_TYPE_SQLA_MAPPED = "sqlalchemy.orm.base.Mapped"
 
-_RelFullNames = {
-    "sqlalchemy.orm.relationships.Relationship",
-    "sqlalchemy.orm.relationships.RelationshipProperty",
-    "sqlalchemy.orm.relationships._RelationshipDeclared",
-    "sqlalchemy.orm.Relationship",
-    "sqlalchemy.orm.RelationshipProperty",
-}
-
 _lookup: Dict[str, Tuple[int, Set[str]]] = {
     "Column": (
         COLUMN,
@@ -74,9 +66,24 @@ _lookup: Dict[str, Tuple[int, Set[str]]] = {
             "sqlalchemy.sql.Column",
         },
     ),
-    "Relationship": (RELATIONSHIP, _RelFullNames),
-    "RelationshipProperty": (RELATIONSHIP, _RelFullNames),
-    "_RelationshipDeclared": (RELATIONSHIP, _RelFullNames),
+    "Relationship": (
+        RELATIONSHIP,
+        {
+            "sqlalchemy.orm.relationships.Relationship",
+            "sqlalchemy.orm.relationships.RelationshipProperty",
+            "sqlalchemy.orm.Relationship",
+            "sqlalchemy.orm.RelationshipProperty",
+        },
+    ),
+    "RelationshipProperty": (
+        RELATIONSHIP,
+        {
+            "sqlalchemy.orm.relationships.Relationship",
+            "sqlalchemy.orm.relationships.RelationshipProperty",
+            "sqlalchemy.orm.Relationship",
+            "sqlalchemy.orm.RelationshipProperty",
+        },
+    ),
     "registry": (
         REGISTRY,
         {
@@ -297,7 +304,7 @@ def type_id_for_callee(callee: Expression) -> Optional[int]:
 
 
 def type_id_for_named_node(
-    node: Union[NameExpr, MemberExpr, SymbolNode],
+    node: Union[NameExpr, MemberExpr, SymbolNode]
 ) -> Optional[int]:
     type_id, fullnames = _lookup.get(node.name, (None, None))