fix： rrocr

mikumifa · Jun 19, 2024 · 3143907 · 3143907
1 parent 2fe3fab
commit 3143907
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tab/login.py b/tab/login.py
@@ -1,8 +1,8 @@
 import gradio as gr
 from loguru import logger
 
-from util.KVDatabase import KVDatabase
 from config import main_request, configDB, global_cookieManager
+from util.KVDatabase import KVDatabase
 
 names = []
 
@@ -38,7 +38,7 @@ def login_tab():
 
         def upload_file(filepath):
             main_request.cookieManager.db.delete("cookie")
-            yield ["已经注销，请选择登录信息文件", gr.update()]
+            yield ["已经注销，请选择登录信息文件", gr.update(), gr.update()]
             try:
                 configDB.insert("cookie_path", filepath)
                 global_cookieManager.db = KVDatabase(filepath)

diff --git a/util/CookieManager.py b/util/CookieManager.py
@@ -54,7 +54,9 @@ def _login_and_save_cookies(
         logger.info("登录成功, 浏览器退出.")
         return self.db.get("cookie")
 
-    def get_cookies(self):
+    def get_cookies(self, force=False):
+        if force:
+            return self.db.get("cookie")
         if not self.db.contains("cookie"):
             return self._login_and_save_cookies()
         else: