feat: add access key

timerring · Feb 17, 2025 · 3215c3b · 3215c3b
1 parent df1303a
commit 3215c3b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/bilitool/login/login_bili.py b/bilitool/login/login_bili.py
@@ -68,11 +68,12 @@ def verify_login(self, auth_code, export):
                         json.dump(body, f, ensure_ascii=False, indent=4)
                     print(f"cookie has been saved to {filename}")
 
+                access_key_value = body['data']['access_token']
                 sessdata_value = body['data']['cookie_info']['cookies'][0]['value']
                 bili_jct_value = body['data']['cookie_info']['cookies'][1]['value']
                 dede_user_id_value = body['data']['cookie_info']['cookies'][2]['value']
                 dede_user_id_ckmd5_value = body['data']['cookie_info']['cookies'][3]['value']
-                Model().save_cookies_info(sessdata_value, bili_jct_value, dede_user_id_value, dede_user_id_ckmd5_value)
+                Model().save_cookies_info(access_key_value, sessdata_value, bili_jct_value, dede_user_id_value, dede_user_id_ckmd5_value)
                 print("Login success!")
                 break
             else:

diff --git a/bilitool/model/config.json b/bilitool/model/config.json
@@ -6,7 +6,8 @@
         "SESSDATA": "",
         "bili_jct": "",
         "DedeUserID": "",
-        "DedeUserID__ckMd5": ""
+        "DedeUserID__ckMd5": "",
+        "access_key": ""
     },
     "upload": {
         "line": "bda2",

diff --git a/bilitool/model/model.py b/bilitool/model/model.py
@@ -67,8 +67,9 @@ def get_headers_with_cookies(self):
     def get_headers_with_cookies_and_refer(self):
         return self.get_config()['headers']
 
-    def save_cookies_info(self, sessdata, bili_jct, dede_user_id, dede_user_id_ckmd5):
+    def save_cookies_info(self, access_key, sessdata, bili_jct, dede_user_id, dede_user_id_ckmd5):
         config_info = self.get_config()
+        config_info['cookies']['access_key'] = access_key
         config_info['cookies']['SESSDATA'] = sessdata
         config_info['cookies']['bili_jct'] = bili_jct
         config_info['cookies']['DedeUserID'] = dede_user_id
@@ -88,6 +89,7 @@ def update_multiple_config(self, action, updates: dict):
 
     def reset_cookies(self):
         config_info = self.get_config()
+        config_info['cookies']['access_key'] = ""
         config_info['cookies']['SESSDATA'] = ""
         config_info['cookies']['bili_jct'] = ""
         config_info['cookies']['DedeUserID'] = ""