From 866b789a3d0f114af2dd70f60258f1acfe85a3ba Mon Sep 17 00:00:00 2001
From: Jerry Yan <792602257@qq.com>
Date: Wed, 8 Apr 2020 15:55:58 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BE=8B=E8=A1=8C=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bilibili.py | 18 ++++++------------
 1 file changed, 6 insertions(+), 12 deletions(-)

diff --git a/bilibili.py b/bilibili.py
index c96ff33..5308abe 100644
--- a/bilibili.py
+++ b/bilibili.py
@@ -44,7 +44,7 @@ class Bilibili:
         :param pwd: password
         :type pwd: str
         :return: if success return True
-                 else return msg json
+                 else raise Exception
         """
         APPKEY = '4409e2ce8ffd12b8'
         ACTIONKEY = 'appkey'
@@ -122,20 +122,16 @@ class Bilibili:
                 'appkey={appkey}&build={build}&captcha=&channel=master&'
                 'guid=XYEBAA3E54D502E37BD606F0589A356902FCF&mobi_app={mobi_app}&'
                 'password={password}&platform={platform}&token=5598158bcd8511e2&ts=0&username={username}'
-                    .format(appkey=APPKEY, build=BUILD, platform=PLATFORM, mobi_app=MOBI_APP, username=user,
-                            password=pwd)),
+                .format(appkey=APPKEY, build=BUILD, platform=PLATFORM, mobi_app=MOBI_APP, username=user, password=pwd)),
         )
-        try:
-            json = r.json()
-        except:
-            return r.text
+        json = r.json()
 
         if json['code'] == -105:
             # need captcha
             raise Exception('TODO: login with captcha')
 
         if json['code'] != 0:
-            return r.text
+            raise Exception(r.text)
 
         access_token = json['data']['token_info']['access_token']
         cookie_dict = access_token_2_cookie(access_token)
@@ -144,14 +140,12 @@ class Bilibili:
             for k, v in cookie_dict.items()
         )
         self.session.headers["cookie"] = cookie
-
-        self.csrf = re.search('bili_jct=(.*?);', cookie).group(1)
-        self.mid = re.search('DedeUserID=(.*?);', cookie).group(1)
+        self.csrf = re.search('bili_jct=(.*?)(;|$)', cookie).group(1)
+        self.mid = re.search('DedeUserID=(.*?)(;|$)', cookie).group(1)
         self.session.headers['Accept'] = 'application/json, text/javascript, */*; q=0.01'
         self.session.headers['Referer'] = 'https://space.bilibili.com/{mid}/#!/'.format(mid=self.mid)
 
         return True
-
     def upload(self,
                parts,
                title,