start to support client

DecryptLogin/login.py

@@ -35,24 +35,14 @@ class Login():
         return 'Welcome to use DecryptLogin!\nYou can visit https://github.com/CharlesPikachu/DecryptLogin for more details.'
 
 
-'''模拟登录器类-仅返回对应平台的实例化类'''
-class Loginer():
-    def __init__(self, **kwargs):
+'''返回对应网站的客户端'''
+class Client():
+    def __init__(self, disable_print_auth=False, **kwargs):
         if not disable_print_auth: print(self)
-        self.supported_apis = {
-            'douban': douban, 'weibo': weibo, 'github': github, 'music163': music163, 
-            'zt12306': zt12306, 'QQZone': QQZone, 'QQQun': QQQun, 'QQId': QQId, 
-            'zhihu': zhihu, 'bilibili': bilibili, 'toutiao': toutiao, 'taobao': taobao, 
-            'jingdong': jingdong, 'ifeng': ifeng, 'sohu': sohu, 'zgconline': zgconline, 
-            'lagou': lagou, 'twitter': twitter, 'eSurfing': eSurfing, 
-            'renren': renren, 'w3cschool': w3cschool, 'fishc': fishc, 'youdao': youdao,
-            'baidupan': baidupan, 'stackoverflow': stackoverflow, 'codalab': codalab, 'pypi': pypi, 
-            'douyu': douyu, 'migu': migu, 'qunar': qunar, 'mieshop': mieshop, 'mpweixin': mpweixin, 
-            'baidutieba': baidutieba, 'dazhongdianping': dazhongdianping, 'jianguoyun': jianguoyun, 
-            'cloud189': cloud189, 'qqmusic': qqmusic, 'ximalaya': ximalaya, 'icourse163': icourse163, 
-            'xiaomihealth': xiaomihealth,
+        self.supported_clients = {
+            'bilibili': BiliBiliClient, 
         }
-        for key, value in self.supported_apis.items():
+        for key, value in self.supported_clients.items():
             setattr(self, key, value)
     '''str'''
     def __str__(self):

DecryptLogin/modules/__init__.py

@@ -1,3 +1,4 @@
 '''initialize'''
 from .core import *
-from .utils import *
+from .utils import *
+from .clients import *

DecryptLogin/modules/clients/__init__.py

@@ -1 +1,2 @@
-'''initialize'''
+'''initialize'''
+from .bilibili import BiliBiliClient

DecryptLogin/modules/clients/baseclient.py

@@ -0,0 +1,77 @@
+'''
+Function:
+    基类客户端
+Author:
+    Charles
+微信公众号:
+    Charles的皮卡丘
+'''
+import os
+import pickle
+import requests
+from ..core import *
+
+
+'''基类客户端'''
+class BaseClient():
+    def __init__(self, website_name=None, reload_history=True, **kwargs):
+        self.supported_apis = {
+            'douban': douban().login, 'weibo': weibo().login, 'github': github().login, 'music163': music163().login, 
+            'zt12306': zt12306().login, 'QQZone': QQZone().login, 'QQQun': QQQun().login, 'QQId': QQId().login, 
+            'zhihu': zhihu().login, 'bilibili': bilibili().login, 'toutiao': toutiao().login, 'taobao': taobao().login, 
+            'jingdong': jingdong().login, 'ifeng': ifeng().login, 'sohu': sohu().login, 'zgconline': zgconline().login, 
+            'lagou': lagou().login, 'twitter': twitter().login, 'eSurfing': eSurfing().login, 
+            'renren': renren().login, 'w3cschool': w3cschool().login, 'fishc': fishc().login, 'youdao': youdao().login, 
+            'baidupan': baidupan().login, 'stackoverflow': stackoverflow().login, 'codalab': codalab().login, 'pypi': pypi().login, 
+            'douyu': douyu().login, 'migu': migu().login, 'qunar': qunar().login, 'mieshop': mieshop().login, 'mpweixin': mpweixin().login, 
+            'baidutieba': baidutieba().login, 'dazhongdianping': dazhongdianping().login, 'jianguoyun': jianguoyun().login, 
+            'cloud189': cloud189().login, 'qqmusic': qqmusic().login, 'ximalaya': ximalaya().login, 'icourse163': icourse163().login, 
+            'xiaomihealth': xiaomihealth().login,
+        }
+        assert website_name in self.supported_apis
+        self.rootdir = os.path.split(os.path.abspath(__file__))[0]
+        self.website_name = website_name
+        self.reload_history = reload_history
+        self.infos_return, self.session = {}, requests.Session()
+    '''模拟登录'''
+    def login(self, username='none', password='none', mode='pc', crack_captcha_func=None, **kwargs):
+        need_new_login = True
+        if self.reload_history:
+            self.infos_return, self.session, need_new_login = self.loadhistory(username)
+        if need_new_login:
+            api = self.supported_apis[self.website_name]
+            self.infos_return, self.session = api(username, password, mode, crack_captcha_func, **kwargs)
+        self.savehistory(username, self.infos_return, self.session)
+        return self.infos_return, self.session
+    '''保存历史数据'''
+    def savehistory(self, username, infos_return, session):
+        history_path = os.path.join(self.rootdir, self.website_name+'.pkl')
+        history_infos = {}
+        if os.path.exists(history_path):
+            fp = open(history_path, 'rb')
+            history_infos = pickle.load(fp)
+            fp.close()
+        history_infos[username] = [infos_return, session]
+        fp = open(history_path, 'wb')
+        pickle.dump(history_infos, fp)
+        fp.close()
+    '''导入历史数据'''
+    def loadhistory(self, username):
+        history_path = os.path.join(self.rootdir, self.website_name+'.pkl')
+        # 不存在历史文件
+        if not os.path.exists(history_path): return None, None, True
+        # 读取history文件
+        fp = open(history_path, 'rb')
+        history_infos = pickle.load(fp)
+        fp.close()
+        # 如果username不存在
+        if username not in history_infos: return None, None, True
+        # 提取对应的数据
+        infos_return, session = history_infos[username]
+        # 检查是否已经过期
+        if self.checksessionstatus(session): return None, None, True
+        # 返回可用的数据
+        return infos_return, session, False
+    '''检查会话是否已经过期, 过期返回True'''
+    def checksessionstatus(self, session):
+        return True

DecryptLogin/modules/clients/bilibili.py

@@ -0,0 +1,25 @@
+'''
+Function:
+    B站客户端
+Author:
+    Charles
+微信公众号:
+    Charles的皮卡丘
+更新日期:
+    2022-03-11
+'''
+from .baseclient import BaseClient
+
+
+'''B站客户端'''
+class BiliBiliClient(BaseClient):
+    def __init__(self, reload_history=True, **kwargs):
+        super(BiliBiliClient, self).__init__(website_name='bilibili', reload_history=reload_history, **kwargs)
+    '''检查会话是否已经过期, 过期返回True'''
+    def checksessionstatus(self, session):
+        url = 'https://api.bilibili.com/x/web-interface/nav/stat'
+        response = session.get(url)
+        response_json = response.json()
+        if str(response_json['code']) == '0' and str(response_json['message']) == '0':
+            return False
+        return True