Resolve pyrefly types for literals.

Reviewed By: DinoV

Differential Revision: D96212432

fbshipit-source-id: 7ec265cd970eb16924019da97cd7a9ad32ab63dc

Author: martindemello

cinderx/PythonLib/cinderx/compiler/static/pyrefly_info.py

@@ -34,6 +34,7 @@ class TypeInfo(TypedDict):
 class TypeTableEntry(TypedDict):
     kind: str
     qname: object
+    promoted_type: TypeKind
 
 
 @dataclass(slots=True, eq=True)
@@ -128,7 +129,17 @@ def lookup(self, node: AST) -> str:
             return ""
         return self._type_to_str(type_index)
 
-    def lookup_class_qname(self, node: AST) -> str:
+    def _get_type_qname(self, type_index: TypeKind) -> str:
+        entry = self._type_table[type_index]
+        if entry["kind"] == "class":
+            # Ignore the generic args
+            return str(entry["qname"])
+        elif entry["kind"] == "literal":
+            if "promoted_type" in entry:
+                return self._get_type_qname(entry["promoted_type"])
+        return ""
+
+    def lookup_typename(self, node: AST) -> str:
         """Look up the qname for an AST node if its type is a simple class.
 
         We treat generic classes as their unparametrised "base" version,
@@ -137,11 +148,7 @@ def lookup_class_qname(self, node: AST) -> str:
         type_index = self._lookup(node)
         if type_index is None:
             return ""
-        entry = self._type_table[type_index]
-        if entry["kind"] == "class":
-            # Ignore the generic args
-            return str(entry["qname"])
-        return ""
+        return self._get_type_qname(type_index)
 
     @classmethod
     def load_json(cls, json_path: str) -> PyreflyTypeInfo:

cinderx/PythonLib/cinderx/compiler/static/pyrefly_type_binder.py

@@ -85,10 +85,10 @@ def __init__(
     def visit(self, node: AST, *args: object) -> NarrowingEffect | None:
         ret = super().visit(node, *args)
         if isinstance(node, ast.expr) and self._type_info is not None:
-            # For now, we only try to get type information for classes,
-            # disregarding their type parameters, and doing nothing if we see a
-            # some other type_info kind like a callable.
-            classname = self._type_info.lookup_class_qname(node)
+            # For now, we only try to get type information for class instances
+            # and literals, disregarding any type parameters, and doing nothing
+            # if we see a some other type_info kind like a callable.
+            classname = self._type_info.lookup_typename(node)
             if classname:
                 resolved = _resolve_classname(classname, self.modules)
                 if resolved is not None: