如何在此 PDF 水印去除脚本中添加批处理操作？

如何解决如何在此 PDF 水印去除脚本中添加批处理操作？

这个脚本不是我详细说明的，它允许我从 PDF 文件中删除水印。问题是它没有做批量操作的选项，我在同一个目录下有1000多个PDF文件

from PyPDF2 import PdfFileReader,PdfFileWriter

def removeWatermark(input_fname: str,output_fname: str):

    with open(input_fname,"rb") as inputFile,open(output_fname,"wb") as outputFile:

        reader = PdfFileReader(inputFile)
        writer = PdfFileWriter()

        for n in range(reader.numPages):

            page = reader.getPage(n)
            del page["/Contents"][-1]

            writer.addPage(page)

        writer.write(outputFile)

if __name__ == "__main__":

    import sys

    if len(sys.argv) != 3:
        raise RuntimeError("Arguments not correct!")

    removeWatermark(sys.argv[1],sys.argv[2])

如果有人能帮助我，我很感激。

依赖：PyPDF2 和 PDFtk

我解决了这个问题：

import os
from PyPDF2 import PdfFileReader,PdfFileWriter

path = os.path.dirname(__file__)
files = os.listdir(path)

def removeWater(inputfilename,outputfilename):

        "input"
        inputFile = open(inputfilename,"rb")
        outputFile = open(outputfilename,"wb")
        
        pdfReader = PdfFileReader(inputFile)
        pdfWriter = PdfFileWriter()
        
        for n in range(pdfReader.numPages):

            page = pdfReader.getPage(n)
            del page["/Contents"][-1]

            pdfWriter.addPage(page)

        pdfWriter.write(outputFile)

if __name__ == '__main__':   
    for filename in files:
            if filename.endswith(('.pdf')):
                removeWater(os.path.splitext(filename)[0] + ".pdf",os.path.splitext(filename)[0] + "- limpo" + ".pdf")