mirror of
https://github.com/astral-sh/uv.git
synced 2025-11-17 02:52:45 +00:00
1512e07a2e
Needed to prevent circular dependencies in my toolchain work (#2931). I think this is probably a reasonable change as we move towards persistent configuration too? Unfortunately `BuildIsolation` needs to be in `uv-types` to avoid circular dependencies still. We might be able to resolve that in the future.
48 lines
1.5 KiB
Rust
48 lines
1.5 KiB
Rust
use pep508_rs::PackageName;
|
|
use rustc_hash::FxHashSet;
|
|
use uv_configuration::{Reinstall, Upgrade};
|
|
|
|
/// Tracks locally installed packages that should not be selected during resolution.
|
|
#[derive(Debug, Default, Clone)]
|
|
pub enum Exclusions {
|
|
#[default]
|
|
None,
|
|
/// Exclude some local packages from consideration, e.g. from `--reinstall-package foo --upgrade-package bar`
|
|
Some(FxHashSet<PackageName>),
|
|
/// Exclude all local packages from consideration, e.g. from `--reinstall` or `--upgrade`
|
|
All,
|
|
}
|
|
|
|
impl Exclusions {
|
|
pub fn new(reinstall: Reinstall, upgrade: Upgrade) -> Self {
|
|
if upgrade.is_all() || reinstall.is_all() {
|
|
Self::All
|
|
} else {
|
|
let mut exclusions: FxHashSet<PackageName> =
|
|
if let Reinstall::Packages(packages) = reinstall {
|
|
FxHashSet::from_iter(packages)
|
|
} else {
|
|
FxHashSet::default()
|
|
};
|
|
|
|
if let Upgrade::Packages(packages) = upgrade {
|
|
exclusions.extend(packages);
|
|
};
|
|
|
|
if exclusions.is_empty() {
|
|
Self::None
|
|
} else {
|
|
Self::Some(exclusions)
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Returns true if the package is excluded and a local distribution should not be used.
|
|
pub fn contains(&self, package: &PackageName) -> bool {
|
|
match self {
|
|
Self::None => false,
|
|
Self::Some(packages) => packages.contains(package),
|
|
Self::All => true,
|
|
}
|
|
}
|
|
}
|