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

Update transpile.rs

Browse source
This commit is contained in:
Shunsuke Shibayama 2024-02-19 01:37:45 +09:00
parent 9e2beff6db
commit 6d61e49bd5
1 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
crates/erg_compiler/transpile.rs
View file

@ -498,6 +498,8 @@ impl PyScriptGenerator {
if self.range_ops_loaded {
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_float.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_array.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_dict.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_set.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_bytes.py"));
} else {
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_int.py"));
@ -506,6 +508,8 @@ impl PyScriptGenerator {
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_str.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_float.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_array.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_dict.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_set.py"));
self.prelude += &Self::replace_import(include_str!("lib/core/_erg_bytes.py"));
}
self.builtin_types_loaded = true;
@ -554,11 +558,12 @@ impl PyScriptGenerator {
},
Expr::Set(set) => match set {
Set::Normal(st) => {
let mut code = "{".to_string();
self.load_builtin_types_if_not();
let mut code = "Set({".to_string();
for elem in st.elems.pos_args {
code += &format!("{},", self.transpile_expr(elem.expr));
}
code += "}";
code += "})";
code
}
other => todo!("transpiling {other}"),
@ -576,7 +581,8 @@ impl PyScriptGenerator {
},
Expr::Dict(dict) => match dict {
Dict::Normal(dic) => {
let mut code = "{".to_string();
self.load_builtin_types_if_not();
let mut code = "Dict({".to_string();
for kv in dic.kvs {
code += &format!(
"({}): ({}),",
@ -584,7 +590,7 @@ impl PyScriptGenerator {
self.transpile_expr(kv.value)
);
}
code += "}";
code += "})";
code
}
other => todo!("transpiling {other}"),