python + chromedriver 实现自动批量保存百度盘分享

长大1/2

python + chromedriver实现自动批量保存百度盘分享
已将提取百度云分享的Python代码封装成了类BaiduPanExtractor

1. from selenium import webdriver
   
2. import threading
   
3. import time
   
4. 
   
5. 
   
6. class BaiduPanExtractor:
   
7.     def __init__(self):
   
8.         self.browser = None
   
9.         self.password = None
   
10.         self.account = None
    
11.         self.executable_path = None
    
12.         self.login_status = False   # 记录登录状态
    
13. 
    
14.     def open_browser(self, executable_path="chromedriver"):
    
15.         '''
    
16.             :param executable_path 指定chromedriver的路径
    
17.         '''
    
18.         try:
    
19.             self.browser = webdriver.Chrome(executable_path=executable_path)
    
20.         except Exception:
    
21.             print("请指定Chromedriver.exe的路径")
    
22. 
    
23.     #   超过2s未检查出来指定元素,默认为该网页无该元素，抛出异常
    
24.     def decorator_exception(fun):
    
25.         def wrap(*args):
    
26.             sleep_time = 0
    
27.             while True:
    
28.                 time.sleep(1)
    
29.                 try:
    
30.                     return fun(*args)
    
31.                 except Exception as reason:
    
32.                     sleep_time = sleep_time + 1
    
33.                     if sleep_time >= 2:
    
34.                         raise reason
    
35.         return wrap
    
36. 
    
37.     #   打开网页百度云盘登录连接
    
38.     def open_baidu_pan(self):
    
39.         self.browser.get("http://pan.baidu.com")
    
40. 
    
41.     #   检查登录状态
    
42.     def check_login_status(self):
    
43.         while not self.login_status:
    
44.             if "https://pan.baidu.com/disk/home?" in self.browser.current_url:
    
45.                 self.login_status = True
    
46. 
    
47.     #   点击”保存到网盘“按钮
    
48.     @decorator_exception
    
49.     def click_save2disk(self):
    
50.         self.browser.find_element_by_class_name("tools-share-save-hb").click()
    
51. 
    
52.     #   点击”新建文件夹“按钮
    
53.     @decorator_exception
    
54.     def click_create_new_folder(self):
    
55.         self.browser.find_element_by_xpath("//a[@data-button-id='b17']").click()
    
56. 
    
57.     #   点击分享页面“确认”按钮
    
58.     @decorator_exception
    
59.     def click_confirm(self):
    
60.         self.browser.find_element_by_xpath("//a[@node-type='confirm']").click()
    
61. 
    
62.     #   点击分享页面“✔”按钮
    
63.     @decorator_exception
    
64.     def click_sure(self):
    
65.         self.browser.find_element_by_class_name("sure").click()
    
66. 
    
67.     #   将文件保存到“file_path”目录下
    
68.     def save(self, file_path):
    
69.         self.locate_folder(file_path)
    
70.         self.click_confirm()
    
71. 
    
72.     #   检查”save_path“路径是否存在
    
73.     def is_path_exist(self, save_path):
    
74.         return self.locate_folder(save_path)
    
75. 
    
76.     #   定位“file_name”文件
    
77.     def locate_folder(self, file_name):
    
78.         pattern = "//span[@node-path='{}']/../..".format(file_name)
    
79.         sleep_time = 0
    
80.         while True:
    
81.             time.sleep(1)
    
82.             try:
    
83.                 self.browser.find_element_by_xpath(pattern).click()
    
84.                 return True
    
85.             except Exception:
    
86.                 sleep_time = sleep_time + 1  # 定位文件时，如果2s内未找到该文件，视为目录中没有该文件
    
87.                 if sleep_time >= 2:
    
88.                     return False
    
89. 
    
90.     #   新建路径未file_path的文件夹
    
91.     @decorator_exception
    
92.     def create_new_folder(self, file_path):
    
93.         self.click_create_new_folder()  # 点击新建文件夹按钮
    
94.         temp = self.browser.find_element_by_class_name("shareFolderInput")  # 定位文件名输入框
    
95.         temp.clear()  # 清空输入框
    
96.         temp.send_keys(file_path[file_path.rfind("/"):])  # 填写文件名
    
97.         self.click_sure()  # 确认填写
    
98. 
    
99.     #   提交用户名和登录密码
    
