recreate animate qr code logic

src/qr_generate/router.py

@@ -1,8 +1,11 @@
 import base64
 import io
+import os
 from fastapi import APIRouter
-from qr_generate.schemas import QRCode, StaticQRCodeRequest, StaticQRCodeResponseModel, AnimatedQRCodeRequest, AnimatedQRCodeResponseModel
-from urllib.request import urlopen
+from qr_generate.schemas import QRCode, StaticQRCodeRequestModel, StaticQRCodeResponseModel, AnimatedQRCodeRequestModel, AnimatedQRCodeResponseModel
+from qrcode_artistic import write_artistic
+import random
+import string
 
 import segno
 
@@ -13,7 +16,7 @@ router = APIRouter(
 
 
 @router.post("/static", response_model=StaticQRCodeResponseModel)
-def generate_static_qr_code(qr_request: StaticQRCodeRequest):
+def generate_static_qr_code(qr_request: StaticQRCodeRequestModel):
     qrcode = segno.make_qr(qr_request.content)
     image_buffer = io.BytesIO()
 
@@ -26,22 +29,36 @@ def generate_static_qr_code(qr_request: StaticQRCodeRequest):
                 data_dark=(qr_request.data_color.red, qr_request.data_color.green, qr_request.data_color.blue))
     
     image_buffer.seek(0)
-    return StaticQRCodeResponseModel(status="success", data=QRCode(image_bytes=base64.b64encode(image_buffer.read()),
+    base64_encoded_image = base64.b64encode(image_buffer.read())
+
+    return StaticQRCodeResponseModel(status="success", data=QRCode(image_base64=base64_encoded_image,
                                                                    content=qr_request.content))
     
+def generate_random_digit_string(length):
+    return ''.join(random.choice(string.digits) for _ in range(length))
+
 @router.post("/animated", response_model=AnimatedQRCodeResponseModel)
-def generate_animated_qr_code(qr_request: AnimatedQRCodeRequest):
-        qrcode = segno.make_qr(qr_request.content)
+def generate_animated_qr_code(qr_request: AnimatedQRCodeRequestModel):
+    try:
         image_buffer = io.BytesIO()
-        
-        video_url = urlopen(qr_request.background_video_link)
-        qrcode.to_artistic(
-            background=video_url,
-            target=image_buffer,
-            scale=qr_request.scale,
-            kind='gif'
-        )
-        
+        unique_picutre_name = generate_random_digit_string(30)
+
+        video_path = unique_picutre_name + ".gif"
+        print(video_path)
+        with open(video_path, "wb") as fh:
+            fh.write(base64.b64decode(qr_request.gif_base64))
+
+        # Создание QR-кода на основе видео
+        qrcode = segno.make(qr_request.content, error="h")
+        write_artistic(qrcode, background=video_path, scale=qr_request.scale, target=image_buffer, kind="gif")
+
+        # Перевод изображения в Base64
         image_buffer.seek(0)
-        return AnimatedQRCodeResponseModel(status="success", data=QRCode(image_bytes=base64.b64encode(image_buffer.read()),
-                                                                        content=qr_request.content))
+        base64_encoded_image = base64.b64encode(image_buffer.read())
+
+        return AnimatedQRCodeResponseModel(status="success", data=QRCode(image_base64=base64_encoded_image,
+                                                                    content=qr_request.content))
+    finally:
+        # Удаление временного файла
+        if os.path.exists(video_path):
+            os.remove(video_path)

src/qr_generate/schemas.py

@@ -3,10 +3,10 @@ from pydantic import BaseModel
 
 class QRCode(BaseModel):
     content: str
-    image_bytes: str
+    image_base64: str
 
 ### models for generate static(without video) qr code / png
-class StaticQRCodeRequest(BaseModel):
+class StaticQRCodeRequestModel(BaseModel):
     content: str
     border_size: int = 1
     scale: int = 15
@@ -20,11 +20,11 @@ class StaticQRCodeResponseModel(BaseModel):
 ### end
 
 
-### models for generate static(without video) qr code / png
-class AnimatedQRCodeRequest(BaseModel):
+### models for animate qr code / gif
+class AnimatedQRCodeRequestModel(BaseModel):
     content: str
     scale: int = 15
-    background_video_link: str
+    gif_base64: str
 
 class AnimatedQRCodeResponseModel(BaseModel):
     status: str

src/util/picture_type_enum.py

@@ -0,0 +1,5 @@
+from enum import Enum
+
+class PictureTypeEnum(Enum):
+    GIF = "gif",
+    PNG = "mp4"