mirrors/uv
1
0
Fork 0
mirror of https://github.com/astral-sh/uv.git synced 2025-07-07 13:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Deduplicate source edges in annotations (#4530)

Browse source 
## Summary

Not relevant today, but it will be once we support universal resolution,
in which a package can be repeated.
This commit is contained in:
Charlie Marsh 2024-06-25 17:10:09 -04:00 • committed by GitHub
parent ad42206e50
commit 08bf6fb87c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 23 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

45
crates/uv-resolver/src/resolution/display.rs
View file

@ -194,19 +194,24 @@ impl std::fmt::Display for DisplayResolutionGraph<'_> {
// If enabled, include annotations to indicate the dependencies that requested each
// package (e.g., `# via mypy`).
if self.include_annotations {
// Display all dependencies.
let mut edges = petgraph
.edges_directed(index, Direction::Incoming)
.map(|edge| &petgraph[edge.source()])
.collect::<Vec<_>>();
edges.sort_unstable_by_key(|package| package.name());
// Display all dependents (i.e., all packages that depend on the current package).
let dependents = {
let mut dependents = petgraph
.edges_directed(index, Direction::Incoming)
.map(|edge| &petgraph[edge.source()])
.map(distribution_types::Name::name)
.collect::<Vec<_>>();
dependents.sort_unstable();
dependents.dedup();
dependents
};
// Include all external sources (e.g., requirements files).
let default = BTreeSet::default();
let source = sources.get(node.name()).unwrap_or(&default);
match self.annotation_style {
AnnotationStyle::Line => match edges.as_slice() {
AnnotationStyle::Line => match dependents.as_slice() {
[] if source.is_empty() => {}
[] if source.len() == 1 => {
let separator = if has_hashes { "\n " } else { " " };
@ -215,19 +220,19 @@ impl std::fmt::Display for DisplayResolutionGraph<'_> {
.to_string();
annotation = Some((separator, comment));
}
edges => {
dependents => {
let separator = if has_hashes { "\n " } else { " " };
let deps = edges
let dependents = dependents
.iter()
.map(|dependency| format!("{}", dependency.name()))
.map(ToString::to_string)
.chain(source.iter().map(ToString::to_string))
.collect::<Vec<_>>()
.join(", ");
let comment = format!("# via {deps}").green().to_string();
let comment = format!("# via {dependents}").green().to_string();
annotation = Some((separator, comment));
}
},
AnnotationStyle::Split => match edges.as_slice() {
AnnotationStyle::Split => match dependents.as_slice() {
[] if source.is_empty() => {}
[] if source.len() == 1 => {
let separator = "\n";
@ -236,25 +241,21 @@ impl std::fmt::Display for DisplayResolutionGraph<'_> {
.to_string();
annotation = Some((separator, comment));
}
[edge] if source.is_empty() => {
[dependent] if source.is_empty() => {
let separator = "\n";
let comment = format!(" # via {}", edge.name()).green().to_string();
let comment = format!(" # via {dependent}").green().to_string();
annotation = Some((separator, comment));
}
edges => {
dependents => {
let separator = "\n";
let deps = source
let dependent = source
.iter()
.map(ToString::to_string)
.chain(
edges
.iter()
.map(|dependency| format!("{}", dependency.name())),
)
.chain(dependents.iter().map(ToString::to_string))
.map(|name| format!(" # {name}"))
.collect::<Vec<_>>()
.join("\n");
let comment = format!(" # via\n{deps}").green().to_string();
let comment = format!(" # via\n{dependent}").green().to_string();
annotation = Some((separator, comment));
}
},