【Python】Python实现解压rar文件

2024-06-15 14:04

文章标签 python 实现 rar 解压

本文主要是介绍【Python】Python实现解压rar文件，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Python实现解压rar文件

零、需求

最近在开发一个填分数的应用，需要用到selenium，那么自然需要用到浏览器，浏览器内置到应用中，但是上传到GitCode的时候被限制了，单个文件大小只能是10M以内。所以只能压缩，分卷，用的时候再解压就好。zip的分卷解压要合并后才能解压，太占用磁盘资源了，因此用rar。7z之前用过，也比较麻烦，现在想试试rar。

壹、实现

准备工作

安装`rarfile`库

pip install rarfile

RAR程序

RAR不是开源的，没有Python实现，本质还是调用exe解压，故需要下载一个WinRAR，安装WinRAR后我们在WinRAR的安装目录中找到UnRAR.exe，把这个复制到我们项目目录下即可，这样其他电脑运行我们的代码时不需要另外安装WinRAR。
项目结构

代码实现

我们需要用到rarfile这个库，这个库的用法跟Python内置的zipfile用法一样，所以学习成本几乎为0，我们要做的就是在使用rarfile之前把UnRAR.exe添加到path环境变量中去。
我们需要把UnRAR.exe所在目录传入这个函数即可，这个函数不会改变系统的设置，只影响当前程序的环境变量，添加环境变量代码如下：

def set_path(path:str):"""添加目录到path环境变量中:param path: 目录:return: None"""os_path = os.environ.get('path')if path in os_path:returnelse:if os.path.exists(path):os.environ['path'] = f'{os_path};{path};'else:raise ValueError('环境变量路径不存在')

之后是解压代码的实现，这个跟zipfile一样用的：

def unrar_file(rar_file, target_path=None, password=None, progress_fun=None):"""解压rar文件:param rar_file: 文件路径:param target_path: 解压目标路径:param password: 密码:param progress_fun: 进度函数，会传递一个 0-1的进度浮点:return: 成功 True"""# 设置环境变量set_path(get_relative_path('chrome'))# 设置密码if not (password is None):password = password.encode('utf-8')logger.debug('解压rar文件：' + str(rar_file))# 打开rar文件rf = rarfile.RarFile(rar_file)try:file_list = rf.infolist()# 统计总大小all_size = 0for i in file_list:all_size += i.file_size# 解压unrar_size = 0for file in file_list:rf.extract(file, path=target_path, pwd=password)unrar_size += file.file_size# 调用显示进度的函数if progress_fun:progress_fun(unrar_size / all_size)except Exception as e:raise efinally:rf.close()return True

代码中get_relative_path函数是通过相对路径获取绝对路径的函数，这个与本文无关就不贴出了，返回就是一个绝对路径。代码中logger是日志对象，相当于就是打印函数，可以改成print输出。代码中progress_fun函数是打印进度的函数，接受一个浮点数，代表进度，这个大家可以自己实现一下，也可以直接删除。