chore: 优化一下

MeoProject · Aug 18, 2024 · 089d1c2 · 089d1c2
1 parent 8919767
commit 089d1c2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/common/default_config.py b/common/default_config.py
@@ -181,7 +181,6 @@
             value: auto
         refresh_login: # cookie池中对于此账号刷新登录的配置，账号间互不干扰
           enable: false
-          interval: 604800
           login_url: http://login.user.kugou.com/v4/login_by_token
 
     tx:

diff --git a/modules/kg/refresh_token.py b/modules/kg/refresh_token.py
@@ -141,7 +141,7 @@ async def refresh_login_for_pool(user_info):
             'KG-Rec': '1',
             'KG-RC': '1',
         }
-        login_url = config.read_config('module.kg.user.refresh_token.login_url')
+        login_url = user_info["refresh_login"]["login_url"]
         req = await signRequest(login_url, params, {'method': 'POST', 'json': data, 'headers': headers})
         body = req.json()
         if body['error_code'] != 0:
@@ -180,7 +180,7 @@ async def refresh_login_for_pool(user_info):
             'KG-Rec': '1',
             'KG-RC': '1',
         }
-        login_url = config.read_config('module.kg.user.refresh_token.login_url')
+        login_url = user_info["refresh_login"]["login_url"]
         req = await signRequest(login_url, params, {'method': 'POST', 'json': data, 'headers': headers})
         body = req.json()
         if body['error_code'] != 0:
@@ -203,7 +203,7 @@ def reg_refresh_login_pool_task():
     for user_info in user_info_pool:
         if (user_info['refresh_login'].get('enable')):
             scheduler.append(
-                f'kgmusic_refresh_login_pooled_{user_info["userid"]}', refresh_login_for_pool, user_info['refresh_login']['interval'], args = {'user_info': user_info})
+                f'kgmusic_refresh_login_pooled_{user_info["userid"]}', refresh_login_for_pool, int(604800), args = {'user_info': user_info})
 
 if (variable.use_cookie_pool):
     reg_refresh_login_pool_task()