You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
2.9 KiB
127 lines
2.9 KiB
import os
|
|
|
|
# import re
|
|
import qrcode
|
|
|
|
# import hashlib
|
|
import logging
|
|
from pyx import *
|
|
|
|
# import subprocess
|
|
from tempfile import mkstemp
|
|
from datetime import datetime
|
|
|
|
|
|
__all__ = ["generate_backup"]
|
|
|
|
TEXT_X_OFFSET = 0.6
|
|
TEXT_Y_OFFSET = 8.2
|
|
PLAINTEXT_MAXLINECHARS = 73
|
|
|
|
QRCODE_HEIGHT = 8
|
|
QRCODE_PER_PAGE = 6
|
|
QRCODE_MAX_BYTE = 140
|
|
QRCODE_X_POS = [1.5, 11, 1.5, 11, 1.5, 11]
|
|
QRCODE_Y_POS = [18.7, 18.7, 10, 10, 1.2, 1.2]
|
|
|
|
PF_STR = "A4"
|
|
PF_OBJ = document.paperformat.A4
|
|
|
|
# suppressing all the warnings
|
|
for name in logging.Logger.manager.loggerDict.keys():
|
|
logging.getLogger(name).setLevel(logging.CRITICAL)
|
|
|
|
|
|
def _generate_barcode(chunkdata):
|
|
"""Generates data barcode image
|
|
|
|
Args:
|
|
chunkdata (str): ASC chunk data
|
|
|
|
Returns:
|
|
class: Represents an image object
|
|
"""
|
|
qr = qrcode.QRCode(
|
|
version=1,
|
|
border=4,
|
|
box_size=10,
|
|
error_correction=qrcode.constants.ERROR_CORRECT_L,
|
|
)
|
|
qr.add_data(chunkdata)
|
|
qr.make(fit=True)
|
|
|
|
im = qr.make_image(fill_color="black", back_color="white")
|
|
|
|
return im
|
|
|
|
|
|
def _finish_page(pdf, canvas, pageno):
|
|
"""Adds numbers to bottom of the page
|
|
|
|
Args:
|
|
pdf (class): PDF object
|
|
canvas (class): Canvas object
|
|
pageno (int): The page number
|
|
"""
|
|
canvas.text(10, 0.6, "Page {}".format(pageno + 1))
|
|
pdf.append(document.page(canvas, paperformat=PF_OBJ, fittosize=0, centered=0))
|
|
|
|
|
|
def generate_backup(ascfile):
|
|
"""Generates PDF backup file
|
|
|
|
Args:
|
|
ascfile (str): ASC (.asc) file path
|
|
"""
|
|
|
|
pageno = 0
|
|
pageid = 0
|
|
codeblocks = []
|
|
chunkdata = "^1 "
|
|
c = canvas.canvas()
|
|
|
|
with open(ascfile) as file:
|
|
ASCDATA = file.read()
|
|
|
|
unit.set(defaultunit="cm")
|
|
pdf = document.document()
|
|
|
|
for char in list(ASCDATA):
|
|
if len(chunkdata) + 1 > QRCODE_MAX_BYTE:
|
|
codeblocks.append(_generate_barcode(chunkdata))
|
|
chunkdata = "^" + str(len(codeblocks) + 1) + " "
|
|
chunkdata += char
|
|
|
|
codeblocks.append(_generate_barcode(chunkdata))
|
|
|
|
for bc in range(len(codeblocks)):
|
|
if pageid >= QRCODE_PER_PAGE:
|
|
_finish_page(pdf, c, pageno)
|
|
c = canvas.canvas()
|
|
pageno += 1
|
|
pageid = 0
|
|
|
|
c.text(
|
|
QRCODE_X_POS[pageid] + TEXT_X_OFFSET,
|
|
QRCODE_Y_POS[pageid] + TEXT_Y_OFFSET,
|
|
"{} ({}/{})".format(
|
|
text.escapestring(ascfile.split(os.sep)[-1]), bc + 1, len(codeblocks)
|
|
),
|
|
)
|
|
|
|
c.insert(
|
|
bitmap.bitmap(
|
|
QRCODE_X_POS[pageid],
|
|
QRCODE_Y_POS[pageid],
|
|
codeblocks[bc],
|
|
height=QRCODE_HEIGHT,
|
|
)
|
|
)
|
|
|
|
pageid += 1
|
|
|
|
_finish_page(pdf, c, pageno)
|
|
|
|
fd, temp_barcode_path = mkstemp(".pdf", "qr_", ".")
|
|
pdf.writetofile(temp_barcode_path)
|
|
os.rename(temp_barcode_path.split(os.sep)[-1], "{}.pdf".format(ascfile))
|
|
|