From 4236d3a7cf607299dbbf0d52a533e947fd6c38ef Mon Sep 17 00:00:00 2001 From: Egor Matveev Date: Sat, 28 Dec 2024 15:33:18 +0300 Subject: [PATCH] fix --- Dockerfile | 2 +- app/routers/take.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/Dockerfile b/Dockerfile index d905ed4..ba79138 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,6 +8,6 @@ RUN pip install -r requirements.txt COPY . . -ENV PYTHONUNBUFFERED 1 +ENV PYTHONUNBUFFERED=1 ENTRYPOINT ["python", "main.py"] diff --git a/app/routers/take.py b/app/routers/take.py index 1f6c3a9..53a1ecb 100644 --- a/app/routers/take.py +++ b/app/routers/take.py @@ -23,9 +23,9 @@ class Response(pydantic.BaseModel): async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response: try: task = await tasks.take_task(queue) + if not task: + return Response(task=None) + return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload)) except Exception as e: - print('GOT ERROR', e) + print("GOT ERROR", e) return Response(task=None) - if not task: - return Response(task=None) - return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload))