-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
46de063
commit d818e41
Showing
24 changed files
with
545 additions
and
19 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,247 @@ | ||
| <# | ||
| .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. | ||
| .Notes | ||
| On Windows, it may be required to enable this Activate.ps1 script by setting the | ||
| execution policy for the user. You can do this by issuing the following PowerShell | ||
| command: | ||
| PS C:\> Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope CurrentUser | ||
| For more information on Execution Policies: | ||
| https://go.microsoft.com/fwlink/?LinkID=135170 | ||
| #> | ||
| 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 VIRTUAL_ENV_PROMPT altogether. | ||
| if (Test-Path -Path Env:VIRTUAL_ENV_PROMPT) { | ||
| Remove-Item -Path env:VIRTUAL_ENV_PROMPT | ||
| } | ||
|
|
||
| # 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("\\/") | ||
| 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 virtual 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 | ||
| } | ||
| $env:VIRTUAL_ENV_PROMPT = $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" |
Oops, something went wrong.