|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231 |
- <#
- .Synopsis
- Activate a Python virtual environment for the current Powershell session.
-
- .Description
- Pushes the python executable for a virtual environment to the front of the
- $Env:PATH environment variable and sets the prompt to signify that you are
- in a Python virtual environment. Makes use of the command line switches as
- well as the `pyvenv.cfg` file values present in the virtual environment.
-
- .Parameter VenvDir
- Path to the directory that contains the virtual environment to activate. The
- default value for this is the parent of the directory that the Activate.ps1
- script is located within.
-
- .Parameter Prompt
- The prompt prefix to display when this virtual environment is activated. By
- default, this prompt is the name of the virtual environment folder (VenvDir)
- surrounded by parentheses and followed by a single space (ie. '(.venv) ').
-
- .Example
- Activate.ps1
- Activates the Python virtual environment that contains the Activate.ps1 script.
-
- .Example
- Activate.ps1 -Verbose
- Activates the Python virtual environment that contains the Activate.ps1 script,
- and shows extra information about the activation as it executes.
-
- .Example
- Activate.ps1 -VenvDir C:\Users\MyUser\Common\.venv
- Activates the Python virtual environment located in the specified location.
-
- .Example
- Activate.ps1 -Prompt "MyPython"
- Activates the Python virtual environment that contains the Activate.ps1 script,
- and prefixes the current prompt with the specified string (surrounded in
- parentheses) while the virtual environment is active.
-
-
- #>
- Param(
- [Parameter(Mandatory = $false)]
- [String]
- $VenvDir,
- [Parameter(Mandatory = $false)]
- [String]
- $Prompt
- )
-
- <# Function declarations --------------------------------------------------- #>
-
- <#
- .Synopsis
- Remove all shell session elements added by the Activate script, including the
- addition of the virtual environment's Python executable from the beginning of
- the PATH variable.
-
- .Parameter NonDestructive
- If present, do not remove this function from the global namespace for the
- session.
-
- #>
- function global:deactivate ([switch]$NonDestructive) {
- # Revert to original values
-
- # The prior prompt:
- if (Test-Path -Path Function:_OLD_VIRTUAL_PROMPT) {
- Copy-Item -Path Function:_OLD_VIRTUAL_PROMPT -Destination Function:prompt
- Remove-Item -Path Function:_OLD_VIRTUAL_PROMPT
- }
-
- # The prior PYTHONHOME:
- if (Test-Path -Path Env:_OLD_VIRTUAL_PYTHONHOME) {
- Copy-Item -Path Env:_OLD_VIRTUAL_PYTHONHOME -Destination Env:PYTHONHOME
- Remove-Item -Path Env:_OLD_VIRTUAL_PYTHONHOME
- }
-
- # The prior PATH:
- if (Test-Path -Path Env:_OLD_VIRTUAL_PATH) {
- Copy-Item -Path Env:_OLD_VIRTUAL_PATH -Destination Env:PATH
- Remove-Item -Path Env:_OLD_VIRTUAL_PATH
- }
-
- # Just remove the VIRTUAL_ENV altogether:
- if (Test-Path -Path Env:VIRTUAL_ENV) {
- Remove-Item -Path env:VIRTUAL_ENV
- }
-
- # Just remove the _PYTHON_VENV_PROMPT_PREFIX altogether:
- if (Get-Variable -Name "_PYTHON_VENV_PROMPT_PREFIX" -ErrorAction SilentlyContinue) {
- Remove-Variable -Name _PYTHON_VENV_PROMPT_PREFIX -Scope Global -Force
- }
-
- # Leave deactivate function in the global namespace if requested:
- if (-not $NonDestructive) {
- Remove-Item -Path function:deactivate
- }
- }
-
- <#
- .Description
- Get-PyVenvConfig parses the values from the pyvenv.cfg file located in the
- given folder, and returns them in a map.
-
- For each line in the pyvenv.cfg file, if that line can be parsed into exactly
- two strings separated by `=` (with any amount of whitespace surrounding the =)
- then it is considered a `key = value` line. The left hand string is the key,
- the right hand is the value.
-
- If the value starts with a `'` or a `"` then the first and last character is
- stripped from the value before being captured.
-
- .Parameter ConfigDir
- Path to the directory that contains the `pyvenv.cfg` file.
- #>
- function Get-PyVenvConfig(
- [String]
- $ConfigDir
- ) {
- Write-Verbose "Given ConfigDir=$ConfigDir, obtain values in pyvenv.cfg"
-
- # Ensure the file exists, and issue a warning if it doesn't (but still allow the function to continue).
- $pyvenvConfigPath = Join-Path -Resolve -Path $ConfigDir -ChildPath 'pyvenv.cfg' -ErrorAction Continue
-
- # An empty map will be returned if no config file is found.
- $pyvenvConfig = @{ }
-
- if ($pyvenvConfigPath) {
-
- Write-Verbose "File exists, parse `key = value` lines"
- $pyvenvConfigContent = Get-Content -Path $pyvenvConfigPath
-
- $pyvenvConfigContent | ForEach-Object {
- $keyval = $PSItem -split "\s*=\s*", 2
- if ($keyval[0] -and $keyval[1]) {
- $val = $keyval[1]
-
- # Remove extraneous quotations around a string value.
- if ("'""".Contains($val.Substring(0,1))) {
- $val = $val.Substring(1, $val.Length - 2)
- }
-
- $pyvenvConfig[$keyval[0]] = $val
- Write-Verbose "Adding Key: '$($keyval[0])'='$val'"
- }
- }
- }
- return $pyvenvConfig
- }
-
-
- <# Begin Activate script --------------------------------------------------- #>
-
- # Determine the containing directory of this script
- $VenvExecPath = Split-Path -Parent $MyInvocation.MyCommand.Definition
- $VenvExecDir = Get-Item -Path $VenvExecPath
-
- Write-Verbose "Activation script is located in path: '$VenvExecPath'"
- Write-Verbose "VenvExecDir Fullname: '$($VenvExecDir.FullName)"
- Write-Verbose "VenvExecDir Name: '$($VenvExecDir.Name)"
-
- # Set values required in priority: CmdLine, ConfigFile, Default
- # First, get the location of the virtual environment, it might not be
- # VenvExecDir if specified on the command line.
- if ($VenvDir) {
- Write-Verbose "VenvDir given as parameter, using '$VenvDir' to determine values"
- } else {
- Write-Verbose "VenvDir not given as a parameter, using parent directory name as VenvDir."
- $VenvDir = $VenvExecDir.Parent.FullName.TrimEnd("\\/")
- $VenvDir = $VenvDir.Insert($VenvDir.Length, "/")
- Write-Verbose "VenvDir=$VenvDir"
- }
-
- # Next, read the `pyvenv.cfg` file to determine any required value such
- # as `prompt`.
- $pyvenvCfg = Get-PyVenvConfig -ConfigDir $VenvDir
-
- # Next, set the prompt from the command line, or the config file, or
- # just use the name of the virtual environment folder.
- if ($Prompt) {
- Write-Verbose "Prompt specified as argument, using '$Prompt'"
- } else {
- Write-Verbose "Prompt not specified as argument to script, checking pyvenv.cfg value"
- if ($pyvenvCfg -and $pyvenvCfg['prompt']) {
- Write-Verbose " Setting based on value in pyvenv.cfg='$($pyvenvCfg['prompt'])'"
- $Prompt = $pyvenvCfg['prompt'];
- }
- else {
- Write-Verbose " Setting prompt based on parent's directory's name. (Is the directory name passed to venv module when creating the virutal environment)"
- Write-Verbose " Got leaf-name of $VenvDir='$(Split-Path -Path $venvDir -Leaf)'"
- $Prompt = Split-Path -Path $venvDir -Leaf
- }
- }
-
- Write-Verbose "Prompt = '$Prompt'"
- Write-Verbose "VenvDir='$VenvDir'"
-
- # Deactivate any currently active virtual environment, but leave the
- # deactivate function in place.
- deactivate -nondestructive
-
- # Now set the environment variable VIRTUAL_ENV, used by many tools to determine
- # that there is an activated venv.
- $env:VIRTUAL_ENV = $VenvDir
-
- if (-not $Env:VIRTUAL_ENV_DISABLE_PROMPT) {
-
- Write-Verbose "Setting prompt to '$Prompt'"
-
- # Set the prompt to include the env name
- # Make sure _OLD_VIRTUAL_PROMPT is global
- function global:_OLD_VIRTUAL_PROMPT { "" }
- Copy-Item -Path function:prompt -Destination function:_OLD_VIRTUAL_PROMPT
- New-Variable -Name _PYTHON_VENV_PROMPT_PREFIX -Description "Python virtual environment prompt prefix" -Scope Global -Option ReadOnly -Visibility Public -Value $Prompt
-
- function global:prompt {
- Write-Host -NoNewline -ForegroundColor Green "($_PYTHON_VENV_PROMPT_PREFIX) "
- _OLD_VIRTUAL_PROMPT
- }
- }
-
- # Clear PYTHONHOME
- if (Test-Path -Path Env:PYTHONHOME) {
- Copy-Item -Path Env:PYTHONHOME -Destination Env:_OLD_VIRTUAL_PYTHONHOME
- Remove-Item -Path Env:PYTHONHOME
- }
-
- # Add the venv to the PATH
- Copy-Item -Path Env:PATH -Destination Env:_OLD_VIRTUAL_PATH
- $Env:PATH = "$VenvExecDir$([System.IO.Path]::PathSeparator)$Env:PATH"
|