pip 批量完全卸载包

创作背景

因为我本机环境中安装的第三方库太多了，所以今天我准备把它们都卸载了，但因为太多了，所以不可能手动一个一个来，于是我便写了个小脚本，本文就记录这个脚本的作用及使用。

如果觉得我这篇文章写的好的话，能不能给我 点个赞 ，评论 、收藏 一条龙(☆▽☆)。如果要点个 关注 的话也不是不可以?。

请各位参加一下文末的 投票 哦，如果 有什么不足之处，还 请各位大佬在评论区提出，不胜感激。

问题分析

要实现上述功能，我们需要解决以下问题：

获取所有已安装的包
获取每个包的依赖
命令行交互，卸载指定包

解决方法

上述三个问题均可以使用 subprocess.Popen 包进行解决。为了方便，第三问题使用 subprocess.run 解决。
网上已经有好多文章都对 subprocess.Popen 和 subprocess.run 的参数进行解释，这里不多赘述。
对于 subprocess.Popen ，除了要执行的命令外，我只设置了 stdin、stdout、stderr 参数。
对于subprocess.run ，除了要执行的命令外，我只设置了如下参数：

universal_newlines ，设置输入输出的数据类型，True 为字符串，否则为字节串。
capture_output ，设置是否显示命令执行结果，True 显示，否则不显示。
input ，这个是关键，使得代码可以与命令行进行交互，即指定命令后，在命令行输入内容执行。在本文中的作用是执行 pip uninstall 【包名】 后输入 y 进行确定。

代码详解

首先导入所需的库：re 、subprocess 。
然后将卸载一个包的代码封装成一个函数，如下（本菜鸡代码水平不足，还请各位大佬指出问题）：

def uninstall_completely(name): # 必备的或不需要卸载的库，可以自行设置 skips = ['pip', 'urllib3', 'setuptools', 'wheel'] if name in skips or name.startswith('-'): return print(f'Start to uninstall { 
     name}') # 初始化 Popen，读取命令 pip show 【包名】 的执行结果 pipe = subprocess.Popen(f'pip show { 
     name}', stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE) # 如果没有安装该包，则退出 if b'WARNING: Package(s) not found: ' in pipe.stderr.read(): print(f'An error occurred when uninstalling { 
     name}: { 
     name} 不存在\n') return # 正则匹配获得所有依赖包名 # stdout.read() 的结果是字节串，需要转换为字符串 requirements = ''.join(re.findall('Requires: (.*?)\r\n', pipe.stdout.read().decode())) print(f"{ 
     name}'s requirements: { 
     requirements}") # 关闭命令行 pipe.terminate() # 卸载指定包 try: # 执行命令 pip uninstall 【包名】 # 执行命令后需要输入是否卸载 [y/n]，因为要卸载，所以指定 input 参数为 'y' obj = subprocess.run(f'pip uninstall { 
     name}', universal_newlines=True, capture_output=True, input='y') # 如果出错，则输出报错原因 if not obj.stderr == '': print(obj.stderr) return # 否则卸载成功 else: print(f'Uninstall { 
     name} successfully.') # 防止中途报错导致程序停止运行 except Exception as e: print(f'An error occurred when uninstalling { 
     name}: { 
     e}') # 输出结果分隔 print('-------------------------------------------') # 卸载指定包的所有依赖包，递归调用本函数 for _ in requirements.split(', '): if r == '': continue uninstall_completely(_)

调用函数代码如下：

for line in subprocess.Popen('pip list', stdout=subprocess.PIPE).stdout.read().decode().split('\n')[2:]: name = line.split(' ')[0] if name == '': continue uninstall_completely(name)

其中：

pip list 可以查看当前安装的所有包。
.decode() 是因为 stdout.read() 的结果是字节串，需要将其转为字符串。
[2:] 去除如下图所示的无用行

如果只卸载单个包的话，直接调用函数。
如果卸载部分包的话，遍历列表并分别调用函数。

改 BUG

写代码的时候 BUG 并不少见，但这次挺少的。出错的原因是读取执行结果时编码错误导致。
具体过程为 run 函数中调用 Popen.communicate() 函数，如下：

with Popen(*popenargs, kwargs) as process: try: stdout, stderr = process.communicate(input, timeout=timeout) except TimeoutExpired as exc: process.kill()

然后调用 Popen._communicate() 函数，如下：

try: stdout, stderr = self._communicate(input, endtime, timeout) except KeyboardInterrupt: ...

又调用 Popen._readerthread() 函数，如下：

self.stdout_thread = threading.Thread(target=self._readerthread, args=(self.stdout, self._stdout_buff))

看一下 Popen._readerthread() ，如下：

def _readerthread(self, fh, buffer): buffer.append(fh.read()) fh.close()

此时会从 Popen.stdout 中读取命令执行结果。

再看一下 Popen.stdout 的初始化代码，如下：

self.text_mode = encoding or errors or text or universal_newlines ... self.stdout = io.open(c2pread, 'rb', bufsize) if self.text_mode: self.stdout = io.TextIOWrapper(self.stdout, encoding=encoding, errors=errors)

此时就明了了，如果指定了 encoding 、errors、text、universal_newlines 中任意一个或多个参数，就意味着输出的结果是 字符串 ，而如果没有指定 encoding 参数的话，默认是使用 gbk 编码，如果和环境中的编码方式不一致的话会导致编码报错。

那我们可以修改一下 Popen 的源码，在 subprocess 中第 767 行 self.text_mode 的定义下一行加入如下代码：

if self.text_mode and encoding is None: encoding = sys.getdefaultencoding()

如果要将字节串转为字符串并且没有指定编码格式的话，就使用环境默认编码。

结尾

有想要一起学习 python 的小伙伴可以 私信我 进群哦。

以上就是我要分享的内容，因为 学识尚浅，会有不足，还 请各位大佬指正。
有什么问题也可在评论区留言。
在这里插入图片描述

发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/209456.html原文链接：https://javaforall.net

pip 批量完全卸载包