1 #/usr/bin/env python
2 #-*- coding:utf-8 -*-
3 #本示例使用两个第三方库来实现为PDF文件添加文字水印
4 #这两个库是pyPdf和reportlab
5 #使用的Python版本是Python 2.7
6 #http://pybrary.net/pyPdf
7 #http://www.reportlab.com
8
9 importos10 #库文件的导入
11 from pyPdf importPdfFileWriter, PdfFileReader12 from reportlab.pdfgen importcanvas13
14 from PIL importImage, ImageDraw, ImageFont15 importos16
17 #指定要使用的字体和大小;/Library/Fonts/是macOS字体目录;Linux的字体目录是/usr/share/fonts/
18 font = ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSerif.ttf', 45)19
20
21 #image: 图片 text:要添加的文本 font:字体
22 def add_text_to_image(image_name, text, font=font):23 image =Image.open(image_name)24 rgba_image = image.convert('RGBA')25 text_overlay = Image.new('RGBA', rgba_image.size, (255, 255, 255, 0))26 image_draw =ImageDraw.Draw(text_overlay)27 text_size_x, text_size_y = image_draw.textsize(text, font=font)28 #设置文本文字位置
29 print(rgba_image)30 #text_xy = (rgba_image.size[0] - text_size_x, rgba_image.size[1] - text_size_y) #底部
31 #text_xy = ((rgba_image.size[0] - text_size_x) / 2, (rgba_image.size[1] - text_size_y) / 2) # 中间
32 text_xy = ((rgba_image.size[0] - text_size_x), 0) #中间
33 #设置文本颜色和透明度
34 #image_draw.text(text_xy, text, font=font, fill=(76, 234, 124, 180))
35 image_draw.text(text_xy, text, font=font, fill=(0, 0, 0))36
37 image_with_text =Image.alpha_composite(rgba_image, text_overlay)38
39 returnimage_with_text40
41 defadd_mark(info_name, text):42 printinfo_name43 #使用reportlab来创建一个PDF文件来作为一个水印文件
44 c = canvas.Canvas("watermark.pdf")45 c.setFont("Courier", 35)46
47 #设置水印文字的灰度
48 c.setFillGray(0.0, 1)49
50 #设置水印文件,并将文字倾斜45度角
51 c.saveState()52 c.translate(280, 750)53 c.rotate(0)54 c.drawCentredString(0, 0, text)55 c.restoreState()56 c.save()57
58 output =PdfFileWriter()59 input1 = PdfFileReader(file(info_name, 'rb'))60 water = PdfFileReader(file('watermark.pdf', 'rb'))61
62 #获取pdf文件的页数
63 pageNum =input1.getNumPages()64 printpageNum65 #给每一页打水印
66 for i inrange(pageNum):67 page =input1.getPage(i)68 page.mergePage(water.getPage(0))69 output.addPage(page)70 returnoutput71
72 defrun():73 os.mkdir("results")74 file_name = "info"
75 files =os.listdir(file_name)76 r =077 for t_file infiles:78 file_all_img = file_name + os.sep +t_file79 ifos.path.isfile(file_all_img):80 continue
81 for file_img inos.listdir(file_all_img):82 file_img_all_img_all = file_all_img + os.sep +file_img83 if file_img_all_img_all.endswith('pdf'):84 out_put =add_mark(file_img_all_img_all, t_file)85 outStream = file('results' + os.sep + '{}.pdf'.format(r), 'wb')86 out_put.write(outStream)87 outStream.close()88 os.remove("watermark.pdf")89 else:90 im_after = add_text_to_image(image_name=file_img_all_img_all, text=t_file)91 im_after.save("results/{}.png".format(r))92 r += 1
93
94
95
96 if __name__ == '__main__':97 run()