用python写pdf_python写个合并pdf的小工具

[Python] 纯文本查看 复制代码import re

from PyPDF4.pdf import PdfFileReader as pr, PdfFileWriter as pw

def MergePDF(dir_path):

pdf_files = list()

merged_file = pw()

for path, _, files in os.walk(dir_path):

pdf_files.extend([os.path.join(path, f)

for f in files if f.lower().endswith('.pdf')]) # 遍历当前文件夹下所有pdf文件并组合成相对路径

pdf_files.sort(key=lambda i: int(re.findall(r'(\d+).*?', i)

[0]) if re.findall(r'(\d+).*?', i) else -1) # 按照路径排序(没有数字的在前面, 有数字的从小到大排序)

for pdf_file in pdf_files:

print(pdf_file)

try:

pdf = pr(open(pdf_file, "rb"))

except:

print(f'{pdf_file}无法解析')

continue

if pdf.isEncrypted:

print(f'{pdf_file} 是加密文件')

res = input('输入密码(回车键跳过)')

while res:

try:

pdf.decrypt(res) # pdf 解密