feat: login via cookie (#47)

Author: timerring

bilitool/cli.py

@@ -37,6 +37,7 @@ def cli():
 
     # Login subcommand
     login_parser = subparsers.add_parser('login', help='Login and save the cookie')
+    login_parser.add_argument('-f', '--file', default='', help='(default is empty) Login via cookie file')
     login_parser.add_argument('--export', action='store_true', help='(default is false) Export the login cookie file')
 
     # Logout subcommand
@@ -98,7 +99,10 @@ def cli():
         sys.exit()
 
     if args.subcommand == 'login':
-        LoginController().login_bilibili(args.export)
+        if args.file:
+            LoginController().login_bilibili_with_cookie_file(args.file)
+        else:
+            LoginController().login_bilibili(args.export)
 
     if args.subcommand == 'logout':
         LoginController().logout_bilibili()

bilitool/controller/login_controller.py

@@ -28,3 +28,6 @@ def logout_bilibili(self):
 
     def check_bilibili_login(self):
         return self.check_bili_login.check_bili_login()
+
+    def login_bilibili_with_cookie_file(self, filename):
+        self.login_bili.get_cookie_file_login(filename)

bilitool/login/login_bili.py

@@ -6,6 +6,7 @@
 import json
 from urllib.parse import urlencode
 from bilitool.model.model import Model
+from bilitool.login.check_bili_login import CheckBiliLogin
 
 
 class LoginBili(object):
@@ -79,3 +80,19 @@ def verify_login(self, auth_code, export):
                 break
             else:
                 time.sleep(3)
+
+    def get_cookie_file_login(self, filename):
+        with open(filename, 'r', encoding='utf-8') as f:
+            body = json.load(f)
+            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']
+            sid_value = body['data']['cookie_info']['cookies'][4]['value']
+            Model().save_cookies_info(access_key_value, sessdata_value, bili_jct_value, dede_user_id_value, dede_user_id_ckmd5_value, sid_value)
+            if CheckBiliLogin().check_bili_login():
+                print("Login success!", flush=True)
+            else:
+                print("Login failed, please check the cookie file again", flush=True)
+