1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
import io
import itertools
from reportlab.pdfgen.canvas import Canvas
from postgresqleu.util.reporttools import cm
from postgresqleu.util.versionutil import fitz_get_page_pixmap, fitz_insert_text, fitz_insert_image
def fill_pdf_fields(pdf, available_fields, fielddata):
import fitz
if 'fields' in fielddata:
pagefields = {int(k): list(v) for k, v in itertools.groupby(fielddata['fields'], lambda x: x['page'])}
else:
pagefields = {}
pdf = fitz.open('pdf', bytes(pdf))
for pagenum, page in enumerate(pdf.pages()):
if pagenum in pagefields:
for f in pagefields[pagenum]:
# Location in the json is top-left corner, but we want bottom-left for the
# PDF. So we add the size of the font in points, which is turned into pixels
# by multiplying by 96/72.
p = fitz.Point(
f['x'],
f['y'] + fielddata['fontsize'] * 96 / 72,
)
# Preview with the field title
txt = None
for fieldname, fieldtext in available_fields:
if not fieldname.startswith('static:'):
break
if fieldname == f['field']:
txt = fieldtext
break
else:
txt = ""
if txt:
fitz_insert_text(page, p, txt, 'Courier-Bold', fielddata['fontsize'])
return pdf.write()
def pdf_watermark_preview(pdfdata):
try:
import fitz
except ImportError:
# Just return without watermark
return pdfdata
wmio = io.BytesIO()
wmcanvas = Canvas(wmio)
wmcanvas.rotate(45)
for y in -5, 0, 5, 10, 15:
t = wmcanvas.beginText()
t.setTextOrigin(cm(6), cm(y))
t.setFont("Times-Roman", 100)
t.setFillColorRGB(0.9, 0.9, 0.9)
t.textLine("PREVIEW PREVIEW")
wmcanvas.drawText(t)
wmcanvas.rotate(-45)
wmcanvas.save()
wmio.seek(0)
wmpdf = fitz.open('pdf', wmio)
wmpixmap = fitz_get_page_pixmap(next(wmpdf.pages()))
pdf = fitz.open('pdf', pdfdata)
for pagenum, page in enumerate(pdf.pages()):
fitz_insert_image(page, page.bound(), wmpixmap, False)
return pdf.write()
|
