queue-service/main.py at main · Klenin-team/queue-service · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from fastapi import FastAPI
from fastapi.responses import JSONResponse
from pydantic import BaseModel
from typing import List
from queue import Queue

testsQueue = FastAPI()
queue = Queue()


class BackendRequest(BaseModel):
    id: str
    code: str
    language: str
    tests: List[List[str]]
    stdio: bool
    input_file: str
    output_file: str


class BackendResponse(BaseModel):
    status: bool


class RunnerResponse(BackendResponse):
    any: bool
    id: str
    code: str
    language: str
    tests: List[List[str]]
    stdio: bool
    input_file: str
    output_file: str


@testsQueue.post("/solution", response_model=BackendResponse)
def queue_put(test: BackendRequest):
    queue.put(test)
    return JSONResponse({"status": True}, status_code=200)


@testsQueue.get("/solution", response_model=RunnerResponse)
def queue_get():
    if queue.empty():
        return JSONResponse({
            "any": False
        }, status_code=200)
    else:
        cur = queue.get()
        return JSONResponse({
            "any": True,
            "id": cur.id,
            "code": cur.code,
            "language": cur.language,
            "tests": cur.tests,
            "stdio": cur.stdio,
            "input_file": cur.input_file,
            "output_file": cur.output_file
        }, status_code=200)