-
Notifications
You must be signed in to change notification settings - Fork 1
/
shipping_label.py
71 lines (56 loc) · 2.15 KB
/
shipping_label.py
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
from io import BytesIO
from PyPDF2 import PdfFileWriter, PdfFileReader
from PIL import Image
from reportlab.lib.utils import ImageReader
from reportlab.pdfgen.canvas import Canvas
def crop_shipping_label(pdf_file):
output = PdfFileWriter()
output.addPage(get_cropped_page(pdf_file))
with open('cropped_label.pdf', 'wb') as fp:
output.write(fp)
def get_cropped_page(pdf_file):
fp = open(pdf_file, 'rb')
# Set strict to False because document may not have unique keys.
reader = PdfFileReader(fp, strict=False)
page = reader.getPage(0)
page.mediaBox.lowerLeft = (75, 425)
page.mediaBox.upperRight = (560, 750)
return page
def process_return_label(pdf_file):
image = list(get_images(pdf_file))[0]
image_reader = ImageReader(image)
with open(pdf_file, 'wb') as fp:
writer = PdfFileWriter()
writer.addPage(get_image_page(image_reader))
writer.write(fp)
def get_images(pdf_file):
with open(pdf_file, 'rb') as fp:
reader = PdfFileReader(fp)
page = reader.getPage(0)
xObject = page['/Resources']['/XObject'].getObject()
for obj in xObject:
if xObject[obj]['/Subtype'] == '/Image':
width, height = (xObject[obj]['/Width'], xObject[obj]['/Height'])
# Ignore smaller images.
if height < 100:
continue
size = width, height
data = xObject[obj].getData()
if xObject[obj]['/ColorSpace'] == '/DeviceRGB':
mode = "RGB"
else:
mode = "P"
encoding = xObject[obj]['/Filter']
if encoding == '/FlateDecode' or '/FlateDecode' in encoding:
yield Image.frombytes(mode, size, data)
else:
raise Exception(
'Unexpected image encoding: {}'.format(encoding))
def get_image_page(image):
inch = 72
bio = BytesIO()
c = Canvas(bio, pagesize=(8.5*inch, 11*inch))
dim = c.drawImage(image, 0.5*inch, 6.3*inch, 495, 290)
# print(dim)
c.save()
return PdfFileReader(bio).getPage(0)