logistic-bot/paperless-asn-qr-codes
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: reformat using black

Browse source
This commit is contained in:
Jan Christian Grünhage 2023-11-25 16:40:09 +01:00
parent 46f3a2fb6f
commit d2cc755e44
No known key found for this signature in database
GPG key ID: EEC1170CE56FA2ED
2 changed files with 94 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

98
paperless_asn_qr_codes/avery_labels.py
View file

@ -18,9 +18,11 @@ from reportlab.lib.units import mm
# (with X,Y=0,0 at the lower right) or a string "form" name of a form # (with X,Y=0,0 at the lower right) or a string "form" name of a form
# previously created with canv.beginForm(). # previously created with canv.beginForm().
@dataclass @dataclass
class LabelInfo: class LabelInfo:
"""Class for modeling label info""" """Class for modeling label info"""
_: KW_ONLY _: KW_ONLY
labels_horizontal: int labels_horizontal: int
labels_vertical: int labels_vertical: int
@ -29,30 +31,76 @@ class LabelInfo:
margin: tuple[float, float] margin: tuple[float, float]
pagesize: tuple[float, float] pagesize: tuple[float, float]
labelInfo: dict[str, LabelInfo] = { labelInfo: dict[str, LabelInfo] = {
"averyL4731": LabelInfo(labels_horizontal= 7, labels_vertical= 27, label_size= (25.4*mm, 10*mm), gutter_size= (2.5*mm, 0), margin= (9*mm, 13.5*mm), pagesize= A4), "averyL4731": LabelInfo(
labels_horizontal=7,
labels_vertical=27,
label_size=(25.4 * mm, 10 * mm),
gutter_size=(2.5 * mm, 0),
margin=(9 * mm, 13.5 * mm),
pagesize=A4,
),
# 2.6 x 1 address labels # 2.6 x 1 address labels
"avery5160": LabelInfo(labels_horizontal= 3, labels_vertical= 10, label_size= (187, 72), gutter_size= (11, 0), margin= (14, 36), pagesize= LETTER), "avery5160": LabelInfo(
"avery5161": LabelInfo(labels_horizontal= 2, labels_vertical= 10, label_size= (288, 72), gutter_size= (0, 0), margin= (18, 36), pagesize= LETTER), labels_horizontal=3,
labels_vertical=10,
label_size=(187, 72),
gutter_size=(11, 0),
margin=(14, 36),
pagesize=LETTER,
),
"avery5161": LabelInfo(
labels_horizontal=2,
labels_vertical=10,
label_size=(288, 72),
gutter_size=(0, 0),
margin=(18, 36),
pagesize=LETTER,
),
# 4 x 2 address labels # 4 x 2 address labels
"avery5163": LabelInfo(labels_horizontal= 2, labels_vertical= 5, label_size= (288, 144), gutter_size= (0, 0), margin= (18, 36), pagesize= LETTER), "avery5163": LabelInfo(
labels_horizontal=2,
labels_vertical=5,
label_size=(288, 144),
gutter_size=(0, 0),
margin=(18, 36),
pagesize=LETTER,
),
# 1.75 x 0.5 return address labels # 1.75 x 0.5 return address labels
"avery5167": LabelInfo(labels_horizontal= 4, labels_vertical= 20, label_size= (126, 36), gutter_size= (0, 0), margin= (54, 36), pagesize= LETTER), "avery5167": LabelInfo(
labels_horizontal=4,
labels_vertical=20,
label_size=(126, 36),
gutter_size=(0, 0),
margin=(54, 36),
pagesize=LETTER,
),
# 3.5 x 2 business cards # 3.5 x 2 business cards
"avery5371": LabelInfo(labels_horizontal= 2, labels_vertical= 5, label_size= (252, 144), gutter_size= (0, 0), margin= (54, 36), pagesize= LETTER), "avery5371": LabelInfo(
labels_horizontal=2,
labels_vertical=5,
label_size=(252, 144),
gutter_size=(0, 0),
margin=(54, 36),
pagesize=LETTER,
),
} }
RETURN_ADDRESS = 5167 RETURN_ADDRESS = 5167
BUSINESS_CARDS = 5371 BUSINESS_CARDS = 5371
class AveryLabel:
class AveryLabel:
def __init__(self, label, **kwargs): def __init__(self, label, **kwargs):
data = labelInfo[label] data = labelInfo[label]
self.across = data.labels_horizontal self.across = data.labels_horizontal
self.down = data.labels_vertical self.down = data.labels_vertical
self.size = data.label_size self.size = data.label_size
self.labelsep = self.size[0]+data.gutter_size[0], self.size[1]+data.gutter_size[1] self.labelsep = (
self.size[0] + data.gutter_size[0],
self.size[1] + data.gutter_size[1],
)
self.margins = data.margin self.margins = data.margin
self.topDown = True self.topDown = True
self.debug = False self.debug = False
@ -61,9 +109,9 @@ class AveryLabel:
self.__dict__.update(kwargs) self.__dict__.update(kwargs)
def open(self, filename): def open(self, filename):
self.canvas = canvas.Canvas( filename, pagesize=self.pagesize ) self.canvas = canvas.Canvas(filename, pagesize=self.pagesize)
if self.debug: if self.debug:
self.canvas.setPageCompression( 0 ) self.canvas.setPageCompression(0)
self.canvas.setLineJoin(1) self.canvas.setLineJoin(1)
self.canvas.setLineCap(1) self.canvas.setLineCap(1)
@ -72,13 +120,13 @@ class AveryLabel:
x = self.position x = self.position
if y == None: if y == None:
if self.topDown: if self.topDown:
x,y = divmod(x, self.down) x, y = divmod(x, self.down)
else: else:
y,x = divmod(x, self.across) y, x = divmod(x, self.across)
return ( return (
self.margins[0]+x*self.labelsep[0], self.margins[0] + x * self.labelsep[0],
self.pagesize[1] - self.margins[1] - (y+1)*self.labelsep[1] self.pagesize[1] - self.margins[1] - (y + 1) * self.labelsep[1],
) )
def advance(self): def advance(self):
@ -100,33 +148,33 @@ class AveryLabel:
# Or, pass a callable and an iterator. We'll do one label # Or, pass a callable and an iterator. We'll do one label
# per iteration of the iterator. # per iteration of the iterator.
def render( self, thing, count, *args ): def render(self, thing, count, *args):
assert callable(thing) or isinstance(thing, str) assert callable(thing) or isinstance(thing, str)
if isinstance(count, Iterator): if isinstance(count, Iterator):
return self.render_iterator( thing, count ) return self.render_iterator(thing, count)
canv = self.canvas canv = self.canvas
for i in range(count): for i in range(count):
canv.saveState() canv.saveState()
canv.translate( *self.topLeft() ) canv.translate(*self.topLeft())
if self.debug: if self.debug:
canv.setLineWidth( 0.25 ) canv.setLineWidth(0.25)
canv.rect( 0, 0, self.size[0], self.size[1] ) canv.rect(0, 0, self.size[0], self.size[1])
if callable(thing): if callable(thing):
thing( canv, self.size[0], self.size[1], *args ) thing(canv, self.size[0], self.size[1], *args)
elif isinstance(thing, str): elif isinstance(thing, str):
canv.doForm(thing) canv.doForm(thing)
canv.restoreState() canv.restoreState()
self.advance() self.advance()
def render_iterator( self, func, iterator ): def render_iterator(self, func, iterator):
canv = self.canvas canv = self.canvas
for chunk in iterator: for chunk in iterator:
canv.saveState() canv.saveState()
canv.translate( *self.topLeft() ) canv.translate(*self.topLeft())
if self.debug: if self.debug:
canv.setLineWidth( 0.25 ) canv.setLineWidth(0.25)
canv.rect( 0, 0, self.size[0], self.size[1] ) canv.rect(0, 0, self.size[0], self.size[1])
func( canv, self.size[0], self.size[1], chunk ) func(canv, self.size[0], self.size[1], chunk)
canv.restoreState() canv.restoreState()
self.advance() self.advance()

