Python
/
uv
mirror of https://github.com/astral-sh/uv
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Detect current venv when `uv` is invoked from within a virtualenv (#3379) 

Fixes #3378.
This commit is contained in:
Godefroid Chapelle 2024-05-06 00:53:25 +02:00 committed by GitHub
parent 6d73db2b1e
commit ef92c38486
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
crates/uv-interpreter/src/virtualenv.rs
View File

@ -73,10 +73,14 @@ pub(crate) fn virtualenv_from_env() -> Option<PathBuf> {
/// Locate a virtual environment by searching the file system. /// Locate a virtual environment by searching the file system.
/// ///
/// Finds a `.venv` directory in the current or any parent directory. /// Checks if the working directory is a virtual environment.
///
/// If not, finds a `.venv` directory in the current or any parent directory.
pub(crate) fn virtualenv_from_working_dir() -> Result<Option<PathBuf>, Error> { pub(crate) fn virtualenv_from_working_dir() -> Result<Option<PathBuf>, Error> {
let current_dir = env::current_dir().expect("Failed to detect current directory"); let current_dir = env::current_dir().expect("Failed to detect current directory");
for dir in current_dir.ancestors() { for dir in current_dir.ancestors() {
// Search for a `.venv` directory.
let dot_venv = dir.join(".venv"); let dot_venv = dir.join(".venv");
if dot_venv.is_dir() { if dot_venv.is_dir() {
if !dot_venv.join("pyvenv.cfg").is_file() { if !dot_venv.join("pyvenv.cfg").is_file() {
@ -85,6 +89,12 @@ pub(crate) fn virtualenv_from_working_dir() -> Result<Option<PathBuf>, Error> {
debug!("Found a virtualenv named .venv at: {}", dot_venv.display()); debug!("Found a virtualenv named .venv at: {}", dot_venv.display());
return Ok(Some(dot_venv)); return Ok(Some(dot_venv));
} }
// Otherwise, if we're _within_ a virtualenv, return it.
if dir.join("pyvenv.cfg").is_file() {
debug!("Found a virtualenv at: {}", dir.display());
return Ok(Some(dir.to_path_buf()));
}
} }
Ok(None) Ok(None)