diff --git a/crates/uv-cli/src/lib.rs b/crates/uv-cli/src/lib.rs index 64276796e..dc4365990 100644 --- a/crates/uv-cli/src/lib.rs +++ b/crates/uv-cli/src/lib.rs @@ -2611,6 +2611,20 @@ pub struct RunArgs { #[arg(long, overrides_with("dev"))] pub no_dev: bool, + /// Include dependencies from the specified local dependency group. + /// + /// May be provided multiple times. + #[arg(long, conflicts_with("only_group"))] + pub group: Vec, + + /// Only include dependencies from the specified local dependency group. + /// + /// May be provided multiple times. + /// + /// The project itself will also be omitted. + #[arg(long, conflicts_with("group"))] + pub only_group: Vec, + /// Run a Python module. /// /// Equivalent to `python -m `. diff --git a/crates/uv-configuration/src/dev.rs b/crates/uv-configuration/src/dev.rs index 8b14d3be1..c482ec404 100644 --- a/crates/uv-configuration/src/dev.rs +++ b/crates/uv-configuration/src/dev.rs @@ -1,3 +1,5 @@ +use std::borrow::Cow; + use either::Either; use uv_normalize::{GroupName, DEV_DEPENDENCIES}; @@ -25,6 +27,15 @@ impl DevMode { pub fn prod(&self) -> bool { matches!(self, Self::Exclude | Self::Include) } + + /// Returns the flag that was used to request development dependencies. + pub fn as_flag(&self) -> &'static str { + match self { + Self::Exclude => "--no-dev", + Self::Include => "--dev", + Self::Only => "--only-dev", + } + } } #[derive(Debug, Clone)] @@ -63,6 +74,23 @@ impl GroupsSpecification { pub fn prod(&self) -> bool { matches!(self, Self::Exclude | Self::Include(_)) } + + /// Returns the option that was used to request the groups, if any. + pub fn as_flag(&self) -> Option> { + match self { + Self::Exclude => None, + Self::Include(groups) => match groups.as_slice() { + [] => None, + [group] => Some(Cow::Owned(format!("--group {group}"))), + [..] => Some(Cow::Borrowed("--group")), + }, + Self::Only(groups) => match groups.as_slice() { + [] => None, + [group] => Some(Cow::Owned(format!("--only-group {group}"))), + [..] => Some(Cow::Borrowed("--only-group")), + }, + } + } } impl DevGroupsSpecification { @@ -138,6 +166,10 @@ impl DevGroupsSpecification { pub fn dev_mode(&self) -> Option<&DevMode> { self.dev.as_ref() } + + pub fn groups(&self) -> &GroupsSpecification { + &self.groups + } } impl From for DevGroupsSpecification { diff --git a/crates/uv/src/commands/project/run.rs b/crates/uv/src/commands/project/run.rs index 38a0e4947..f7a2041b0 100644 --- a/crates/uv/src/commands/project/run.rs +++ b/crates/uv/src/commands/project/run.rs @@ -17,8 +17,8 @@ use uv_cache::Cache; use uv_cli::ExternalCommand; use uv_client::{BaseClientBuilder, Connectivity}; use uv_configuration::{ - Concurrency, DevGroupsSpecification, DevMode, EditableMode, ExtrasSpecification, - InstallOptions, LowerBound, SourceStrategy, + Concurrency, DevGroupsSpecification, EditableMode, ExtrasSpecification, InstallOptions, + LowerBound, SourceStrategy, }; use uv_distribution::LoweredRequirement; use uv_fs::which::is_executable; @@ -336,11 +336,14 @@ pub(crate) async fn run( if !extras.is_empty() { warn_user!("Extras are not supported for Python scripts with inline metadata"); } - if matches!(dev.dev_mode(), Some(DevMode::Exclude)) { - warn_user!("`--no-dev` is not supported for Python scripts with inline metadata"); + if let Some(dev_mode) = dev.dev_mode() { + warn_user!( + "`{}` is not supported for Python scripts with inline metadata", + dev_mode.as_flag() + ); } - if matches!(dev.dev_mode(), Some(DevMode::Only)) { - warn_user!("`--only-dev` is not supported for Python scripts with inline metadata"); + if let Some(flag) = dev.groups().as_flag() { + warn_user!("`{flag}` is not supported for Python scripts with inline metadata"); } if package.is_some() { warn_user!( @@ -413,11 +416,14 @@ pub(crate) async fn run( if !extras.is_empty() { warn_user!("Extras have no effect when used alongside `--no-project`"); } - if matches!(dev.dev_mode(), Some(DevMode::Exclude)) { - warn_user!("`--no-dev` has no effect when used alongside `--no-project`"); + if let Some(dev_mode) = dev.dev_mode() { + warn_user!( + "`{}` has no effect when used alongside `--no-project`", + dev_mode.as_flag() + ); } - if matches!(dev.dev_mode(), Some(DevMode::Only)) { - warn_user!("`--only-dev` has no effect when used alongside `--no-project`"); + if let Some(flag) = dev.groups().as_flag() { + warn_user!("`{flag}` has no effect when used alongside `--no-project`"); } if locked { warn_user!("`--locked` has no effect when used alongside `--no-project`"); @@ -433,11 +439,14 @@ pub(crate) async fn run( if !extras.is_empty() { warn_user!("Extras have no effect when used outside of a project"); } - if matches!(dev.dev_mode(), Some(DevMode::Exclude)) { - warn_user!("`--no-dev` has no effect when used outside of a project"); + if let Some(dev_mode) = dev.dev_mode() { + warn_user!( + "`{}` has no effect when used outside of a project", + dev_mode.as_flag() + ); } - if matches!(dev.dev_mode(), Some(DevMode::Only)) { - warn_user!("`--only-dev` has no effect when used outside of a project"); + if let Some(flag) = dev.groups().as_flag() { + warn_user!("`{flag}` has no effect when used outside of a project"); } if locked { warn_user!("`--locked` has no effect when used outside of a project"); diff --git a/crates/uv/src/settings.rs b/crates/uv/src/settings.rs index 104c2f97a..c0e4ca408 100644 --- a/crates/uv/src/settings.rs +++ b/crates/uv/src/settings.rs @@ -252,6 +252,8 @@ impl RunSettings { no_all_extras, dev, no_dev, + group, + only_group, module: _, only_dev, no_editable, @@ -280,8 +282,7 @@ impl RunSettings { flag(all_extras, no_all_extras).unwrap_or_default(), extra.unwrap_or_default(), ), - // TODO(zanieb): Support `--group` here - dev: DevGroupsSpecification::from_args(dev, no_dev, only_dev, vec![], vec![]), + dev: DevGroupsSpecification::from_args(dev, no_dev, only_dev, group, only_group), editable: EditableMode::from_args(no_editable), with, with_editable, diff --git a/crates/uv/tests/it/run.rs b/crates/uv/tests/it/run.rs index 76f7a2c26..8d1e4478f 100644 --- a/crates/uv/tests/it/run.rs +++ b/crates/uv/tests/it/run.rs @@ -416,6 +416,30 @@ fn run_pep723_script() -> Result<()> { "# })?; + // Running a script with `--group` should warn. + uv_snapshot!(context.filters(), context.run().arg("--group").arg("foo").arg("main.py"), @r###" + success: false + exit_code: 1 + ----- stdout ----- + + ----- stderr ----- + Reading inline script metadata from `main.py` + × No solution found when resolving script dependencies: + ╰─▶ Because there are no versions of add and you require add, we can conclude that your requirements are unsatisfiable. + "###); + + // If the script can't be resolved, we should reference the script. + let test_script = context.temp_dir.child("main.py"); + test_script.write_str(indoc! { r#" + # /// script + # requires-python = ">=3.11" + # dependencies = [ + # "add", + # ] + # /// + "# + })?; + uv_snapshot!(context.filters(), context.run().arg("--no-project").arg("main.py"), @r###" success: false exit_code: 1 @@ -925,6 +949,161 @@ fn run_with_editable() -> Result<()> { Ok(()) } +#[test] +fn run_group() -> Result<()> { + let context = TestContext::new("3.12"); + + let pyproject_toml = context.temp_dir.child("pyproject.toml"); + pyproject_toml.write_str( + r#" + [project] + name = "project" + version = "0.1.0" + requires-python = ">=3.12" + dependencies = ["typing-extensions"] + + [dependency-groups] + foo = ["anyio"] + bar = ["iniconfig"] + dev = ["sniffio"] + "#, + )?; + + let test_script = context.temp_dir.child("main.py"); + test_script.write_str(indoc! { r#" + try: + import anyio + print("imported `anyio`") + except ImportError: + print("failed to import `anyio`") + + try: + import iniconfig + print("imported `iniconfig`") + except ImportError: + print("failed to import `iniconfig`") + + try: + import typing_extensions + print("imported `typing_extensions`") + except ImportError: + print("failed to import `typing_extensions`") + "# + })?; + + context.lock().assert().success(); + + uv_snapshot!(context.filters(), context.run().arg("main.py"), @r###" + success: true + exit_code: 0 + ----- stdout ----- + failed to import `anyio` + failed to import `iniconfig` + imported `typing_extensions` + + ----- stderr ----- + Resolved 6 packages in [TIME] + Prepared 2 packages in [TIME] + Installed 2 packages in [TIME] + + sniffio==1.3.1 + + typing-extensions==4.10.0 + "###); + + uv_snapshot!(context.filters(), context.run().arg("--only-group").arg("bar").arg("main.py"), @r###" + success: true + exit_code: 0 + ----- stdout ----- + failed to import `anyio` + imported `iniconfig` + imported `typing_extensions` + + ----- stderr ----- + Resolved 6 packages in [TIME] + Prepared 1 package in [TIME] + Installed 1 package in [TIME] + + iniconfig==2.0.0 + "###); + + uv_snapshot!(context.filters(), context.run().arg("--group").arg("foo").arg("main.py"), @r###" + success: true + exit_code: 0 + ----- stdout ----- + imported `anyio` + imported `iniconfig` + imported `typing_extensions` + + ----- stderr ----- + Resolved 6 packages in [TIME] + Prepared 2 packages in [TIME] + Installed 2 packages in [TIME] + + anyio==4.3.0 + + idna==3.6 + "###); + + uv_snapshot!(context.filters(), context.run().arg("--group").arg("foo").arg("--group").arg("bar").arg("main.py"), @r###" + success: true + exit_code: 0 + ----- stdout ----- + imported `anyio` + imported `iniconfig` + imported `typing_extensions` + + ----- stderr ----- + Resolved 6 packages in [TIME] + Audited 5 packages in [TIME] + "###); + + uv_snapshot!(context.filters(), context.run().arg("--group").arg("foo").arg("--no-project").arg("main.py"), @r###" + success: true + exit_code: 0 + ----- stdout ----- + imported `anyio` + imported `iniconfig` + imported `typing_extensions` + + ----- stderr ----- + warning: `--group foo` has no effect when used alongside `--no-project` + "###); + + uv_snapshot!(context.filters(), context.run().arg("--group").arg("foo").arg("--group").arg("bar").arg("--no-project").arg("main.py"), @r###" + success: true + exit_code: 0 + ----- stdout ----- + imported `anyio` + imported `iniconfig` + imported `typing_extensions` + + ----- stderr ----- + warning: `--group` has no effect when used alongside `--no-project` + "###); + + uv_snapshot!(context.filters(), context.run().arg("--group").arg("dev").arg("--no-project").arg("main.py"), @r###" + success: true + exit_code: 0 + ----- stdout ----- + imported `anyio` + imported `iniconfig` + imported `typing_extensions` + + ----- stderr ----- + warning: `--group dev` has no effect when used alongside `--no-project` + "###); + + uv_snapshot!(context.filters(), context.run().arg("--dev").arg("--no-project").arg("main.py"), @r###" + success: true + exit_code: 0 + ----- stdout ----- + imported `anyio` + imported `iniconfig` + imported `typing_extensions` + + ----- stderr ----- + warning: `--dev` has no effect when used alongside `--no-project` + "###); + + Ok(()) +} + #[test] fn run_locked() -> Result<()> { let context = TestContext::new("3.12"); diff --git a/docs/reference/cli.md b/docs/reference/cli.md index 34a83c5ac..d6a6dbc9c 100644 --- a/docs/reference/cli.md +++ b/docs/reference/cli.md @@ -163,6 +163,10 @@ uv run [OPTIONS] [COMMAND]

Instead of checking if the lockfile is up-to-date, uses the versions in the lockfile as the source of truth. If the lockfile is missing, uv will exit with an error. If the pyproject.toml includes changes to dependencies that have not been included in the lockfile yet, they will not be present in the environment.

May also be set with the UV_FROZEN environment variable.

+
--group group

Include dependencies from the specified local dependency group.

+ +

May be provided multiple times.

+
--help, -h

Display the concise help for this command

--index index

The URLs to use when resolving dependencies, in addition to the default index.

@@ -311,6 +315,12 @@ uv run [OPTIONS] [COMMAND]

The project itself will also be omitted.

+
--only-group only-group

Only include dependencies from the specified local dependency group.

+ +

May be provided multiple times.

+ +

The project itself will also be omitted.

+
--package package

Run the command in a specific package in the workspace.

If the workspace member does not exist, uv will exit with an error.