diff --git a/req.py b/req.py index 9981f76..72cd96b 100644 --- a/req.py +++ b/req.py @@ -1,7 +1,11 @@ import requests +import urllib3 +import random +import re +import time from retrying import retry -requests.packages.urllib3.disable_warnings() +urllib3.disable_warnings() # 静态变量 BASE_URL = 'https://pan.baidu.com' @@ -54,7 +58,8 @@ class ReqAction: with open('cookie.txt', 'r', encoding='utf-8') as f: self.request_header['Cookie'] = f.readline() - @retry(stop_max_attempt_number=3, wait_fixed=1000) + # 获取bdstoken函数 + @retry(stop_max_attempt_number=3, wait_fixed=random.randint(1000, 3000)) def get_bdstoken(self): url = f'{BASE_URL}/api/gettemplatevariable?clienttype=0&app_id=250528&web=1&fields=[%22bdstoken%22,%22token%22,%22uk%22,%22isdocuser%22,%22servertime%22]' response = self.session.get(url=url, headers=self.request_header, timeout=20, allow_redirects=True, @@ -62,9 +67,96 @@ class ReqAction: print(response.text) return response.json()['errno'] if response.json()['errno'] != 0 else response.json()['result']['bdstoken'] + # 获取目录列表函数 + @retry(stop_max_attempt_number=5, wait_fixed=random.randint(1000, 3000)) + def get_dir_list(self, dir_path): + url = f'{BASE_URL}/api/list?order=time&desc=1&showempty=0&web=1&page=1&num=1000&dir={dir_path}&bdstoken={self.bdstoken}' + response = self.session.get(url=url, headers=self.request_header, timeout=15, allow_redirects=False, + verify=False) + print(response.text) + return response.json()['errno'] if response.json()['errno'] != 0 else response.json()['list'] + + # 新建目录函数 + @retry(stop_max_attempt_number=5, wait_fixed=random.randint(1000, 3000)) + def create_dir(self, target_directory_name): + url = f'{BASE_URL}/api/create?a=commit&bdstoken={self.bdstoken}' + post_data = {'path': target_directory_name, 'isdir': '1', 'block_list': '[]', } + response = self.session.post(url=url, headers=self.request_header, data=post_data, timeout=15, + allow_redirects=False, verify=False) + print(response.text) + return response.json()['errno'] + + # 更新 cookie 函数 + def update_cookie(self, bdclnd): + if 'BDCLND=' in self.request_header['Cookie']: + self.request_header['Cookie'] = re.sub(r'BDCLND=(\S+);?', f'BDCLND={bdclnd};', + self.request_header['Cookie']) + else: + self.request_header['Cookie'] += f';BDCLND={bdclnd}' + + # 验证提取码函数 + @retry(stop_max_attempt_number=6, wait_fixed=1700) + def verify_pass_code(self, link_url, pass_code): + check_url = f'{BASE_URL}/share/verify?surl={link_url[25:48]}&bdstoken={self.bdstoken}&t={str(int(round(time.time() * 1000)))}&channel=chunlei&web=1&clienttype=0' + post_data = {'pwd': pass_code, 'vcode': '', 'vcode_str': '', } + response = self.session.post(url=check_url, headers=self.request_header, data=post_data, timeout=10, + allow_redirects=False, verify=False) + print(response.text) + return response.json()['errno'] if response.json()['errno'] != 0 else response.json()['randsk'] + + # 验证链接函数 + @retry(stop_max_attempt_number=12, wait_fixed=1700) + def verify_links(self, link_url, pass_code): + if pass_code: + bdclnd = self.verify_pass_code(link_url, pass_code) + if isinstance(bdclnd, int): + return bdclnd + self.update_cookie(bdclnd) + + response = self.session.get(url=link_url, headers=self.request_header, timeout=15, allow_redirects=True, + verify=False).content.decode("utf-8") + print(response) + shareid_list = re.findall('"shareid":(\\d+?),"', response) + user_id_list = re.findall('"share_uk":"(\\d+?)","', response) + fs_id_list = re.findall('"fs_id":(\\d+?),"', response) + info_title_list = re.findall('(.+)', response) + + if not shareid_list: + return 1 + elif not user_id_list: + return 2 + elif not fs_id_list: + return info_title_list[0] if info_title_list else 3 + else: + return [shareid_list[0], user_id_list[0], fs_id_list] + + # 转存文件函数 + @retry(stop_max_attempt_number=9, wait_fixed=random.randint(1000, 3000)) + def transfer_files(self, verify_links_reason, target_directory_name): + url = f'{BASE_URL}/share/transfer?shareid={verify_links_reason[0]}&from={verify_links_reason[1]}&bdstoken={self.bdstoken}&channel=chunlei&web=1&clienttype=0' + post_data = {'fsidlist': f'[{",".join(i for i in verify_links_reason[2])}]', + 'path': f'/{target_directory_name}', } + response = self.session.post(url=url, headers=self.request_header, data=post_data, timeout=15, + allow_redirects=False, verify=False) + print(response.text) + return response.json()['errno'] + def main(self): + target_dir = '/Temp/leshe/1' + dir_name = '2' + link_list = [] self.get_cookies() self.bdstoken = self.get_bdstoken() + dir_list = self.get_dir_list('/Temp/leshe') + print([_dir['path'] for _dir in dir_list]) + if target_dir in [_dir['path'] for _dir in dir_list]: + print('找到了') + # if dir_name and dir_name not in [_dir('path') for _dir in self.dir_list]: + # self.create_dir(r'/Temp/leshe/1') + # 执行转存 + for link in link_list: + verified_links = self.verify_links(link[0], link[1]) + self.transfer_files(verified_links, target_dir) if __name__ == '__main__': diff --git a/requirement.txt b/requirement.txt new file mode 100644 index 0000000..fa86397 Binary files /dev/null and b/requirement.txt differ