在日常工作中，要保护一下我们的商业机密嘛~ 如果按名单向客户逐一发文件，如果就发几份，那么可以采取用pdf软件逐份编辑，打个水印、加个密码咯。但面对成百上千份文件，就需要批量处理了！菜鸡在熬夜加班后终于发掘了这个东西的用法，分享给大家~

私信小编01即可获取大量Python学习资料

首先简单设置一下

# encoding:utf-8import os;import pandas as pd;import numpy as np;import openpyxl;import shutil;

插入页面：向文件夹1中的每一个文件中都插入一页 新文件.pdf

# -*- coding: utf-8 -*-# 读入pdf路径为path0，写出pdf路径为path1path0 = 'D:\\举例\\文件夹1'path1 = 'D:\\举例\\文件夹2'filelist=os.listdir(path0)# 读入新文件，假设新文件仅有一页，文件夹1中的文件均仅有2页pdf2 = open( 'D:\\举例\\新文件.pdf','rb')pdf_reader2 = PdfFileReader(pdf2)for file in filelist:    what = os.path.join(path0,file)    pdf1 = open(what,'rb')    pdf_reader = PdfFileReader(pdf1)    pdf_writer = PdfFileWriter()     # 第一页为文件夹1中的文件第一页    pdf_writer.addPage(pdf_reader.getPage(0))    # 第二页为新文件第一页    pdf_writer.addPage(pdf_reader2.getPage(0))    # 第三页为文件夹1中的文件第二页    pdf_writer.addPage(pdf_reader.getPage(1))# 写入合并的pdf    pdf_writer.encrypt(user_pwd="123123")    output = os.path.join(path1,file)    with open(output, 'wb') as out:         pdf_writer.write(out)

复制、改名称：先把要发的客户名称制作成name.txt，写个函数复制一下文件，把文件按客户名称命名好。

# 写函数：复制原文件并逐一对pdf文件按照name.txt更改文件名为【机构名称】，name.txt文件内仅有一列数据【机构名称】def get_files():    path='D:\\举例\\文件夹1'        name = pd.read_table('D:\\举例\\文件夹1\\name.txt');        name2 = np.array(name['机构名称']) ;        n = name.iloc[:,0].size;        for file in range(1,n+1):           path_o = 'D:\\举例\\文件夹1\\正文.pdf'                filename_new = name2[file]                path_n = os.path.join(path,filename_new+'.pdf')                shutil.copyfile(path_o, path_n);                print(filename_new);# 执行函数get_files()

然后准备上pypdf2的包，安排好汉字字体，准备当调包侠~

# 安装pyPDF2，准备批量打水印#pip install pypdf2from PyPDF2 import PdfFileReader, PdfFileWriter,PdfFileMerger#!/usr/bin/env python#!pip install reportlab# encoding:utf-8from PyPDF2 import PdfFileReader, PdfFileWriterfrom reportlab.lib.units import cmfrom reportlab.pdfbase import pdfmetrics from reportlab.pdfgen import canvasfrom reportlab.pdfbase.ttfonts import TTFont pdfmetrics.registerFont(TTFont('song', 'C:/Windows/Fonts/simsun.ttc'))#宋体import xlrd

制作水印文件：开始安排水印文件，因为客户名称可能3个字也可能“乱七八糟真长真无语绝了有限公司”这种，水印无法自动调节大小，因此把字数不同长度做成不同的水印文件，一会儿循环的时候先判断一下~ 这边还有很多长度不同的水印函数，暂略。

# 因name.txt中【机构名称】字数不一，因此水印按照字数分组设置大小、位置、旋转角度等。# 5个字以下的水印# encoding:utf-8def create_watermark5(content):    """水印信息"""    # 默认大小为21cm*29.7cm    file_name = "mark.pdf"    c = canvas.Canvas(file_name, pagesize=(30*cm, 30*cm))    # 移动坐标原点(坐标系左下为(0,0))    c.translate(4*cm, 14*cm)    # 设置字体    c.setFont('song', 120)    # 指定描边的颜色    c.setStrokeColorRGB(0, 1, 0)    # 指定填充颜色    c.setFillColorRGB(0, 1, 0)    # 画一个矩形    # c.rect(cm, cm, 7*cm, 17*cm, fill=1)    # 旋转45度,坐标系被旋转    c.rotate(45)    # 指定填充颜色    c.setFillColorRGB(0, 0, 0, 0.1)    # 设置透明度,1为不透明    # c.setFillAlpha(0.1)    # 画几个文本,注意坐标系旋转的影响    c.drawString(3*cm, 0*cm, content)    c.setFillAlpha(0.6)    # 关闭并保存pdf文件    c.save()    return file_name

打水印&加密码：来吧，下面准备打水印函数，主要用来完成把水印文件贴在原文件上面这个步骤~ 同时还在函数中插入了加密码这个功能。

