如何通过Python脚本切换当前Shell的venv虚拟环境？

你用os.system('source myvenv/bin/activate')无效的本质原因是进程环境空间完全隔离：Python脚本是你当前Shell的子进程，os.system启动的又是Python进程的子Shell——这个子Shell里的环境变量修改仅在自身生命周期内有效，一旦子Shell退出，所有修改都会被清空，根本影响不到父Shell（你运行Python脚本的那个Shell）。

既然没法直接修改父进程环境，换个思路：让Python根据你的动态逻辑生成修改环境的Shell命令，再让父Shell执行这些命令。具体实现如下：

1. 编写Python脚本（activate_venv.py） #

这个脚本负责动态确定目标venv路径，然后输出对应的Shell环境配置命令：

import os
import sys

# 这里可以替换成你的动态逻辑，比如从命令行参数/项目配置中获取venv路径
# 示例：支持通过命令行参数传入venv名称，如python activate_venv.py myvenv
venv_path = sys.argv[1] if len(sys.argv) > 1 else "myvenv"
venv_bin = os.path.abspath(os.path.join(venv_path, "bin"))

# 构造需要输出的Shell命令
print(f'export VIRTUAL_ENV="{os.path.abspath(venv_path)}"')
print(f'export PATH="{venv_bin}:$PATH"')
# 保留原PS1的前缀，避免重复添加
print(f'export PS1="({venv_path})${{PS1#\\({venv_path}\\)}}"')
# 添加deactivate函数，实现退出虚拟环境的功能
print('deactivate() {')
print(f'    export PATH="${{PATH#"{venv_bin}:"}}"')
print('    unset VIRTUAL_ENV')
print(f'    export PS1="${{PS1#\\({venv_path}\\)}}"')
print('    unset -f deactivate')
print('}')

2. 在父Shell中执行脚本 #

运行以下命令，让父Shell直接执行Python输出的命令：

eval $(python activate_venv.py)

如果需要指定自定义venv路径，直接传参数即可：

eval $(python activate_venv.py my_custom_venv)

执行后，你的Shell环境会直接被修改：PATH优先使用venv的bin目录，PS1会加上(myvenv)前缀，还能通过deactivate命令退出虚拟环境，和手动执行source的效果完全一致。

3. 优化使用体验（可选） #

如果觉得每次敲eval $(...)麻烦，可以在.bashrc或.zshrc里添加别名：

alias activate-venv='eval $(python /path/to/activate_venv.py)'

之后直接敲activate-venv或activate-venv my_custom_venv就能快速激活目标虚拟环境。

内容的提问来源于stack exchange，提问作者gnarlyninja