first commit

.gitignore

@@ -0,0 +1,2 @@
+venv/
+**/__pycache__/

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.12
+
+WORKDIR /app
+
+COPY ./requirements.txt /app/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "app.src.main:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -0,0 +1,45 @@
+annotated-types==0.6.0
+anyio==4.0.0
+certifi==2023.7.22
+click==8.1.7
+colorama==0.4.6
+distlib==0.3.7
+dnspython==2.4.2
+email-validator==2.1.0.post1
+exceptiongroup==1.1.3
+fastapi==0.108.0
+filelock==3.12.4
+h11==0.14.0
+httpcore==0.17.3
+httptools==0.6.1
+httpx==0.24.1
+idna==3.4
+itsdangerous==2.1.2
+Jinja2==3.1.2
+MarkupSafe==2.1.3
+netifaces==0.11.0
+orjson==3.9.10
+packaging==23.2
+pillow==10.2.0
+ping3==4.0.4
+platformdirs==3.10.0
+psutil==5.9.6
+pydantic==2.5.3
+pydantic-extra-types==2.4.0
+pydantic-settings==2.1.0
+pydantic_core==2.14.6
+pydentic==0.0.1.dev3
+python-dotenv==1.0.0
+python-multipart==0.0.6
+python-stdnum==1.19
+PyYAML==6.0.1
+qrcode-artistic==3.0.2
+segno==1.6.0
+sniffio==1.3.0
+starlette==0.32.0.post1
+typing_extensions==4.9.0
+ujson==5.9.0
+uvicorn==0.25.0
+virtualenv==20.24.5
+watchfiles==0.21.0
+websockets==12.0

src/main.py

@@ -0,0 +1,9 @@
+from fastapi import FastAPI
+
+from qr_generate.router import router as qr_generate_router
+
+app = FastAPI(
+    title="Animated QR Maker"
+)
+
+app.include_router(qr_generate_router)

src/qr_generate/router.py

@@ -0,0 +1,47 @@
+import base64
+import io
+from fastapi import APIRouter
+from qr_generate.schemas import QRCode, StaticQRCodeRequest, StaticQRCodeResponseModel, AnimatedQRCodeRequest, AnimatedQRCodeResponseModel
+from urllib.request import urlopen
+
+import segno
+
+router = APIRouter(
+    prefix="/qr_generate",
+    tags=["Generate QR Code"]
+)
+
+
+@router.post("/static", response_model=StaticQRCodeResponseModel)
+def generate_static_qr_code(qr_request: StaticQRCodeRequest):
+    qrcode = segno.make_qr(qr_request.content)
+    image_buffer = io.BytesIO()
+
+    qrcode.save(image_buffer,
+                kind="png",
+                scale=qr_request.scale,
+                dark=(qr_request.qr_code_color.red, qr_request.qr_code_color.green, qr_request.qr_code_color.blue),
+                border=qr_request.border_size,
+                light=(qr_request.border_color.red, qr_request.border_color.green, qr_request.border_color.blue),
+                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()),
+                                                                   content=qr_request.content))
+    
+@router.post("/animated", response_model=AnimatedQRCodeResponseModel)
+def generate_animated_qr_code(qr_request: AnimatedQRCodeRequest):
+        qrcode = segno.make_qr(qr_request.content)
+        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'
+        )
+        
+        image_buffer.seek(0)
+        return AnimatedQRCodeResponseModel(status="success", data=QRCode(image_bytes=base64.b64encode(image_buffer.read()),
+                                                                        content=qr_request.content))

src/qr_generate/schemas.py

@@ -0,0 +1,32 @@
+from util.rgb_model import RGBModel
+from pydantic import BaseModel
+
+class QRCode(BaseModel):
+    content: str
+    image_bytes: str
+
+### models for generate static(without video) qr code / png
+class StaticQRCodeRequest(BaseModel):
+    content: str
+    border_size: int = 1
+    scale: int = 15
+    qr_code_color: RGBModel = RGBModel(red=0, green=0, blue=0)
+    data_color: RGBModel = RGBModel(red=0, green=0, blue=0)
+    border_color: RGBModel
+
+class StaticQRCodeResponseModel(BaseModel):
+    status: str
+    data: QRCode
+### end
+
+
+### models for generate static(without video) qr code / png
+class AnimatedQRCodeRequest(BaseModel):
+    content: str
+    scale: int = 15
+    background_video_link: str
+
+class AnimatedQRCodeResponseModel(BaseModel):
+    status: str
+    data: QRCode
+### end

src/util/rgb_model.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+
+class RGBModel(BaseModel):
+    red: int = 255
+    green: int = 255
+    blue: int = 255