From ae23e62be8f232bf4f427d689ff8738c1255b30c Mon Sep 17 00:00:00 2001 From: Leonardo Bishop Date: Sat, 15 Mar 2025 21:31:49 +0000 Subject: Clean up some code & add example config --- config-example.yaml | 7 +++++++ imggen.py | 3 +-- main.py | 14 +++++++------- 3 files changed, 15 insertions(+), 9 deletions(-) create mode 100644 config-example.yaml diff --git a/config-example.yaml b/config-example.yaml new file mode 100644 index 0000000..6367635 --- /dev/null +++ b/config-example.yaml @@ -0,0 +1,7 @@ +tito: + mac_token: + +printer: + maj: + min: + diff --git a/imggen.py b/imggen.py index cdc6314..460b432 100644 --- a/imggen.py +++ b/imggen.py @@ -1,6 +1,5 @@ from PIL import Image, ImageDraw, ImageFont import math -import printer font_bold = ImageFont.truetype("/usr/share/fonts/TTF/IosevkaSS08-Bold.ttc", 70, encoding="unic") font_norm = ImageFont.truetype("/usr/share/fonts/TTF/IosevkaSS08-Regular.ttc", 40, encoding="unic") @@ -17,7 +16,7 @@ def name(name: str): draw = ImageDraw.Draw(img) draw.text((0, 0), name, font=font_bold, fill="black") - if font_bold.getlength(name) > MAX_WIDTH: + if txt_len > MAX_WIDTH: wpercent = (MAX_WIDTH / float(img.size[0])) hsize = int((float(img.size[1]) * float(wpercent))) img = img.resize((MAX_WIDTH, hsize), Image.Resampling.LANCZOS) diff --git a/main.py b/main.py index d4c5149..2648460 100644 --- a/main.py +++ b/main.py @@ -56,14 +56,15 @@ async def create_checkin(request: Request): pizza_pref = list(filter(lambda a: a['question'] == 'What is your pizza preference?', content['answers'])) if len(pizza_pref) == 0: - return "ok" + return Response(status_code=204) + + d_reqs = None + q_d_reqs = list(filter(lambda a: a['question'] == 'Do you have any dietary restrictions?', content['answers'])) + if len(q_d_reqs) > 0: + d_reqs = q_d_reqs[0]['response'] - d_reqs = list(filter(lambda a: a['question'] == 'Do you have any dietary restrictions?', content['answers'])) logger.info(f"Printing food token for {attendee_name}...") - if len(d_reqs) == 0: - printer.print_food(attendee_name, pizza_pref[0]['response'], str(group), None) - else: - printer.print_food(attendee_name, pizza_pref[0]['response'], str(group), d_reqs[0]['response']) + printer.print_food(attendee_name, pizza_pref[0]['response'], str(group), d_reqs) serial = serial + 1 if serial % 10 == 0: @@ -73,7 +74,6 @@ async def create_checkin(request: Request): return Response(status_code=204) finally: lock.release() - if __name__ == '__main__': uvicorn.run(app, log_level="info", host="0.0.0.0", port=3000) -- cgit v1.2.3-70-g09d2