python将当前文件夹下的所有excel文件打包成一份需要密码才能打开的zip文件,设置一份初始密码
时间: 2024-09-10 18:09:49 浏览: 61
Python中可以使用`zipfile`模块配合`os`模块来完成这个任务,同时利用`getpass`模块获取用户输入的密码作为加密口令。以下是简单的步骤:
```python
import os
import zipfile
import getpass
# 获取当前目录
current_dir = os.getcwd()
# 设置初始密码
password = getpass.getpass("请输入ZIP文件的密码: ")
# 创建一个ZipFile对象,指定为写入模式,并指定加密级别
with zipfile.ZipFile('output.zip', 'w', zipfile.ZIP_DEFLATED, compresslevel=9) as zipf:
# 遍历当前目录下的所有Excel文件
for filename in os.listdir(current_dir):
if filename.endswith('.xlsx') or filename.endswith('.xls'): # 检查是否为Excel文件
file_path = os.path.join(current_dir, filename)
zipf.setpassword(password.encode()) # 设置加密口令
zipf.write(file_path) # 将文件添加到压缩包
# 提示用户输入密码,以便打开压缩包
print("Zip文件已创建,密码是:", password)
```
在这个脚本中,用户会被提示输入一个密码,该密码会用于加密生成的`output.zip`文件。当尝试打开这个压缩包时,需要提供相同的密码。
阅读全文