From 80270f9dbd7bff33bc2e137b0e0d213b524e13b6 Mon Sep 17 00:00:00 2001
From: zhao-zg <61549925+zhao-zg@users.noreply.github.com>
Date: Tue, 30 Jul 2024 10:33:07 +0800
Subject: [PATCH] Update api.py

---
 api.py | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/api.py b/api.py
index 701b161..de12ed2 100644
--- a/api.py
+++ b/api.py
@@ -96,9 +96,29 @@ async def loginPhone():
     r = mr("pass", uid=u.uid, msg=f"{u.account}处理中, 到/check查询处理进度")
     return r
 # 传入账号密码，启动登录线程
-#@app.route("/loginPassword", methods=["POST"])
-#async def login():
-#    loginPassword()
+@app.route("/login", methods=["POST"])
+async def login():
+    data = await request.get_json()
+    try:
+        u = account(data)
+    except Exception as e:
+        r = mr("error", msg=str(e))
+        return r
+    # 检测重复提交
+    if workList.get(u.uid):
+        workList[u.uid].SMS_CODE = None
+        r = mr("pass", uid=u.uid, msg=f"{u.account}已经在处理了，请稍后再试")
+        return r
+
+    # 新增记录
+    workList[u.uid] = u
+    # 非阻塞启动登录线程
+    asyncio.create_task(THREAD_DO_LOGIN(workList, u.uid, ocr))
+    # 更新信息，响应api请求
+    workList[u.uid].status = "pending"
+    r = mr("pass", uid=u.uid, msg=f"{u.account}处理中, 到/check查询处理进度")
+    return r
+    
 # 传入账号密码，启动登录线程
 @app.route("/loginPassword", methods=["POST"])
 async def loginPassword():