Merge pull request #71 from codeguru42/66-fast-api

66 fast api

Author: codeguru42

docker-compose-dev.yml

@@ -1,7 +1,7 @@
 services:
   api:
     build: .
-    command: "poetry run python3 /api/manage.py runserver 0.0.0.0:8000"
+    command: "poetry run uvicorn main:app --host 0.0.0.0 --port 8000"
     env_file:
       - env/dev.env
     environment:
@@ -16,7 +16,7 @@ services:
       - .:/api
   worker:
     build: .
-    command: "poetry run celery -A go_capture.tasks worker"
+    command: "poetry run celery -A tasks worker"
     depends_on:
       - broker
     environment:

docker-compose-prod.yml

@@ -1,6 +1,6 @@
 services:
   api:
-    command: "poetry run gunicorn go_capture.wsgi:application --bind 0.0.0.0:8000 -t 60"
+    command: "poetry run uvicorn main:app --host 0.0.0.0 --port 8000"
     environment:
       - CELERY_BROKER_URL=amqp://guest:guest@broker:5672
       - FIREBASE_CREDENTIALS_FILE=/run/secrets/firebase-credentials
@@ -15,7 +15,7 @@ services:
     volumes:
       - media:/api/media
   worker:
-    command: "poetry run celery -A go_capture.tasks worker"
+    command: "poetry run celery -A tasks worker"
     depends_on:
       - broker
     environment:

go_capture/__init__.py

@@ -0,0 +1,38 @@
+import io
+from pathlib import Path
+from typing import Annotated
+
+from fastapi import FastAPI, File, UploadFile, Form
+
+import settings
+from sgf.process_image import process_image
+from tasks import process_image_task
+
+app = FastAPI()
+
+
+@app.get("/health_check/")
+def health_check():
+    return {"message": "Healthy!"}
+
+
+@app.post("/capture/")
+def capture(image: Annotated[UploadFile, File()]):
+    output_file = io.StringIO()
+    process_image(image.file, output_file)
+    output_file.seek(0)
+    return {
+        "sgf": output_file.read(),
+        "image_filename": image.filename,
+    }
+
+
+@app.post("/capture_async/")
+def capture_async(
+    image: Annotated[UploadFile, File()], fcm_registration_token: Annotated[str, Form()]
+):
+    filename = Path(image.filename)
+    output_path = settings.IMAGES_DIR / filename
+    with output_path.open("wb") as output_file:
+        output_file.write(image.file.read())
+    process_image_task.delay(str(output_path.absolute()), fcm_registration_token)