Implement as GetSpecialAttr.

Author: dnwpark

tests/test_type_eval.py

@@ -27,6 +27,7 @@
     GetArgs,
     GetAttr,
     GetName,
+    GetSpecialAttr,
     GetType,
     GetAnnotations,
     IsSub,
@@ -40,7 +41,6 @@
     Slice,
     SpecialFormEllipsis,
     StrConcat,
-    TypeName,
     Uppercase,
     _BoolLiteral,
 )
@@ -975,23 +975,82 @@ class Container2[T]: ...
     assert eval_typing(GetArg[t, Container, Literal[1]]) == Never
 
 
+class OuterType:
+    class InnerType:
+        pass
+
+
 def test_eval_typename_01():
-    d = eval_typing(TypeName[int])
+    d = eval_typing(GetSpecialAttr[int, Literal["__name__"]])
     assert d == Literal["int"]
-    d = eval_typing(TypeName[str])
+    d = eval_typing(GetSpecialAttr[str, Literal["__name__"]])
     assert d == Literal["str"]
-    d = eval_typing(TypeName[list[int]])
+    d = eval_typing(GetSpecialAttr[list[int], Literal["__name__"]])
     assert d == Literal["list"]
-    d = eval_typing(TypeName[list[str]])
+    d = eval_typing(GetSpecialAttr[list[str], Literal["__name__"]])
     assert d == Literal["list"]
 
     class C:
         pass
 
-    d = eval_typing(TypeName[C])
+    d = eval_typing(GetSpecialAttr[OuterType, Literal["__name__"]])
+    assert d == Literal["OuterType"]
+    d = eval_typing(GetSpecialAttr[OuterType.InnerType, Literal["__name__"]])
+    assert d == Literal["InnerType"]
+    d = eval_typing(GetSpecialAttr[C, Literal["__name__"]])
     assert d == Literal["C"]
 
-    d = eval_typing(TypeName[GetA1[int, str]])
+    d = eval_typing(GetSpecialAttr[GetA1[int, str], Literal["__name__"]])
+    assert d == Literal["int"]
+
+
+def test_eval_typename_02():
+    d = eval_typing(GetSpecialAttr[int, Literal["__module__"]])
+    assert d == Literal["builtins"]
+    d = eval_typing(GetSpecialAttr[str, Literal["__module__"]])
+    assert d == Literal["builtins"]
+    d = eval_typing(GetSpecialAttr[list[int], Literal["__module__"]])
+    assert d == Literal["builtins"]
+    d = eval_typing(GetSpecialAttr[list[str], Literal["__module__"]])
+    assert d == Literal["builtins"]
+
+    class C:
+        pass
+
+    d = eval_typing(GetSpecialAttr[OuterType, Literal["__module__"]])
+    assert d == Literal["tests.test_type_eval"]
+    d = eval_typing(GetSpecialAttr[OuterType.InnerType, Literal["__module__"]])
+    assert d == Literal["tests.test_type_eval"]
+    d = eval_typing(GetSpecialAttr[C, Literal["__module__"]])
+    assert d == Literal["tests.test_type_eval"]
+
+    d = eval_typing(GetSpecialAttr[GetA1[int, str], Literal["__module__"]])
+    assert d == Literal["builtins"]
+
+
+def test_eval_typename_03():
+    d = eval_typing(GetSpecialAttr[int, Literal["__qualname__"]])
+    assert d == Literal["int"]
+    d = eval_typing(GetSpecialAttr[str, Literal["__qualname__"]])
+    assert d == Literal["str"]
+    d = eval_typing(GetSpecialAttr[list[int], Literal["__qualname__"]])
+    assert d == Literal["list"]
+    d = eval_typing(GetSpecialAttr[list[str], Literal["__qualname__"]])
+    assert d == Literal["list"]
+
+    class C:
+        pass
+
+    d = eval_typing(GetSpecialAttr[OuterType, Literal["__qualname__"]])
+    assert d == Literal["OuterType"]
+    d = eval_typing(
+        GetSpecialAttr[OuterType.InnerType, Literal["__qualname__"]]
+    )
+    assert d == Literal["OuterType.InnerType"]
+    d = eval_typing(GetSpecialAttr[C, Literal["__qualname__"]])
+    assert d == Literal["test_eval_typename_03.<locals>.C"]
+
+    d = eval_typing(GetSpecialAttr[GetA1[int, str], Literal["__qualname__"]])
     assert d == Literal["int"]
 
 

typemap/type_eval/_eval_operators.py

@@ -26,6 +26,7 @@
     GetArg,
     GetArgs,
     GetAttr,
+    GetSpecialAttr,
     InitField,
     IsSubSimilar,
     IsSubtype,
@@ -40,7 +41,6 @@
     Slice,
     SpecialFormEllipsis,
     StrConcat,
-    TypeName,
     Uncapitalize,
     Uppercase,
     _BoolLiteral,
@@ -907,10 +907,29 @@ def _eval_GetArgs(tp, base, *, ctx) -> typing.Any:
     return tuple[*args]  # type: ignore[valid-type]
 
 
-@type_eval.register_evaluator(TypeName)
+@type_eval.register_evaluator(GetSpecialAttr)
 @_lift_over_unions
-def _eval_TypeName(tp, *, ctx) -> typing.Any:
-    return typing.Literal[tp.__name__]
+def _eval_GetSpecialAttr(tp, attr, *, ctx) -> typing.Any:
+    if not (
+        _typing_inspect.is_generic_alias(attr)
+        and attr.__origin__ is typing.Literal
+        and isinstance(attr.__args__[0], str)
+    ):
+        raise TypeError(
+            f"Invalid type argument to GetSpecialAttr: "
+            f"{attr} is not a string Literal"
+        )
+    if attr.__args__[0] == "__name__":
+        return typing.Literal[tp.__name__]
+    elif attr.__args__[0] == "__module__":
+        return typing.Literal[tp.__module__]
+    elif attr.__args__[0] == "__qualname__":
+        return typing.Literal[tp.__qualname__]
+    else:
+        raise TypeError(
+            f"Invalid type argument to GetSpecialAttr: "
+            f"{attr} must be one of __name__, __module__, or __qualname__"
+        )
 
 
 @type_eval.register_evaluator(GetAnnotations)

typemap/typing.py

@@ -143,7 +143,7 @@ class GetArgs[Tp, Base]:
     pass
 
 
-class TypeName[T: type]:
+class GetSpecialAttr[T: type, Attr: str]:
     pass