From 064b1656786172189ce4be29dd9a96cc5dbafefb Mon Sep 17 00:00:00 2001 From: DingZQ Date: Sun, 19 Feb 2023 17:12:54 +0800 Subject: [PATCH] Add method to lock/unlock --- main.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/main.py b/main.py index 9224067..a482e43 100644 --- a/main.py +++ b/main.py @@ -100,17 +100,20 @@ async def fastapi_dump_output(output: str) -> str: return dump_output(output) @app.get("/isprojectlocked/") -async def fastapi_is_locked(network: str): - return lockedPrjs.get(network) != None +async def fastapi_is_locked(network: str, req: Request): + client_host = req.client.host + return lockedPrjs.get(network) != client_host @app.post("/lockproject/") -async def fastapi_lock_project(network: str, id: str): - lockedPrjs[network] = id - return True +async def fastapi_lock_project(network: str, req: Request): + client_host = req.client.host + lockedPrjs[network] = client_host + return client_host @app.post("/unlockproject/") -async def fastapi_unlock_project(network: str, id: str): - if lockedPrjs[network] == id: +async def fastapi_unlock_project(network: str, req: Request): + client_host = req.client.host + if lockedPrjs[network] == client_host: del lockedPrjs[network] return True