允许在python脚本中处理可能的未知文件名

Question

我正在使用一个脚本，它将输入的pdf覆盖到另一个本质上是信头的pdf上。 但是，我不确定如何允许该过程自动化以允许一次处理多个文件，而无需事先知道该文件的名称。 我正在使用python 2.7。

from pyPdf import PdfFileWriter, PdfFileReader


output = PdfFileWriter()
input1 = PdfFileReader(file("example.pdf", "rb"))

# add page 1 from input1 to output document, unchanged
output.addPage(input1.getPage(0))

# add page 2 from input1, but first add a watermark from another pdf:
page2 = input1.getPage(0)
watermark = PdfFileReader(file("template.pdf", "rb"))
page2.mergePage(watermark.getPage(0))

output.addPage(page2)


# finally, write "output" to document-output.pdf
outputStream = file("example.pdf", "wb")
output.write(outputStream)
outputStream.close()

Answer 1

导入os并使用os.listdir在特定目录中查找文件：

https://docs.python.org/2/library/os.html#os.listdir

Answer 2

您的问题似乎可以通过命令行参数轻松解决。

您可以导入sys模块并使用sys.argv来获取命令行参数列表。 在此链接上有一个教程应与2.7一起使用。 sys.argv是一个数组，而sys.argv [1]是您传递的第一个参数，因为sys.argv [0]是可执行文件的名称。

在命令行中，您可以执行以下命令：

python your_program.py file_name_to_work_with.pdf

如果使用的话，可以使用IDLE插入命令行参数。 如果您需要知道如何发表评论。

Answer 3

像argparse这样的东西可能会最安全

argparse为您提供非常灵活的命令行参数操作，以实现脚本的控制界面。

$ python your_program.py -t <template> -i <input1> <input2>

Answer 4

感谢您的所有贡献。 我最终使用的是：

from pyPdf import PdfFileWriter, PdfFileReader
import os, sys, glob, shutil

path = glob.glob ('p:\letter2pdf\processing\*.pdf')
template = 'p:/letter2pdf/processing/template/template.pdf'
dest1 = 'p:\letter2pdf\output'
dest2 = 'p:\letter2pdf\processing'

for files in sorted(path):

    output = PdfFileWriter()
    input1 = PdfFileReader(file(files,"rb"))

    print "title = %s" % (input1.getDocumentInfo().title)

# add page 1 from input1 to output document, unchanged
    output.addPage(input1.getPage(0))

#add a watermark from another pdf:
    page2 = input1.getPage(0)
    watermark = PdfFileReader(file (template, "rb"))
    page2.mergePage(watermark.getPage(0))



# finally, write "output" to document-output.pdf
    outputStream = file(input1.getDocumentInfo().title, "wb")
    output.write(outputStream)
    outputStream.close()