Use cargo to install pyo3

Author: sanders41

src/main.rs

@@ -27,7 +27,8 @@ use crate::{
     config::Config,
     dev_dependency_installer::install_dev_dependencies,
     project_generator::generate_project,
-    project_info::{get_project_info, ProjectInfo},
+    project_info::{get_project_info, ProjectInfo, ProjectManager},
+    rust_files::cargo_add_pyo3,
 };
 
 #[cfg(feature = "fastapi")]
@@ -42,6 +43,10 @@ fn create(project_info: &ProjectInfo) -> Result<()> {
         .output()
         .expect("Failed to initialize git");
 
+    if matches!(project_info.project_manager, ProjectManager::Maturin) {
+        cargo_add_pyo3(project_info)?;
+    }
+
     install_dev_dependencies(project_info)?;
 
     #[cfg(feature = "fastapi")]

src/package_version.rs

@@ -115,33 +115,6 @@ impl PreCommitHookVersion {
     }
 }
 
-#[derive(Debug)]
-pub struct RustPackageVersion {
-    pub name: String,
-    pub version: String,
-    pub features: Option<Vec<String>>,
-}
-
-impl LatestVersion for RustPackageVersion {
-    fn get_latest_version(&mut self) -> Result<()> {
-        let url = format!("https://crates.io/api/v1/crates/{}", self.name);
-        let client = reqwest::blocking::Client::new();
-        let response = client
-            .get(url)
-            .header(reqwest::header::USER_AGENT, "python-project-generator")
-            .timeout(Duration::new(5, 0))
-            .send()?
-            .text()?;
-        let info: serde_json::Value = serde_json::from_str(&response)?;
-        self.name = info["crate"]["id"].to_string().replace('"', "");
-        self.version = info["crate"]["max_stable_version"]
-            .to_string()
-            .replace('"', "");
-
-        Ok(())
-    }
-}
-
 pub fn default_pre_commit_rev(hook: &PreCommitHook) -> String {
     match hook {
         PreCommitHook::MyPy => "v1.18.2".to_string(),

src/rust_files.rs

@@ -1,64 +1,17 @@
 use anyhow::Result;
-use colored::*;
-use rayon::prelude::*;
 
 use crate::{
     file_manager::save_file_with_content,
     licenses::license_str,
-    package_version::{LatestVersion, RustPackageVersion},
     project_info::{LicenseType, ProjectInfo},
 };
 
-fn build_latest_dependencies() -> String {
-    let mut version_string = String::new();
-    let mut packages = vec![RustPackageVersion {
-        name: "pyo3".to_string(),
-        version: "0.27.1".to_string(),
-        features: Some(vec!["extension-module".to_string()]),
-    }];
-
-    packages.par_iter_mut().for_each(|package| {
-        if package.get_latest_version().is_err() {
-            let error_message = format!(
-                "Error retrieving latest crate version for {}. Using default.",
-                package.name
-            );
-            println!("\n{}", error_message.yellow());
-        }
-    });
-
-    for package in packages {
-        if let Some(features) = &package.features {
-            let mut feature_str = "[".to_string();
-            for feature in features {
-                feature_str.push_str(&format!(r#""{feature}", "#));
-            }
-
-            feature_str.truncate(feature_str.len() - 2);
-            feature_str.push(']');
-
-            version_string.push_str(&format!(
-                "{} = {{ version = \"{}\", features = {} }}\n",
-                package.name, package.version, feature_str
-            ));
-        } else {
-            version_string.push_str(&format!(
-                "{} = {{ version = \"{}\" }}\n",
-                package.name, package.version
-            ));
-        }
-    }
-
-    version_string.trim().to_string()
-}
-
 fn create_cargo_toml_file(
     project_slug: &str,
     project_description: &str,
     source_dir: &str,
     license_type: &LicenseType,
 ) -> String {
-    let versions = build_latest_dependencies();
     let license = license_str(license_type);
     let name = source_dir.replace([' ', '-'], "_");
 
@@ -76,7 +29,6 @@ name = "_{name}"
 crate-type = ["cdylib"]
 
 [dependencies]
-{versions}
 "#
     )
 }
@@ -95,6 +47,22 @@ pub fn save_cargo_toml_file(project_info: &ProjectInfo) -> Result<()> {
     Ok(())
 }
 
+pub fn cargo_add_pyo3(project_info: &ProjectInfo) -> Result<()> {
+    use anyhow::bail;
+
+    let output = std::process::Command::new("cargo")
+        .args(["add", "pyo3", "--features", "extension-module"])
+        .current_dir(project_info.base_dir())
+        .output()?;
+
+    if !output.status.success() {
+        let stderr = String::from_utf8_lossy(&output.stderr);
+        bail!("Failed to add pyo3 dependency: {stderr}");
+    }
+
+    Ok(())
+}
+
 fn create_lib_file(project_info: &ProjectInfo) -> String {
     let module = project_info.module_name();
     format!(

src/snapshots/python_project__rust_files__tests__save_cargo_toml_file.snap

@@ -1,6 +1,5 @@
 ---
 source: src/rust_files.rs
 expression: content
-snapshot_kind: text
 ---
-"[package]\nname = \"my-project\"\nversion = \"1.0.0\"\ndescription = \"This is a test\"\nedition = \"2021\"\nlicense = \"MIT\"\nreadme = \"README.md\"\n\n[lib]\nname = \"_my_project\"\ncrate-type = [\"cdylib\"]\n\n[dependencies]\npyo3 = { version = \"1.0.0\", features = [\"extension-module\"] }\n"
+"[package]\nname = \"my-project\"\nversion = \"1.0.0\"\ndescription = \"This is a test\"\nedition = \"2021\"\nlicense = \"MIT\"\nreadme = \"README.md\"\n\n[lib]\nname = \"_my_project\"\ncrate-type = [\"cdylib\"]\n\n[dependencies]\n"