language-servers/pylyzer
1
0
Fork 0
mirror of https://github.com/mtshiba/pylyzer.git synced 2025-08-04 14:28:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Support operator checks for user-defined classes

Browse source
This commit is contained in:
Shunsuke Shibayama 2023-02-24 14:57:08 +09:00
parent f360c2c763
commit 4b3ff53295
5 changed files with 39 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

9
crates/py2erg/convert.rs
View file

@ -843,7 +843,8 @@ impl ASTConverter {
// self.y = y
// self.z = z
// ↓
// {x: Int, y: Int, z: Never}, .__call__(x: Int, y: Int, z: Obj): Self = .unreachable()
// requirement : {x: Int, y: Int, z: Never}
// returns : .__call__(x: Int, y: Int, z: Obj): Self = .unreachable()
fn extract_init(&mut self, base_type: &mut Option<Expr>, init_def: Def) -> Option<Def> {
self.check_init_sig(&init_def.sig)?;
let l_brace = Token::new(
@ -988,8 +989,8 @@ impl ASTConverter {
.map(|id| &id.inspect()[..] == "__init__")
.unwrap_or(false)
{
if let Some(init_def) = self.extract_init(&mut base_type, def) {
attrs.push(ClassAttr::Def(init_def));
if let Some(call_def) = self.extract_init(&mut base_type, def) {
attrs.insert(0, ClassAttr::Def(call_def));
init_is_defined = true;
}
} else {
@ -1041,7 +1042,7 @@ impl ASTConverter {
}
}
if !init_is_defined {
attrs.push(ClassAttr::Def(self.gen_default_init(0)));
attrs.insert(0, ClassAttr::Def(self.gen_default_init(0)));
}
(base_type, ClassAttrs::new(attrs))
}