mirrors/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-08-04 10:48:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Remove condensed display type enum (#17902)

Browse source 
## Summary

See: https://github.com/astral-sh/ruff/pull/17889#discussion_r2076556002
This commit is contained in:
Charlie Marsh 2025-05-06 21:04:03 -04:00 • committed by GitHub
parent a2e9a7732a
commit 443f62e98d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 23 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

67
crates/ty_python_semantic/src/types/display.rs
View file

@ -15,7 +15,6 @@ use crate::types::{
UnionType, WrapperDescriptorKind,
};
use crate::{Db, FxOrderSet};
use rustc_hash::FxHashMap;
impl<'db> Type<'db> {
pub fn display(&self, db: &'db dyn Db) -> DisplayType {
@ -537,31 +536,35 @@ struct DisplayUnionType<'db> {
impl Display for DisplayUnionType<'_> {
fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
fn is_condensable(ty: Type<'_>) -> bool {
matches!(
ty,
Type::IntLiteral(_)
| Type::StringLiteral(_)
| Type::BytesLiteral(_)
| Type::BooleanLiteral(_)
)
}
let elements = self.ty.elements(self.db);
// Group condensed-display types by kind.
let mut grouped_condensed_kinds = FxHashMap::default();
for element in elements {
if let Ok(kind) = CondensedDisplayTypeKind::try_from(*element) {
grouped_condensed_kinds
.entry(kind)
.or_insert_with(Vec::new)
.push(*element);
}
}
let condensed_types = elements
.iter()
.copied()
.filter(|element| is_condensable(*element))
.collect::<Vec<_>>();
let mut join = f.join(" | ");
let mut condensed_types = Some(condensed_types);
for element in elements {
if let Ok(kind) = CondensedDisplayTypeKind::try_from(*element) {
let Some(condensed_kind) = grouped_condensed_kinds.remove(&kind) else {
continue;
};
join.entry(&DisplayLiteralGroup {
literals: condensed_kind,
db: self.db,
});
if is_condensable(*element) {
if let Some(condensed_types) = condensed_types.take() {
join.entry(&DisplayLiteralGroup {
literals: condensed_types,
db: self.db,
});
}
} else {
join.entry(&DisplayMaybeParenthesizedType {
ty: *element,
@ -572,8 +575,6 @@ impl Display for DisplayUnionType<'_> {
join.finish()?;
debug_assert!(grouped_condensed_kinds.is_empty());
Ok(())
}
}
@ -599,28 +600,6 @@ impl Display for DisplayLiteralGroup<'_> {
}
}
/// Enumeration of literal types that are displayed in a "condensed way" inside `Literal` slices.
///
/// For example, `Literal[1] | Literal[2] | Literal["s"]` is displayed as `"Literal[1, 2, "s"]"`.
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash)]
enum CondensedDisplayTypeKind {
LiteralExpression,
}
impl TryFrom<Type<'_>> for CondensedDisplayTypeKind {
type Error = ();
fn try_from(value: Type<'_>) -> Result<Self, Self::Error> {
match value {
Type::IntLiteral(_)
| Type::StringLiteral(_)
| Type::BytesLiteral(_)
| Type::BooleanLiteral(_) => Ok(Self::LiteralExpression),
_ => Err(()),
}
}
}
impl<'db> IntersectionType<'db> {
fn display(&'db self, db: &'db dyn Db) -> DisplayIntersectionType<'db> {
DisplayIntersectionType { db, ty: self }