Exclude distributions from lock

Author: charliermarsh

crates/uv-distribution-types/src/prioritized_distribution.rs

@@ -371,6 +371,10 @@ impl PrioritizedDist {
                 self.0.markers.or(implied_markers(&dist.filename));
             }
         }
+        // Track the hashes.
+        if !compatibility.is_excluded() {
+            self.0.hashes.extend(hashes);
+        }
         // Track the highest-priority wheel.
         if let Some((.., existing_compatibility)) = self.best_wheel() {
             if compatibility.is_more_compatible(existing_compatibility) {
@@ -379,7 +383,6 @@ impl PrioritizedDist {
         } else {
             self.0.best_wheel_index = Some(self.0.wheels.len());
         }
-        self.0.hashes.extend(hashes);
         self.0.wheels.push((dist, compatibility));
     }
 
@@ -394,6 +397,10 @@ impl PrioritizedDist {
         if compatibility.is_compatible() {
             self.0.markers = MarkerTree::TRUE;
         }
+        // Track the hashes.
+        if !compatibility.is_excluded() {
+            self.0.hashes.extend(hashes);
+        }
         // Track the highest-priority source.
         if let Some((.., existing_compatibility)) = &self.0.source {
             if compatibility.is_more_compatible(existing_compatibility) {
@@ -402,7 +409,6 @@ impl PrioritizedDist {
         } else {
             self.0.source = Some((dist, compatibility));
         }
-        self.0.hashes.extend(hashes);
     }
 
     /// Return the highest-priority distribution for the package version, if any.
@@ -501,6 +507,7 @@ impl PrioritizedDist {
             .0
             .wheels
             .iter()
+            .filter(|(_, compatibility)| !compatibility.is_excluded())
             .map(|(wheel, _)| wheel.clone())
             .collect();
         let sdist = self.0.source.as_ref().map(|(sdist, _)| sdist.clone());
@@ -514,7 +521,12 @@ impl PrioritizedDist {
     /// If this prioritized dist has an sdist, then this creates a source
     /// distribution.
     pub fn source_dist(&self) -> Option<RegistrySourceDist> {
-        let mut sdist = self.0.source.as_ref().map(|(sdist, _)| sdist.clone())?;
+        let mut sdist = self
+            .0
+            .source
+            .as_ref()
+            .filter(|(_, compatibility)| !compatibility.is_excluded())
+            .map(|(sdist, _)| sdist.clone())?;
         assert!(
             sdist.wheels.is_empty(),
             "source distribution should not have any wheels yet"
@@ -623,6 +635,11 @@ impl WheelCompatibility {
         matches!(self, Self::Compatible(_, _, _))
     }
 
+    /// Return `true` if the distribution is excluded.
+    pub fn is_excluded(&self) -> bool {
+        matches!(self, Self::Incompatible(IncompatibleWheel::ExcludeNewer(_)))
+    }
+
     /// Return `true` if the current compatibility is more compatible than another.
     ///
     /// Compatible wheels are always higher more compatible than incompatible wheels.
@@ -650,6 +667,14 @@ impl SourceDistCompatibility {
         matches!(self, Self::Compatible(_))
     }
 
+    /// Return `true` if the distribution is excluded.
+    pub fn is_excluded(&self) -> bool {
+        matches!(
+            self,
+            Self::Incompatible(IncompatibleSource::ExcludeNewer(_))
+        )
+    }
+
     /// Return the higher priority compatibility.
     ///
     /// Compatible source distributions are always higher priority than incompatible source distributions.