github/XHS_Downloader
1
0
Fork 0
mirror of https://github.com/JoeanAmier/XHS-Downloader.git synced 2025-12-26 04:48:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码复用性

Browse Source
This commit is contained in:
JoeamAmier 2023-09-01 19:41:44 +08:00
parent c5a3e9e4f6
commit 674e0db065
1 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
source/__init__.py
View File

@ -30,32 +30,43 @@ class XHS:
self.image = Image()
self.video = Video()
self.explore = Explore()
self.download = Download(self.manager, path, folder, self.headers, proxies, chunk)
self.download = Download(
self.manager,
path,
folder,
self.headers,
proxies,
chunk)
def get_image(self, container: dict, html: str, download):
def __get_image(self, container: dict, html: str, download):
urls = self.image.get_image_link(html)
if download:
self.download.run(urls, container["作品ID"])
self.download.run(urls, self.__naming_rules(container))
container["下载地址"] = urls
def get_video(self, container: dict, html: str, download):
def __get_video(self, container: dict, html: str, download):
url = self.video.get_video_link(html)
if download:
self.download.run(url, container["作品ID"])
self.download.run(url, self.__naming_rules(container))
container["下载地址"] = url
def extract(self, url: str, download=False) -> dict:
if not self.check(url):
if not self.__check(url):
return {}
html = self.html.get_html(url)
if not html:
return {}
data = self.explore.run(html)
if data["作品类型"] == "视频":
self.get_video(data, html, download)
self.__get_video(data, html, download)
else:
self.get_image(data, html, download)
self.__get_image(data, html, download)
return data
def check(self, url: str):
def __check(self, url: str):
return self.links.match(url)
@staticmethod
def __naming_rules(data: dict) -> str:
"""下载文件默认使用作品 ID 作为文件名，可修改此方法自定义文件名格式"""
return data["作品ID"]