PDF加水印命令行程序参数变量添加失败及PyPDF2 TypeError异常求助

先帮你一步步排查并解决遇到的两个问题：

1. 修复TypeError异常 #

这个错误是因为你混淆了PyPDF2.PdfFileWriter和PyPDF2.PdfFileReader的核心用法：

• PdfFileReader是用来读取现有PDF的类，必须传入打开的文件对象
• PdfFileWriter是用来生成新PDF的类，不需要在初始化时传入文件对象（后续通过write()方法写入文件）

你的错误代码：

watermark = PyPDF2.PdfFileWriter(open(mark, 'rb'))
# ...
template = PyPDF2.PdfFileWriter(open(inputs[i], 'rb'))

修正为：

watermark = PyPDF2.PdfFileReader(open(mark, 'rb'))
# ...
template = PyPDF2.PdfFileReader(open(inputs[i], 'rb'))

另外你的Marker函数里还有变量名错误：参数传入的是pdf_list，但函数内部用了未定义的template，需要替换成pdf_list；同时每次处理一个PDF时，要创建新的PdfFileWriter实例，避免多个PDF的页面被合并到同一个输出文件中覆盖。

2. 解决“无法在参数中添加变量”的问题 #

如果你的需求是让程序支持更灵活的命令行参数（比如指定输出目录、自定义输出文件名模板等），可以用Python的argparse模块替代直接读取sys.argv，这样就能轻松支持参数变量配置。

完整修正后的代码 #

下面是整合了所有修复点的代码，同时解决了两个问题：

import PyPDF2
import argparse

def add_watermark(watermark_reader, input_pdf_reader, output_path):
    # 每次处理新PDF都创建独立的Writer实例
    writer = PyPDF2.PdfFileWriter()
    # 取水印PDF的第一页作为水印
    watermark_page = watermark_reader.getPage(0)
    
    for page_num in range(input_pdf_reader.getNumPages()):
        page = input_pdf_reader.getPage(page_num)
        page.mergePage(watermark_page)
        writer.addPage(page)
    
    with open(output_path, 'wb') as output_file:
        writer.write(output_file)

if __name__ == "__main__":
    # 用argparse构建灵活的命令行参数支持
    parser = argparse.ArgumentParser(description='给PDF文件添加水印')
    parser.add_argument('watermark', help='水印PDF文件的路径')
    parser.add_argument('inputs', nargs='+', help='需要添加水印的PDF文件路径（可多个）')
    parser.add_argument('--output-dir', default='.', help='水印文件的保存目录（默认当前目录）')
    
    args = parser.parse_args()
    
    # 读取水印PDF
    watermark_reader = PyPDF2.PdfFileReader(open(args.watermark, 'rb'))
    
    # 批量处理每个输入PDF
    for pdf_path in args.inputs:
        # 生成带标识的输出文件名，避免覆盖
        output_filename = f"watermarked_{pdf_path.split('/')[-1]}"
        output_path = f"{args.output_dir}/{output_filename}"
        
        input_reader = PyPDF2.PdfFileReader(open(pdf_path, 'rb'))
        add_watermark(watermark_reader, input_reader, output_path)
        print(f"已生成水印PDF：{output_path}")

改进后的运行示例 #

现在你可以灵活使用参数变量，比如指定输出目录：

python watermarker.py wtr.pdf single.pdf twist.pdf --output-dir ./watermarked_files

也可以在命令行中使用shell变量（适用于Linux/macOS）：

# 定义变量存储需要处理的PDF列表
INPUT_PDFS="single.pdf twist.pdf super.pdf"
# 传入变量运行程序
python watermarker.py wtr.pdf $INPUT_PDFS --output-dir ./output

额外注意事项 #

• 如果你使用的是PyPDF2 2.x版本，部分API名称有变化（比如getNumPages()改为get_num_pages()），如果遇到版本兼容问题，可以安装指定版本：pip install PyPDF2==1.26.0
• 确保水印PDF只有一页，否则代码只会取第一页作为水印

内容的提问来源于stack exchange，提问作者Mohan Sasi