fix(gen): add bases for ConstraintType and TypeVar

midas/checker/preamble.py

@@ -54,6 +54,11 @@ class Preamble(Environment):
             returns=self._list_of(map_out),  # TODO: replace with Iterable[U]
             type_vars=[map_in, map_out],
         )
+        self._def_function(
+            name="input",
+            pos=[Param("prompt", TopType(), required=False)],
+            returns=self._types.get_type("str"),
+        )
 
     def _list_of(self, item_type: Type) -> Type:
         return self._types.apply_generic(self._types.get_type("list"), [item_type])

midas/checker/registry.py

@@ -404,6 +404,9 @@ class TypesRegistry:
             case ConstraintType(type=base):
                 return self.lookup_member(base, member_name)
 
+            case TypeVar(bound=bound) if bound is not None:
+                return self.lookup_member(bound, member_name)
+
             case UnknownType():
                 return UnknownType()
 

midas/generator/stubs.py

@@ -127,6 +127,12 @@ class StubsGenerator:
                     body_subsitutions | substitutions,
                 )
 
+            case ConstraintType(type=base):
+                return self.get_bases(base)
+
+            case TypeVar(bound=bound) if bound is not None:
+                return [self.dump_type(bound)], {}
+
             case _:
                 return [], {}