Python
/
ruff
mirror of https://github.com/astral-sh/ruff
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

is_subclass_of

This commit is contained in:
Jack O'Connor 2025-08-19 19:44:18 -07:00
parent 5d9a8af775
commit 2be369f299
1 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
crates/ty_python_semantic/src/types/class.rs
View File

@ -1405,6 +1405,19 @@ impl<'db> ClassSingletonType<'db> {
Self::NewType(new_type) => new_type.dataclass_transformer_params(db),
}
}
pub(super) fn is_subclass_of(
self,
db: &'db dyn Db,
specialization: Option<Specialization<'db>>,
other: ClassType<'db>,
) -> bool {
match self {
Self::Literal(literal) => literal.is_subclass_of(db, specialization, other),
// A NewType can't be specialized.
Self::NewType(new_type) => new_type.is_subclass_of(db, other),
}
}
}
impl<'db> From<ClassSingletonType<'db>> for Type<'db> {
@ -3249,7 +3262,7 @@ pub struct NewTypeClass<'db> {
#[returns(ref)]
name: Name,
parent: Box<ClassType<'db>>,
parent: ClassType<'db>,
}
impl<'db> NewTypeClass<'db> {
@ -3332,6 +3345,15 @@ impl<'db> NewTypeClass<'db> {
) -> Option<DataclassTransformerParams> {
self.parent(db).dataclass_transformer_params(db)
}
pub(super) fn is_subclass_of(self, db: &'db dyn Db, other: ClassType<'db>) -> bool {
let parent = self.parent(db);
if parent == other {
true
} else {
parent.is_subclass_of(db, other)
}
}
}
impl<'db> get_size2::GetSize for NewTypeClass<'_> {}