diff --git a/sehuatang/shehuatang.py b/sehuatang/shehuatang.py index 567dbea..b1d246b 100644 --- a/sehuatang/shehuatang.py +++ b/sehuatang/shehuatang.py @@ -17,6 +17,7 @@ from reportlab.pdfbase import pdfmetrics from datetime import datetime from PIL import Image as PILImage import re # 用于正则表达式提取磁力链接 +from PyPDF2 import PdfReader, PdfWriter def download_image(url): @@ -168,9 +169,12 @@ def fetch_and_create_pdf(url): # 生成PDF doc.build(content) + # 获取PDF文件的绝对路径 absolute_pdf_path = os.path.abspath(pdf_filename) print(f"PDF saved as {absolute_pdf_path}") + # 加密PDF + add_pdf_encryption(absolute_pdf_path) # 关闭浏览器 driver.quit() @@ -178,6 +182,26 @@ def fetch_and_create_pdf(url): return absolute_pdf_path +def add_pdf_encryption(pdf_file, password="4000"): + """ 使用PyPDF2为PDF添加加密保护 """ + pdf_writer = PdfWriter() + pdf_reader = PdfReader(pdf_file) + + # 将所有页面添加到PDF写入器中 + for page_num in range(len(pdf_reader.pages)): + pdf_writer.add_page(pdf_reader.pages[page_num]) + + # 添加密码 + pdf_writer.encrypt(password) + + # 保存加密后的PDF + with open(pdf_file, "wb") as output_pdf: + pdf_writer.write(output_pdf) + + print(f"PDF加密成功,密码为: {password}") + + + def pdf_file_path(): url = 'https://www.sehuatang.net/forum.php?mod=forumdisplay&fid=103&filter=typeid&typeid=481' pdf_path = fetch_and_create_pdf(url)