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

fix(checker): allow iterating on unknown

Browse Source
This commit is contained in:
Louis Heredero
2026-06-29 11:13:47 +02:00
parent 3392bc347d
commit 6f6245d283
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
midas/checker/python.py
View File

@@ -458,13 +458,16 @@ class PythonTyper(
pass pass
def visit_for_stmt(self, stmt: p.ForStmt) -> None: def visit_for_stmt(self, stmt: p.ForStmt) -> None:
item_type: Optional[Type] = self._get_iterator_type(stmt.iterator) item_type: Type = UnknownType()
if item_type is None: iterator_type: Type = self.type_of(stmt.iterator)
iterator_type: Type = self.compute_type(stmt.iterator) if iterator_type != UnknownType():
self.reporter.error( maybe_item_type = self._get_iterator_type(stmt.iterator, iterator_type)
stmt.iterator.location, f"{iterator_type} is not iterable" if maybe_item_type is None:
) self.reporter.error(
item_type = UnknownType() stmt.iterator.location, f"{iterator_type} is not iterable"
)
else:
item_type = maybe_item_type
self._assign(stmt.location, stmt.target, item_type) self._assign(stmt.location, stmt.target, item_type)
self.judge(stmt.target, item_type) self.judge(stmt.target, item_type)
@@ -1155,9 +1158,8 @@ class PythonTyper(
return False return False
return True return True
def _get_iterator_type(self, expr: p.Expr) -> Optional[Type]: def _get_iterator_type(self, expr: p.Expr, type: Type) -> Optional[Type]:
# TODO: lookup __iter__ # TODO: lookup __iter__
type: Type = self.type_of(expr)
getitem: Optional[Type] = self.types.lookup_member(type, "__getitem__") getitem: Optional[Type] = self.types.lookup_member(type, "__getitem__")
if getitem is None: if getitem is None:
return None return None