From 71c460a1bfd3e034d603e37ded4c19f20e636d92 Mon Sep 17 00:00:00 2001
From: Shunsuke Shibayama <sbym1346@gmail.com>
Date: Tue, 1 Oct 2024 12:15:19 +0900
Subject: [PATCH] fix: transpiler bug

---
 crates/erg_compiler/transpile.rs | 25 ++++++++++++++++++-------
 1 file changed, 18 insertions(+), 7 deletions(-)

diff --git a/crates/erg_compiler/transpile.rs b/crates/erg_compiler/transpile.rs
index c7b577ec9..ca56ab6dc 100644
--- a/crates/erg_compiler/transpile.rs
+++ b/crates/erg_compiler/transpile.rs
@@ -454,8 +454,11 @@ impl PyScriptGenerator {
     pub fn transpile(&mut self, hir: HIR) -> PyScript {
         let mut code = String::new();
         for chunk in hir.module.into_iter() {
-            code += &self.transpile_expr(chunk);
-            code.push('\n');
+            let expr = self.transpile_expr(chunk);
+            if !expr.is_empty() {
+                code += &expr;
+                code.push('\n');
+            }
         }
         code = std::mem::take(&mut self.prelude) + &code;
         PyScript {
@@ -639,13 +642,18 @@ impl PyScriptGenerator {
             Expr::Compound(comp) => {
                 let mut code = "".to_string();
                 for expr in comp.into_iter() {
-                    code += &self.transpile_expr(expr);
-                    code += &format!("\n{}", "    ".repeat(self.level));
+                    let expr = self.transpile_expr(expr);
+                    if !expr.is_empty() {
+                        code += &expr;
+                        code += &format!("\n{}", "    ".repeat(self.level));
+                    }
                 }
                 code
             }
             Expr::Import(acc) => {
-                let full_name = Str::from(acc.show());
+                let full_name = acc
+                    .qual_name()
+                    .map_or(acc.show(), |s| s.replace(".__init__", ""));
                 let root = PyCodeGenerator::get_root(&acc);
                 self.prelude += &format!(
                     "{} = __import__(\"{full_name}\")\n",
@@ -1118,8 +1126,11 @@ impl PyScriptGenerator {
                     }
                 }
             }
-            code += &self.transpile_expr(chunk);
-            code.push('\n');
+            let expr = self.transpile_expr(chunk);
+            if !expr.is_empty() {
+                code += &expr;
+                code.push('\n');
+            }
         }
         self.level -= 1;
         code