From c6c00321b042b6c594e849ced50730874a5bc245 Mon Sep 17 00:00:00 2001
From: 7a1dd609-d238-4580-9d5f-ee8412b0f5bc
 <7a1dd609-d238-4580-9d5f-ee8412b0f5bc@bankofchina.com>
Date: Wed, 15 Jan 2025 23:52:36 +0800
Subject: [PATCH] feat: add login test fun in unlock

---
 main.py | 25 ++++++++++++++++++++-----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/main.py b/main.py
index a720dc3..549b92d 100644
--- a/main.py
+++ b/main.py
@@ -202,11 +202,26 @@ def unlock_process():
         return jsonify({"apiName": "unlock", "apiInfo": "Unlock All DX Master Charts.", "apiUsage": "/unlock?userid=<userId>"})
     else:
         userId = int(userId)
-        unlock(userId)
-        returnCode = 200
-        status = "200 OK"
-        info = "Succeed."
-        log = {"UserLoginApiStatus": 1, "UploadUserPlaylogApiStatus": 1, "UpsertUserAllApi": 1, "UserLogoutApiStatus": 1}
+        timestamp = int(time.time())
+        login_data = login(userId, timestamp)
+        if login_data['returnCode'] == 102:
+            returnCode = 403
+            status = "403 Forbidden"
+            info = "Unable to operate. Please refresh QrCode."
+            log = {}
+        elif login_data['returnCode'] == 100:
+            returnCode = 403
+            status = "403 Forbidden"
+            info = "Unable to operate. Probably User has logged in."
+            log = {}
+        elif login_data['returnCode'] == 1:
+            userdata_result = userdata(userId)
+            logout(userId, timestamp)
+            unlock(userId)
+            returnCode = 200
+            status = "200 OK"
+            info = "Succeed."
+            log = {"UserLoginApiStatus": 1, "UploadUserPlaylogApiStatus": 1, "UpsertUserAllApi": 1, "UserLogoutApiStatus": 1}
         data = jsonify({"status": status, "timestamp": timestamp, "info": info, "apiName": "unlock", "date": datetime.now(pytz.timezone('Asia/Shanghai')).strftime('%a, %d %b %Y %H:%M:%S GMT+8'), "userId": userId})
         return data