From af1aba41e7f8d413486514566cb5d0d53931a1c3 Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Tue, 16 Jun 2026 00:36:43 +0200
Subject: [PATCH] feat(gen): handle for loops

---
 midas/generator/generator.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/midas/generator/generator.py b/midas/generator/generator.py
index ceabdd5..bef6e16 100644
--- a/midas/generator/generator.py
+++ b/midas/generator/generator.py
@@ -161,5 +161,13 @@ class Generator(p.Stmt.Visitor[ast.stmt], p.Expr.Visitor[ast.expr]):
     def visit_pass(self, stmt: p.Pass) -> ast.stmt:
         return ast.Pass()
 
+    def visit_for_stmt(self, stmt: p.ForStmt) -> ast.stmt:
+        return ast.For(
+            target=stmt.target.accept(self),
+            iter=stmt.iterator.accept(self),
+            body=self._visit_body(stmt.body),
+            orelse=[],
+        )
+
     def _visit_body(self, stmts: list[p.Stmt]) -> list[ast.stmt]:
         return [stmt.accept(self) for stmt in stmts]