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

[ty] Add truncation policy for displaying overloads on single line

This commit is contained in:
Dhruv Manilawala 2025-12-12 20:14:59 +05:30
parent 76ffc56e85
commit 3420cdda09
1 changed files with 38 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
crates/ty_python_semantic/src/types/display.rs
View File

@ -746,13 +746,23 @@ impl<'db> FmtDetailed<'db> for DisplayRepresentation<'db> {
}
let separator = if self.settings.multiline { "\n" } else { ", " };
let mut join = f.join(separator);
for signature in signatures {
let display_limit = OVERLOAD_POLICY
.display_limit(signatures.len(), self.settings.multiline);
for signature in signatures.iter().take(display_limit) {
join.entry(
&signature
.bind_self(self.db, Some(typing_self_ty))
.display_with(self.db, self.settings.clone()),
);
}
if !self.settings.multiline {
let omitted = signatures.len().saturating_sub(display_limit);
join.entry(&DisplayOmitted {
count: omitted,
singular: "overload",
plural: "overloads",
});
}
join.finish()?;
if !self.settings.multiline {
f.write_str("]")?;
@ -1152,9 +1162,19 @@ impl<'db> FmtDetailed<'db> for DisplayFunctionType<'db> {
}
let separator = if self.settings.multiline { "\n" } else { ", " };
let mut join = f.join(separator);
for signature in signatures {
let display_limit =
OVERLOAD_POLICY.display_limit(signatures.len(), self.settings.multiline);
for signature in signatures.iter().take(display_limit) {
join.entry(&signature.display_with(self.db, self.settings.clone()));
}
if !self.settings.multiline {
let omitted = signatures.len().saturating_sub(display_limit);
join.entry(&DisplayOmitted {
count: omitted,
singular: "overload",
plural: "overloads",
});
}
join.finish()?;
if !self.settings.multiline {
f.write_str("]")?;
@ -1470,6 +1490,11 @@ impl TupleSpecialization {
}
}
const OVERLOAD_POLICY: TruncationPolicy = TruncationPolicy {
max: 3,
max_when_elided: 2,
};
impl<'db> CallableType<'db> {
pub(crate) fn display<'a>(&'a self, db: &'db dyn Db) -> DisplayCallableType<'a, 'db> {
Self::display_with(self, db, DisplaySettings::default())
@ -1521,9 +1546,19 @@ impl<'db> FmtDetailed<'db> for DisplayCallableType<'_, 'db> {
}
let separator = if self.settings.multiline { "\n" } else { ", " };
let mut join = f.join(separator);
for signature in signatures {
let display_limit =
OVERLOAD_POLICY.display_limit(signatures.len(), self.settings.multiline);
for signature in signatures.iter().take(display_limit) {
join.entry(&signature.display_with(self.db, self.settings.clone()));
}
if !self.settings.multiline {
let omitted = signatures.len().saturating_sub(display_limit);
join.entry(&DisplayOmitted {
count: omitted,
singular: "overload",
plural: "overloads",
});
}
join.finish()?;
if !self.settings.multiline {
f.write_char(']')?;