fix: implement type comments for for loops

Author: jvllmr

src/unparser.rs

@@ -211,6 +211,7 @@ impl Unparser {
         }
         self.fill("def ");
         self.write_str(&node.name);
+
         if node.type_params.len() > 0 {
             self.write_str("[");
             let mut type_params_iter = node.type_params.iter().peekable();
@@ -407,6 +408,10 @@ impl Unparser {
         self.write_str(" in ");
         self.unparse_expr(&node.iter);
         self.write_str(":");
+        if let Some(type_comment) = &node.type_comment {
+            self.write_str(" #type: ignore");
+            self.write_str(type_comment);
+        }
         self.block(|block_self| {
             for value in &node.body {
                 block_self.unparse_stmt(value);
@@ -427,6 +432,10 @@ impl Unparser {
         self.write_str(" in ");
         self.unparse_expr(&node.iter);
         self.write_str(":");
+        if let Some(type_comment) = &node.type_comment {
+            self.write_str(" #type: ignore");
+            self.write_str(type_comment);
+        }
         self.block(|block_self| {
             for value in &node.body {
                 block_self.unparse_stmt(value);

test_files/for_loop.py

@@ -0,0 +1,7 @@
+for name_3 in name_0:  # type: ignore[name-defined] # noqa: F821
+    pass
+
+
+async def main():
+    async for name_3 in name_0:  # type: ignore[name-defined] # noqa: F821
+        pass