如何实现Python软件自定义PDF导出扩展名并避免PDF阅读器直接读取？

针对你的需求——用自定义扩展名保存PDF内容且不让普通PDF阅读器直接打开，这里有几个比“加换行再恢复”更靠谱的方案：

1. 给PDF内容加自定义文件头标记 #

这是最简单易维护的方式：保存时在PDF二进制内容前加一段独有的自定义标识，读取时先移除这段标识再还原成正常PDF。自定义标识选个不会和PDF内容冲突的字符串就行，比单纯加换行更可靠。

保存示例代码： #

def save_as_potato(pdf_path, output_path):
    with open(pdf_path, 'rb') as pdf_file:
        pdf_content = pdf_file.read()
    
    # 自定义专属标记，避免和PDF内容混淆
    custom_marker = b'POTATO_APP_V1.0\n'
    modified_content = custom_marker + pdf_content
    
    with open(output_path, 'wb') as potato_file:
        potato_file.write(modified_content)

读取示例代码： #

def load_from_potato(potato_path, output_pdf_path):
    with open(potato_path, 'rb') as potato_file:
        content = potato_file.read()
    
    custom_marker = b'POTATO_APP_V1.0\n'
    if content.startswith(custom_marker):
        pdf_content = content[len(custom_marker):]
        with open(output_pdf_path, 'wb') as pdf_file:
            pdf_file.write(pdf_content)
    else:
        raise ValueError("无效的.potato文件")

2. 轻量对称加密（推荐） #

如果需要一点保密性，不想让别人轻易破解内容，用简单的对称加密算法加密PDF后保存为自定义扩展名，读取时解密即可。这个方案不仅能阻止PDF阅读器直接打开，还能给内容加一层保护。

示例代码（使用pycryptodome库）： #

先安装依赖：pip install pycryptodome

加密保存：

from Crypto.Cipher import AES
from Crypto.Util.Padding import pad
import os

def encrypt_save_potato(pdf_path, output_path, password):
    with open(pdf_path, 'rb') as f:
        pdf_data = f.read()
    
    # 生成符合AES要求的32字节密钥
    key = password.ljust(32)[:32].encode('utf-8')
    iv = os.urandom(16)  # 随机初始向量，增强安全性
    cipher = AES.new(key, AES.MODE_CBC, iv)
    
    # 把初始向量和加密内容一起写入文件
    encrypted_data = iv + cipher.encrypt(pad(pdf_data, AES.block_size))
    
    with open(output_path, 'wb') as f:
        f.write(encrypted_data)

解密读取：

from Crypto.Cipher import AES
from Crypto.Util.Padding import unpad

def decrypt_load_potato(potato_path, output_pdf_path, password):
    with open(potato_path, 'rb') as f:
        encrypted_data = f.read()
    
    key = password.ljust(32)[:32].encode('utf-8')
    iv = encrypted_data[:16]  # 取出开头的初始向量
    cipher = AES.new(key, AES.MODE_CBC, iv)
    
    pdf_data = unpad(cipher.decrypt(encrypted_data[16:]), AES.block_size)
    
    with open(output_pdf_path, 'wb') as f:
        f.write(pdf_data)

3. 封装成自定义容器格式 #

如果想做得更规范、扩展性更强，可以把PDF内容和额外元数据（比如软件版本、创建时间）一起封装成自定义二进制容器，保存为.potato文件。这种方式以后要加新功能（比如嵌入缩略图、配置信息）会很方便。

保存示例： #

def save_potato_container(pdf_path, output_path):
    with open(pdf_path, 'rb') as f:
        pdf_data = f.read()
    
    # 容器结构：固定标记(4字节) + PDF内容长度(4字节大端) + PDF内容
    marker = b'POTA'
    content_length = len(pdf_data).to_bytes(4, byteorder='big')
    container_data = marker + content_length + pdf_data
    
    with open(output_path, 'wb') as f:
        f.write(container_data)

读取示例： #

def load_potato_container(potato_path, output_pdf_path):
    with open(potato_path, 'rb') as f:
        marker = f.read(4)
        if marker != b'POTA':
            raise ValueError("不是有效的.potato容器文件")
        content_length = int.from_bytes(f.read(4), byteorder='big')
        pdf_data = f.read(content_length)
    
    with open(output_pdf_path, 'wb') as f:
        f.write(pdf_data)

内容的提问来源于stack exchange，提问作者Nicolas Bartual