Python
/
pylyzer
mirror of https://github.com/mtshiba/pylyzer
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update gen_decl.rs

This commit is contained in:
Shunsuke Shibayama 2023-02-14 20:01:33 +09:00
parent 04c4a4dc54
commit f37fb6f521
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
crates/py2erg/gen_decl.rs
View File

@ -39,13 +39,15 @@ pub fn gen_decl_er(hir: HIR, status: CheckStatus) -> DeclFile {
for chunk in hir.module.into_iter() { for chunk in hir.module.into_iter() {
match chunk { match chunk {
Expr::Def(def) => { Expr::Def(def) => {
let name = def.sig.ident().inspect().replace('\0', "");
let typ = def.sig.ident().ref_t().to_string(); let typ = def.sig.ident().ref_t().to_string();
let typ = escape_type(typ); let typ = escape_type(typ);
let decl = format!(".{}: {typ}", def.sig.ident().inspect()); let decl = format!(".{name}: {typ}");
code += &decl; code += &decl;
} }
Expr::ClassDef(def) => { Expr::ClassDef(def) => {
let decl = format!(".{}: ClassType", def.sig.ident().inspect()); let name = def.sig.ident().inspect().replace('\0', "");
let decl = format!(".{name}: ClassType");
code += &decl; code += &decl;
} }
_ => {} _ => {}