?i dZddlZddlmZddlZddlmZmZdZdZ dZ de d e d ejfd Z d ejde d ed dfdZded ejfdZdede de d efdZy)u qr_merge.py — Core library for inserting QR codes into a CarrierQR blank PDF. Shared by the CLI (add_qr_codes.py) and the HTTP service (service.py). N)Image) PdfReader PdfWriter)(iy)i ri rurlsize_pxreturnc0tjtjjdd}|j ||j d|j ddjd }|j||ftjS) u8Return a square QR code PIL Image at size_px × size_px. )error_correctionbox_sizeborderT)fitblackwhite) fill_color back_colorRGB) qrcodeQRCode constantsERROR_CORRECT_Madd_datamake make_imageconvertresizerLANCZOS)r r qrimgs /var/www/html/qr/qr_merge.pygenerate_qr_imager&sx ))99 B KKGGG --7w- ? G G NC ::w(%-- 88baserectc|\}}}}||z dtzz }||z dtzz }t||} t|| } |tz|| z dzz} |tz|| z dzz} |j| | | fy)z5Paste a centred QR code into rect on base (in-place).N) PADDING_PXminr&paste) r(r r)x1y1x2y2avail_wavail_hqr_sizeqr_imgpaste_xpaste_ys r% _composite_qrr9#sNBB2gJ&G2gJ&G'7#GW-F:o7!2q 88G:o7!2q 88GJJv)*r' pdf_bytesc\ttj|}|jd}|ddj D]\\}}|j ddk(st jtj|jjdcStd)zAExtract the embedded JPEG from PDF bytes and return as PIL Image.rz /Resourcesz/XObjectz/Subtypez/ImagerzNo image XObject found in PDF) rioBytesIOpagesitemsgetropen_datar ValueError)r:readerpagenamexobjs r%extract_base_imagerH/s rzz), -F <<?D<(4::<E d 88J 8 +::bjj45==eD DE 4 55r'blank_pdf_bytes submittal_urlproduction_urlct|}t||tt||tt t j |}|jdj}t j }|j|dd|jdt j }tj|j|dd|jdt |}t} | j|jdt j } | j| | j!S)z Main entry point. Given the blank carrier PDF bytes and two URLs, return the completed carrier PDF bytes with QR codes inserted. rJPEG_)formatqualityPDFH)rO resolution)rHr9 LEFT_RECT RIGHT_RECTrr<r=r>mediaboxsaveseekrrAradd_pagewritegetvalue) rIrJrKr(rD media_boximg_bufpdf_buf tmp_readerwriterouts r%build_carrier_pdfrb9s o .D$ y1$ 3rzz/2 3F Q((IjjlGIIgfbI1 LLOjjlG JJwWUrB LLO7#J [F OOJ$$Q'( **,C LL <<>r')__doc__r<PILrrpypdfrrrTrUr,strintr&tupler9bytesrHrbr'r%rks   & % $   93 9 9 9 + +# +U +t +6%6EKK6     r'