language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-08-04 18:58:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: transpiler bug

Browse source
This commit is contained in:
Shunsuke Shibayama 2024-10-01 12:15:19 +09:00
parent dcad268dc5
commit 71c460a1bf
1 changed files with 18 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

25
crates/erg_compiler/transpile.rs
View file

@ -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