Add option to use uv in pyo3 projects (#550)

* Add option to use uv in pyo3 projects

* Fix CI

* Install uv

* Fix commands

* Fix commands

* Fix ci_run.sh

Author: sanders41

.github/workflows/testing.yml

@@ -223,6 +223,8 @@ jobs:
           curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
       - name: Cache Rust dependencies
         uses: Swatinem/[email protected]
+      - name: Install uv
+        run: curl -LsSf https://astral.sh/uv/install.sh | sh
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
@@ -234,11 +236,9 @@ jobs:
       - name: Install Dependencies
         working-directory: ${{ env.WORKING_DIR }}
         run: |
-          python -m pip install -U pip
-          python -m pip install -r requirements-dev.txt
-          python -m pip install -e .
-          maturin build --out dist
-          python -m pip install --no-index --find-links=dist/ my-project
+          uv lock
+          uv sync --frozen
+          uv run maturin build
       - name: MyPy
         working-directory: ${{ env.WORKING_DIR }}
         run: just mypy
@@ -247,7 +247,7 @@ jobs:
         run: just ruff-check
       - name: ruff format
         working-directory: ${{ env.WORKING_DIR }}
-        run: python -m ruff format --check .
+        run: uv run ruff format --check .
       - name: Generated Project Clippy
         working-directory: ${{ env.WORKING_DIR }}
         run: just clippy
@@ -268,6 +268,8 @@ jobs:
           curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
       - name: Cache Rust dependencies
         uses: Swatinem/[email protected]
+      - name: Install uv
+        run: curl -LsSf https://astral.sh/uv/install.sh | sh
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
@@ -279,20 +281,18 @@ jobs:
       - name: Install Dependencies
         working-directory: ${{ env.WORKING_DIR }}
         run: |
-          python -m pip install -U pip
-          python -m pip install -r requirements-dev.txt
-          python -m pip install -e .
-          maturin build --out dist
-          python -m pip install --no-index --find-links=dist/ my-project
+          uv lock
+          uv sync --frozen
+          uv run maturin build
       - name: Pre-commit check
         working-directory: ${{ env.WORKING_DIR }}
         run: |
-          pre-commit install
+          uv run pre-commit install
           git add .
-          pre-commit run --all-files
+          uv run pre-commit run --all-files
       - name: Test with pytest
         working-directory: ${{ env.WORKING_DIR }}
-        run: pytest
+        run: uv run pytest
   setuptools-linting:
     strategy:
       fail-fast: false

scripts/ci_run.sh

@@ -42,8 +42,10 @@ use_dependabot=""
 use_continuous_deployment=""
 use_release_drafter=""
 use_multi_os_ci=""
+pyo3_python_manager=""
 
-./target/release/python-project create -s << EOF
+if [[ project_manager -eq 3 ]]; then
+  ./target/release/python-project create -s << EOF
 $project_name
 $project_slug
 $source_dir
@@ -57,13 +59,37 @@ $python_version
 $min_python_version
 $gha_versions
 $project_manager
+$pyo3_python_manager
 $application
 $max_line_length
 $use_dependabot
 $use_continuous_deployment
 $use_release_drafter
 $use_multi_os_ci
 EOF
+else
+  ./target/release/python-project create -s << EOF
+$project_name
+$project_slug
+$source_dir
+$project_description
+$creator
+$creator_email
+$license
+$copyright_year
+$version
+$python_version
+$min_python_version
+$gha_versions
+$project_manager
+$application
+$max_line_length
+$use_dependabot
+$use_continuous_deployment
+$use_release_drafter
+$use_multi_os_ci
+EOF
+fi
 
 if [ ! -d $project_slug ]; then
   echo "Directory not created"

src/cli.rs

@@ -1,6 +1,8 @@
 use clap::{Parser, Subcommand, ValueEnum};
 
-use crate::project_info::{Day, DependabotSchedule, LicenseType, ProjectManager};
+use crate::project_info::{
+    Day, DependabotSchedule, LicenseType, ProjectManager, Pyo3PythonManager,
+};
 
 #[derive(Clone, Debug, ValueEnum)]
 pub enum ApplicationOrLib {
@@ -87,6 +89,12 @@ pub enum Param {
     /// Remove the saved project manager
     ResetProjectManager,
 
+    /// Save a default PyO3 python manager
+    Pyo3PythonManager { value: Pyo3PythonManager },
+
+    /// Remove the saved PyO3 Python manager
+    ResetPyo3PythonManager,
+
     /// Save a default value for is async project
     IsAsyncProject { value: BooleanChoice },
 

src/config.rs

@@ -11,6 +11,7 @@ use serde::{Deserialize, Serialize};
 
 use crate::project_info::{
     is_valid_python_version, Day, DependabotSchedule, LicenseType, ProjectManager,
+    Pyo3PythonManager,
 };
 
 #[derive(Debug, Deserialize, Serialize, PartialEq, Eq)]
@@ -21,6 +22,7 @@ pub struct Config {
     pub python_version: Option<String>,
     pub min_python_version: Option<String>,
     pub project_manager: Option<ProjectManager>,
+    pub pyo3_python_manager: Option<Pyo3PythonManager>,
     pub is_async_project: Option<bool>,
     pub is_application: Option<bool>,
     pub github_actions_python_test_versions: Option<Vec<String>>,
@@ -51,6 +53,7 @@ impl Default for Config {
             python_version: None,
             min_python_version: None,
             project_manager: None,
+            pyo3_python_manager: None,
             is_async_project: None,
             is_application: None,
             github_actions_python_test_versions: None,
@@ -84,6 +87,7 @@ impl Config {
                             python_version: config.python_version,
                             min_python_version: config.min_python_version,
                             project_manager: config.project_manager,
+                            pyo3_python_manager: config.pyo3_python_manager,
                             is_async_project: config.is_async_project,
                             is_application: config.is_application,
                             github_actions_python_test_versions: config
@@ -202,6 +206,16 @@ impl Config {
         Ok(())
     }
 
+    pub fn save_pyo3_python_manager(&self, value: Pyo3PythonManager) -> Result<()> {
+        self.handle_save_config(|config| &mut config.pyo3_python_manager, Some(value))?;
+        Ok(())
+    }
+
+    pub fn reset_pyo3_python_manager(&self) -> Result<()> {
+        self.handle_save_config(|config| &mut config.pyo3_python_manager, None)?;
+        Ok(())
+    }
+
     pub fn save_is_async_project(&self, value: bool) -> Result<()> {
         self.handle_save_config(|config| &mut config.is_async_project, Some(value))?;
         Ok(())
@@ -408,6 +422,7 @@ impl Config {
         }
 
         print_config_value("Project Manager", &config.project_manager);
+        print_config_value("PyO3 Python Manager", &config.pyo3_python_manager);
         print_config_value("Async Project", &config.is_async_project);
         print_config_value("Max Line Length", &config.max_line_length);
         print_config_value("Use Dependabot", &config.use_dependabot);
@@ -659,6 +674,28 @@ mod tests {
         assert_eq!(result.project_manager, None);
     }
 
+    #[test]
+    fn test_save_pyo3_python_manger() {
+        let config = mock_config();
+        let expected = Pyo3PythonManager::Uv;
+        config.save_pyo3_python_manager(expected.clone()).unwrap();
+        let result = config.load_config();
+
+        assert_eq!(result.pyo3_python_manager, Some(expected));
+    }
+
+    #[test]
+    fn test_reset_pyo3_project_manager() {
+        let config = mock_config();
+        config
+            .save_pyo3_python_manager(Pyo3PythonManager::Uv)
+            .unwrap();
+        config.reset_pyo3_python_manager().unwrap();
+        let result = config.load_config();
+
+        assert_eq!(result.pyo3_python_manager, None);
+    }
+
     #[test]
     fn test_save_is_async_project() {
         let config = mock_config();