github/XHS_Downloader
1
0
Fork 0
mirror of https://github.com/JoeanAmier/XHS-Downloader.git synced 2026-03-22 06:57:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化项目代码

Browse Source
This commit is contained in:
yongquan
2024-01-09 19:18:49 +08:00
parent 039f9f9785
commit 0d22a8f4c0
9 changed files with 65 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
source/module/manager.py
View File

@@ -12,6 +12,7 @@ from aiohttp import ClientTimeout
from source.translator import Chinese
from source.translator import English
from .static import COOKIE
from .static import HEADERS
from .static import USERAGENT
__all__ = ["Manager"]
@@ -40,9 +41,9 @@ class Manager:
self.temp = root.joinpath("./temp")
self.path = self.__check_path(path)
self.folder = self.__check_folder(folder)
self.headers = {
"User-Agent": user_agent or USERAGENT,
"Cookie": cookie or COOKIE}
self.blank_headers = HEADERS | {
"User-Agent": user_agent or USERAGENT, }
self.headers = self.blank_headers | {"Cookie": cookie or COOKIE}
self.retry = retry
self.chunk = chunk
self.record_data = record_data
@@ -51,11 +52,11 @@ class Manager:
self.proxy = proxy
self.request_session = ClientSession(
headers=self.headers | {
"Referer": "https://www.xiaohongshu.com/", },
"Referer": "https://www.xiaohongshu.com/explore", },
timeout=ClientTimeout(connect=timeout),
)
self.download_session = ClientSession(
headers={"User-Agent": self.headers["User-Agent"]},
headers=self.blank_headers,
timeout=ClientTimeout(connect=timeout))
self.prompt = language