Add uv export support for PEP 751

Author: charliermarsh

crates/uv-configuration/src/export_format.rs

@@ -11,4 +11,8 @@ pub enum ExportFormat {
         clap(name = "requirements.txt", alias = "requirements-txt")
     )]
     RequirementsTxt,
+    /// Export in `pylock.toml` format.
+    #[serde(rename = "pylock.toml", alias = "pylock-toml")]
+    #[cfg_attr(feature = "clap", clap(name = "pylock.toml", alias = "pylock-toml"))]
+    PylockToml,
 }

crates/uv-pypi-types/src/simple_json.rs

@@ -185,11 +185,15 @@ impl Default for Yanked {
 /// A dictionary mapping a hash name to a hex encoded digest of the file.
 ///
 /// PEP 691 says multiple hashes can be included and the interpretation is left to the client.
-#[derive(Debug, Clone, Eq, PartialEq, Default, Deserialize)]
+#[derive(Debug, Clone, Eq, PartialEq, Default, Deserialize, Serialize)]
 pub struct Hashes {
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub md5: Option<SmallString>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub sha256: Option<SmallString>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub sha384: Option<SmallString>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub sha512: Option<SmallString>,
 }
 
@@ -490,6 +494,21 @@ impl From<Hashes> for HashDigests {
     }
 }
 
+impl From<HashDigests> for Hashes {
+    fn from(value: HashDigests) -> Self {
+        let mut hashes = Hashes::default();
+        for digest in value.into_iter() {
+            match digest.algorithm() {
+                HashAlgorithm::Md5 => hashes.md5 = Some(digest.digest),
+                HashAlgorithm::Sha256 => hashes.sha256 = Some(digest.digest),
+                HashAlgorithm::Sha384 => hashes.sha384 = Some(digest.digest),
+                HashAlgorithm::Sha512 => hashes.sha512 = Some(digest.digest),
+            }
+        }
+        hashes
+    }
+}
+
 impl From<HashDigest> for HashDigests {
     fn from(value: HashDigest) -> Self {
         Self(Box::new([value]))

crates/uv-resolver/src/lib.rs

@@ -5,8 +5,8 @@ pub use exclusions::Exclusions;
 pub use flat_index::{FlatDistributions, FlatIndex};
 pub use fork_strategy::ForkStrategy;
 pub use lock::{
-    Installable, Lock, LockError, LockVersion, Package, PackageMap, RequirementsTxtExport,
-    ResolverManifest, SatisfiesResult, TreeDisplay, VERSION,
+    Installable, Lock, LockError, LockVersion, Package, PackageMap, PylockToml,
+    RequirementsTxtExport, ResolverManifest, SatisfiesResult, TreeDisplay, VERSION,
 };
 pub use manifest::Manifest;
 pub use options::{Flexibility, Options, OptionsBuilder};

crates/uv-resolver/src/lock/export/mod.rs

@@ -14,10 +14,12 @@ use uv_pep508::MarkerTree;
 use uv_pypi_types::ConflictItem;
 
 use crate::graph_ops::{marker_reachability, Reachable};
+pub use crate::lock::export::pylock_toml::PylockToml;
 pub use crate::lock::export::requirements_txt::RequirementsTxtExport;
 use crate::universal_marker::resolve_conflicts;
 use crate::{Installable, Package};
 
+mod pylock_toml;
 mod requirements_txt;
 
 /// A flat requirement, with its associated marker.