HEL/midas
1
0
Fork 0
Code Issues Pull Requests Releases Activity

feat(gen): generate alias stubs

Browse Source
This commit is contained in:
Louis Heredero
2026-07-01 14:43:30 +02:00
parent 3260ae4a1e
commit a59a58d21a
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
midas/generator/stubs.py
View File

@@ -91,6 +91,21 @@ class StubsGenerator:
def generate_stub(self, name: str, type: Type): def generate_stub(self, name: str, type: Type):
base_type: Type = type base_type: Type = type
# TODO: improve
match type:
case DerivedType(name=name_) | GenericType(name=name_) if name_ == name:
pass
case UnitType() if name == "None":
pass
case TopType() if name == "Any":
pass
case _:
alias = ast.Assign(
targets=[ast.Name(id=name)], value=self.dump_type(type)
)
self.add_stub(alias)
return
members: dict[str, Member] = self.types._members.get(name, {}) members: dict[str, Member] = self.types._members.get(name, {})
if isinstance(base_type, (BaseType, TopType, UnitType)) and len(members) == 0: if isinstance(base_type, (BaseType, TopType, UnitType)) and len(members) == 0:
return return