《Python编程快速上手—让繁琐工作自动化》第13章实践项目

##13.6.1 PDF偏执狂

import os,PyPDF2,sys

os.chdir(r'C:\Users\Yuno\Desktop')
for foldername,subfoldernames,filenames in os.walk('.'): #遍历当前工作目录找到pdf文件
    for filename in filenames:
        if filename.endswith('.pdf'):
            try:
                
                pdfReader = PyPDF2.PdfFileReader\
                            (open(os.path.join\
                                  (foldername,filename),'rb'))         #为PDF创建的PdfFileReader对象                           
                                
                pdfWriter = PyPDF2.PdfFileWriter()
                for i in range(pdfReader.numPages):                #遍历拷贝到PdfFileWriter对象                        
                    pdfWriter.addPage(pdfReader.getPage(i))
                    pdfWriter.encrypt(sys.arge[1,])       #加密            
                    pdfFile = open\
                              (filename.replace\
                                   ('.pdf',