self/queues
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

master #40

Merged
emmatveev merged 20 commits from master into dev 2024-12-29 13:32:45 +03:00
Conversation 0 Commits 20 Files Changed 7 +5 -5
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 4236d3a7cf - Show all commits

2
Dockerfile
View File

@ -8,6 +8,6 @@ RUN pip install -r requirements.txt
COPY . . COPY . .
ENV PYTHONUNBUFFERED 1 ENV PYTHONUNBUFFERED=1
ENTRYPOINT ["python", "main.py"] ENTRYPOINT ["python", "main.py"]

8
app/routers/take.py
View File

@ -23,9 +23,9 @@ class Response(pydantic.BaseModel):
async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response: async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response:
try: try:
task = await tasks.take_task(queue) 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: except Exception as e:
print('GOT ERROR', e) print("GOT ERROR", e)
return Response(task=None) 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))