fix: card creation fix

generators/warehouse_place_qr_code_generator/generator.py

@@ -1,16 +1,50 @@
+import os
 from io import BytesIO
 
+from reportlab.graphics import renderPDF
 from reportlab.lib.units import mm
 from reportlab.pdfgen.canvas import Canvas
 from reportlab.platypus import Paragraph, SimpleDocTemplate, Frame
 from reportlab_qrcode import QRCodeImage
+from svglib.svglib import svg2rlg
 
+from constants import STATIC_PATH
 from generators.base_pdf_card_generator.base_pdf_card_generator import BasePdfCardGenerator
 from services.warehouse_management import WmsService
 
 
 class WarehousePlaceQRCodeGenerator(BasePdfCardGenerator):
+    @staticmethod
+    def _get_big_font_size_from_str(value: str | int) -> int:
+        value = str(value)
+        dots = value.count(".")
+        string = value.replace(".", "")
+        length = len(string) + dots / 3
+
+        if length > 8:
+            return 30
+        elif length > 5:
+            return 37
+        return 48
+
+    @staticmethod
+    def _get_mid_font_size_from_str(value: str | int) -> int:
+        value = str(value)
+        dots = value.count(".")
+        string = value.replace(".", "")
+        length = len(string) + dots / 3
+
+        if length > 8:
+            return 22
+        elif length > 6:
+            return 26
+        elif length > 4:
+            return 30
+        return 34
+
     async def generate(self, place_id: int, is_short: bool) -> BytesIO:
+        icon_path = os.path.join(STATIC_PATH, "icons", "denco.svg")
+
         service = WmsService(self._session)
         place_code = await service.get_code_of_place(place_id)
 
@@ -18,25 +52,28 @@ class WarehousePlaceQRCodeGenerator(BasePdfCardGenerator):
         doc: SimpleDocTemplate = self._create_doc(buffer)
 
         def on_first_page(canvas: Canvas, doc):
-            denco_paragraph = Paragraph('DENCO', self.small_centered_style)
+            svg_icon_scale = 0.04
+            drawing = svg2rlg(icon_path)
+            drawing.scale(svg_icon_scale, svg_icon_scale)
 
             if is_short:
-                qr = QRCodeImage(place_code, size=20 * mm)
-                qr.drawOn(canvas, 39 * mm, -4)
+                qr = QRCodeImage(place_code, size=17 * mm)
+                qr.drawOn(canvas, 42 * mm, -4)
 
-                number_paragraph = Paragraph(str(place_code), self._get_paragraph_style(26))
-                number_frame = Frame(x1=0 * mm, y1=-15 * mm, width=42 * mm, height=30 * mm)
-                denco_frame = Frame(x1=0 * mm, y1=-2 * mm, width=42 * mm, height=20 * mm)
+                font_size = self._get_mid_font_size_from_str(place_code)
+                number_paragraph = Paragraph(str(place_code), self._get_paragraph_style(font_size))
+                number_frame = Frame(x1=0 * mm, y1=-15 * mm, width=45 * mm, height=30 * mm)
+                renderPDF.draw(drawing, canvas, 3, 35)
             else:
-                qr = QRCodeImage(place_code, size=25 * mm)
-                qr.drawOn(canvas, 34 * mm, -4)
+                qr = QRCodeImage(place_code, size=20 * mm)
+                qr.drawOn(canvas, 39 * mm, -2 * mm)
 
-                number_paragraph = Paragraph(str(place_code), self._get_paragraph_style(34))
+                font_size = self._get_big_font_size_from_str(place_code)
+                number_paragraph = Paragraph(str(place_code), self._get_paragraph_style(font_size))
                 number_frame = Frame(x1=0 * mm, y1=9 * mm, width=60 * mm, height=30 * mm)
-                denco_frame = Frame(x1=0 * mm, y1=-13 * mm, width=58 * mm, height=20 * mm)
+                renderPDF.draw(drawing, canvas, 67, 3)
 
             number_frame.addFromList([number_paragraph], canvas)
-            denco_frame.addFromList([denco_paragraph], canvas)
 
         empty_paragraph = Paragraph("", self.small_centered_style)
         elements = [empty_paragraph]