100.     @decorator_exception
     
101.     def submit_login_form(self):
     
102.         username = self.browser.find_element_by_class_name("pass-text-input-userName")
     
103.         password = self.browser.find_element_by_class_name("pass-text-input-password")
     
104.         username.clear()
     
105.         password.clear()
     
106.         username.send_keys(self.account)
     
107.         password.send_keys(self.password)
     
108.         self.browser.find_element_by_id("TANGRAM__PSP_4__submit").click()
     
109.         while not self.login_status:
     
110.             pass
     
111. 
     
112.     #   打开分享界面
     
113.     def extract_share(self, share_url, share_url_code):
     
114.         self.browser.get(share_url)
     
115.         self.browser.find_element_by_id("accessCode").send_keys(share_url_code)
     
116.         self.browser.find_element_by_class_name("submit-a").click()
     
117.         current_url = self.browser.current_url
     
118.         while current_url == self.browser.current_url:
     
119.             pass
     
120. 
     
121.     #   将文件保存在save_path路径下
     
122.     def save_file(self, save_path):
     
123.         catalog_list = save_path.split("/")
     
124.         catalog_list.remove("")
     
125.         length = len(catalog_list)
     
126.         file_path = ""
     
127.         for i in range(length):
     
128.             file_path = file_path + "/" + catalog_list[i]
     
129.             if not self.is_path_exist(file_path):
     
130.                 self.create_new_folder(file_path)
     
131.             if file_path == save_path:
     
132.                 self.save(save_path)
     
133. 
     
134.     #   登录百度云盘
     
135.     def login(self,account=None,password=None):
     
136.         '''
     
137.             :param account 百度云的账户
     
138.             :param password 百度云账户密码
     
139.             若用户名或密码为空，则打开浏览器手动登录
     
140.             若用户名或密码不为空，则打开浏览器自动提交用户名和密码，（百度云可能会有验证，这时需要手动验证）
     
141.         '''
     
142.         self.account = account
     
143.         self.password = password
     
144.         self.open_baidu_pan()
     
145.         threading.Thread(target=self.check_login_status).start()
     
146.         while not self.login_status:
     
147.             if self.account and self.password:
     
148.                 self.submit_login_form()
     
149. 
     
150.     def check_share_type(self):
     
151.         if "#list" in self.browser.current_url:
     
152.             return "FOLDER"
     
153.         return "FILE"
     
154. 
     
155.     #   点击“文件名”按钮
     
156.     @decorator_exception
     
157.     def click_file_name_checkbox(self):
     
158.         self.browser.find_element_by_xpath("//div[@class='QxJxtg']/div/ul/li/div").click()
     
159. 
     
160.     #   将链接为share_url的文件保存在save_path路径下
     
161.     def save2baidu_pan_disk(self, share_url, share_url_code, save_path):
     
162.         '''
     
163.             :param share_url 分享连接
     
164.             :param share_url_code 分享连接提取码
     
165.             :param save_path 资源保存位置 默认为根路径
     
166.         '''
     
167.         self.extract_share(share_url, share_url_code)
     
168. 
     
169.         if self.check_share_type() == "FOLDER":
     
170.             self.click_file_name_checkbox()
     
171. 
     
172.         self.click_save2disk()
     
173.         self.save_file(save_path)

复制代码

使用方法：

1. chromedriver_executable = "指定你的chromedriver的路径"
   
2. account = "你的百度云"
   
3. password = "你的百度云密码"
   
4. 
   
5. baidupan_extractor = BaiduPanExtractor()
   
6. baidupan_extractor.open_browser(chromedriver_executable)
   
7. baidupan_extractor.login(account = account,password=password)
   
8. baidupan_extractor.save2baidu_pan_disk(share_url="百度云分享链接", share_url_code="百度云分享链接提取码",save_path="指定文件保存路径")

复制代码

账户和密码可以不用输入，运行代码后会打开浏览器，弹出百度网盘登录界面，可在该界面输入账户和密码登录并验证，之后浏览器会自动进入到你填写的百度云分享连接，并提取文件

写脚本的原因是，在github上发现某大神分享了很关于计算机知识的电子书



一个个手动点击保存过于繁琐于是写了个脚本将该页的链接和提取码爬取了下来并配合chromedriver实现模拟手动点击输入提取码，点击提取文件等过程。