From 87c9790d3b5ee32e0e0e68b2335ccf11224cc0fa Mon Sep 17 00:00:00 2001
From: DingZQ <zqding@hotmail.com>
Date: Fri, 24 Feb 2023 22:48:09 +0800
Subject: [PATCH] Refine

---
 main.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 49c9185..92128c5 100644
--- a/main.py
+++ b/main.py
@@ -121,13 +121,20 @@ async def fastapi_is_locked_by_me(network: str, req: Request):
     client_host = req.client.host
     return lockedPrjs.get(network) == client_host
 
+# 0 successfully locked
+# 1 already locked by you
+# 2 locked by others
 @app.post("/lockproject/")
 async def fastapi_lock_project(network: str, req: Request):
     client_host = req.client.host
     if not network in lockedPrjs.keys():
         lockedPrjs[network] = client_host
-        return client_host
-    return ""
+        return 0
+    else:
+        if lockedPrjs.get(network) == client_host:
+            return 1
+        else:
+            return 2
 
 @app.post("/unlockproject/")
 async def fastapi_unlock_project(network: str, req: Request):