mirror of https://github.com/astral-sh/uv
33 lines
1.1 KiB
Rust
33 lines
1.1 KiB
Rust
use rustc_hash::FxHashMap;
|
|
|
|
use distribution_types::{CompatibleDist, ResolvedDist};
|
|
use uv_normalize::PackageName;
|
|
|
|
use crate::candidate_selector::Candidate;
|
|
|
|
/// A set of package versions pinned to specific files.
|
|
///
|
|
/// For example, given `Flask==3.0.0`, the [`FilePins`] would contain a mapping from `Flask` to
|
|
/// `3.0.0` to the specific wheel or source distribution archive that was pinned for that version.
|
|
#[derive(Debug, Default)]
|
|
pub(crate) struct FilePins(FxHashMap<PackageName, FxHashMap<pep440_rs::Version, ResolvedDist>>);
|
|
|
|
impl FilePins {
|
|
/// Pin a candidate package.
|
|
pub(crate) fn insert(&mut self, candidate: &Candidate, dist: &CompatibleDist) {
|
|
self.0.entry(candidate.name().clone()).or_default().insert(
|
|
candidate.version().clone(),
|
|
dist.for_installation().to_owned(),
|
|
);
|
|
}
|
|
|
|
/// Return the pinned file for the given package name and version, if it exists.
|
|
pub(crate) fn get(
|
|
&self,
|
|
name: &PackageName,
|
|
version: &pep440_rs::Version,
|
|
) -> Option<&ResolvedDist> {
|
|
self.0.get(name)?.get(version)
|
|
}
|
|
}
|