32
paperless_asn_qr_codes/main.py
View file

@ -4,29 +4,37 @@ from reportlab.lib.units import mm, cm
from reportlab_qrcode import QRCodeImage from reportlab_qrcode import QRCodeImage
from paperless_asn_qr_codes import avery_labels from paperless_asn_qr_codes import avery_labels
def render(c,x,y):
def render(c, x, y):
global startASN global startASN
barcode_value = f"ASN{startASN:07d}" barcode_value = f"ASN{startASN:07d}"
startASN = startASN + 1 startASN = startASN + 1
qr = QRCodeImage(barcode_value, size=y*0.9) qr = QRCodeImage(barcode_value, size=y * 0.9)
qr.drawOn(c,1*mm,y*0.05) qr.drawOn(c, 1 * mm, y * 0.05)
c.setFont("Helvetica", 2*mm) c.setFont("Helvetica", 2 * mm)
c.drawString(y, (y-2*mm)/2, barcode_value) c.drawString(y, (y - 2 * mm) / 2, barcode_value)
def main(): def main():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
prog='paperless-asn-qr-codes', prog="paperless-asn-qr-codes",
description='CLI Tool for generating paperless ASN labels with QR codes') description="CLI Tool for generating paperless ASN labels with QR codes",
parser.add_argument('start_asn') )
parser.add_argument('output_file') parser.add_argument("start_asn")
parser.add_argument('--format', choices=avery_labels.labelInfo.keys(), default="averyL4731") parser.add_argument("output_file")
parser.add_argument(
"--format", choices=avery_labels.labelInfo.keys(), default="averyL4731"
)
args = parser.parse_args() args = parser.parse_args()
global startASN global startASN
startASN = int(args.start_asn) startASN = int(args.start_asn)
label = avery_labels.AveryLabel(args.format) label = avery_labels.AveryLabel(args.format)
label.open(args.output_file) label.open(args.output_file)
# by default, we render all labels possible on a single sheet # by default, we render all labels possible on a single sheet
count = avery_labels.labelInfo[args.format].labels_horizontal*avery_labels.labelInfo[args.format].labels_vertical count = (
label.render(render, count ) avery_labels.labelInfo[args.format].labels_horizontal
* avery_labels.labelInfo[args.format].labels_vertical
)
label.render(render, count)
label.close() label.close()