# 打水印函数def add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out):    """把水印添加到pdf中"""    pdf_output = PdfFileWriter()    input_stream = open(pdf_file_in, 'rb')    pdf_input = PdfFileReader(input_stream, strict=False)    # 获取PDF文件的页数    pageNum = pdf_input.getNumPages()    # 读入水印pdf文件    pdf_watermark = PdfFileReader(open(pdf_file_mark, 'rb'), strict=False)    # 给每一页打水印    for i in range(pageNum):        page = pdf_input.getPage(i)        page.mergePage(pdf_watermark.getPage(0))        page.compressContentStreams()  # 压缩内容        pdf_output.addPage(page)    # 添加密码    pdf_writer.encrypt(user_pwd="abc")    output = os.path.join(path1,file)    with open(output, 'wb') as out:         pdf_writer.write(out)      pdf_output.write(open(pdf_file_out, 'wb'))

设置打水印路径

# 设置打水印读入、写出路径，打水印读入文件路径为path00，写出文件路径为path11# 注意：前置步骤需将文件夹1中全部未打水印的文件复制到文件夹2中path00 = 'D:\\举例\\文件夹1'path11 = 'D:\\举例\\文件夹2'filelist00=os.listdir(path00)filelist11=os.listdir(path11)

循环开始

# 写条件循环语句，完成不同字数长度的文件水印# encoding:utf-8if __name__ == '__main__':    for i,j in zip(filelist00,filelist11):        pdf_file_in = os.path.join(path00,i)        pdf_file_out = os.path.join(path11,j)        mark_name = os.path.splitext(i)[0]        if len(mark_name) <= 5:            pdf_file_mark = create_watermark5(mark_name)        elif len(mark_name) <= 8:            pdf_file_mark = create_watermark8(mark_name)        elif len(mark_name) <= 13:            pdf_file_mark = create_watermark13(mark_name)        elif len(mark_name) <= 15:            pdf_file_mark = create_watermark15(mark_name)        else:             pdf_file_mark = create_watermark18(mark_name)        add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out)        print(mark_name)

有的公司还要加盖电子章，可以仿照打水印的操作，做一个电子章的页面，重新调用打水印函数将电子章贴上去~

# 写电子章函数，可重复调用add_watermark进行打批量电子章def create_stamp(content):    """盖章"""    # 默认大小为21cm*29.7cm    file_name = "stamp.pdf"    c = canvas.Canvas(file_name, pagesize=(30*cm, 30*cm))    # 移动坐标原点(坐标系左下为(0,0))    c.translate(10*cm, 0.5*cm)    # 设置字体    c.setFont('song', 120)    # 指定描边的颜色    c.setStrokeColorRGB(0, 1, 0)    # 指定填充颜色    c.setFillColorRGB(0, 1, 0)    # 画一个矩形    # c.rect(cm, cm, 7*cm, 17*cm, fill=1)    # 旋转45度,坐标系被旋转    c.rotate(45)    # 指定填充颜色    c.setFillColorRGB(0, 0, 0, 0.1)    # 设置透明度,1为不透明    # c.setFillAlpha(0.1)    # 画几个文本,注意坐标系旋转的影响    c.drawString(3*cm, 0*cm, content)    c.setFillAlpha(0.6)    # 关闭并保存pdf文件    c.save()    return file_name

分类放置：打好水印、加好密码的文件可以按照名字自动移动至相应文件夹，避免手动分类烦恼喔~

# 按分支机构地区进行文件分类，location的格式是两列数据，分别为【机构名称】、【分支机构地区】# encoding:utf-8name11 = pd.read_table('D:\\举例\\location.txt')name11 = pd.DataFrame(name11)# 循环逻辑：如【机构名称】中带有“公司”二字，划分至企业文件夹，其余按照location.txt划分至对应的【分支机构地区】# -*- coding: utf-8 -*-for file in filelist:    filename=os.path.splitext(file)[0]    if  filename.find('公司') > 0:        old_path = os.path.join(path0,file)        new_pat = os.path.join(path1,'企业')        new_path = os.path.join(new_pat,file)        os.rename(old_path,new_path)        print("移动到企业")    else:        location = name11[name11['机构名称'] == filename]        location = location['分支机构地区'].iloc[0]        old_path = os.path.join(path0,file)        new_pat = os.path.join(path1,location)        new_path = os.path.join(new_pat,file)        os.rename(old_path,new_path)        print("移动到分支机构")

至此，文件已复制为按照【机构名称】命名的文件，文件上每页均有【机构名称】字样、由左下角至右上角的灰色大水印，每一个文件都需要输入密码“abc”才能打开，且按照分类移动至所属文件夹。

Pypdf2作为Python批量处理pdf文件的包，其功能完整，操作难度小，可以说有效提升办公效率。