recreate animate qr code logic

src/qr_generate/router.py

@@ -1,12 +1,8 @@
 import base64
 import io
-import os
 from fastapi import APIRouter
 from qr_generate.schemas import QRCode, StaticQRCodeRequestModel, StaticQRCodeResponseModel, AnimatedQRCodeRequestModel, AnimatedQRCodeResponseModel
-from qrcode_artistic import write_artistic
-import random
-import string
-
+from urllib.request import urlopen
 import segno
 
 router = APIRouter(
@@ -34,31 +30,22 @@ def generate_static_qr_code(qr_request: StaticQRCodeRequestModel):
     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: AnimatedQRCodeRequestModel):
-    try:
         image_buffer = io.BytesIO()
-        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
+        
+        qrcode = segno.make_qr(qr_request.content)
+        video_url = urlopen(qr_request.gif_link)
+        
+        qrcode.to_artistic(
+            background=video_url,
+            target=image_buffer,
+            scale=qr_request.scale,
+            kind='gif'
+        )
+        
         image_buffer.seek(0)
         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)
+                                                                        content=qr_request.content))

src/qr_generate/schemas.py

@@ -24,7 +24,7 @@ class StaticQRCodeResponseModel(BaseModel):
 class AnimatedQRCodeRequestModel(BaseModel):
     content: str
     scale: int = 15
-    gif_base64: str
+    gif_link: str
 
 class AnimatedQRCodeResponseModel(BaseModel):
     status: str