fix: 修复下载私密账号作品时出现 404 响应码的问题

BREAKING CHANGE: 下载作品文件时将携带 Cookie

Author: JoeanAmier

docs/Release_Notes.md

@@ -1,16 +1,17 @@
 **更新内容：**
 
-1. 修复 API 模式直播接口获取原始数据报错的问题
-2. 优化 API 模式搜索接口的响应提示
-3. 修复 MacOS 下载直播报错的问题
-4. 搜索数据为空时不再保存至文件
-5. 支持设置直播下载的默认清晰度
-6. 重构复制粘贴写入 Cookie 功能
-7. 重构调用 ffmpeg 下载直播功能
-8. 优化提取链接的正则表达式
-9. 优化配置文件参数容错机制
-10. 恢复数据请求延时机制
-11. 优化程序运行提示
-12. 其他细节优化
+1. 修复下载私密账号作品时出现 `404` 响应码的问题
+2. 修复 API 模式直播接口获取原始数据报错的问题
+3. 优化 API 模式搜索接口的响应提示
+4. 修复 MacOS 下载直播报错的问题
+5. 搜索数据为空时不再保存至文件
+6. 支持设置直播下载的默认清晰度
+7. 重构复制粘贴写入 Cookie 功能
+8. 重构调用 ffmpeg 下载直播功能
+9. 优化提取链接的正则表达式
+10. 优化配置文件参数容错机制
+11. 恢复数据请求延时机制
+12. 优化程序运行提示
+13. 其他细节优化
 
 <p><strong>更新公告：本项目名称由 <code>TikTokDownloader</code> 变更为 <code>DouK-Downloader</code></strong></p>

src/config/parameter.py

@@ -24,7 +24,7 @@
 from ..module import FFMPEG
 from ..record import BaseLogger, LoggerManager
 from ..storage import RecordManager
-from ..tools import Cleaner, cookie_dict_to_str, create_client
+from ..tools import Cleaner, cookie_dict_to_str, create_client, DownloaderError
 from ..translation import _
 
 if TYPE_CHECKING:
@@ -139,7 +139,7 @@ def __init__(
         self.cookie_state: bool = self.__check_cookie_state()
         self.cookie_tiktok_state: bool = self.__check_cookie_state(True)
         self.set_uif_id()
-        self.set_download_headers()
+        # self.set_download_headers()
 
         self.root = self.__check_root(root)
         self.folder_name = self.__check_folder_name(folder_name)
@@ -290,7 +290,7 @@ def __get_cookie_tiktok_cache(
     ) -> str:
         return self.__check_cookie_tiktok(cookie)[1]
 
-    async def __add_cookie(
+    def __add_cookie(
         self,
         parameters: tuple[dict, ...],
         cookie: dict | str,
@@ -303,6 +303,7 @@ async def __add_cookie(
                         False,
                     )
                     cookie |= i
+            return None
         elif isinstance(cookie, str):
             for i in parameters:
                 if i:
@@ -312,6 +313,7 @@ async def __add_cookie(
                     )
                     cookie += f"; {cookie_dict_to_str(i)}"
             return cookie
+        raise DownloaderError
 
     async def __get_tt_wid_params(self) -> dict:
         if tt_wid := await TtWid.get_tt_wid(
@@ -569,7 +571,10 @@ async def update_params(self) -> None:
                         ms_token,
                         tt_wid,
                     ),
-                    self.headers,
+                    (
+                        self.headers,
+                        self.headers_download,
+                    ),
                     self.cookie_dict,
                     self.cookie_str,
                 )
@@ -598,7 +603,10 @@ async def update_params(self) -> None:
                         ms_token,
                         tt_wid,
                     ),
-                    self.headers_tiktok,
+                    (
+                        self.headers_tiktok,
+                        self.headers_download_tiktok,
+                    ),
                     self.cookie_dict_tiktok,
                     self.cookie_str_tiktok,
                 )
@@ -627,7 +635,10 @@ async def update_params_offline(self) -> None:
                 API.params["msToken"] = ms_token
                 await self.__update_cookie(
                     ({MsToken.NAME: ms_token},),
-                    self.headers,
+                    (
+                        self.headers,
+                        self.headers_download,
+                    ),
                     self.cookie_dict,
                     self.cookie_str,
                 )
@@ -646,7 +657,10 @@ async def update_params_offline(self) -> None:
                 APITikTok.params["msToken"] = ms_token.get(MsTokenTikTok.NAME, "")
                 await self.__update_cookie(
                     (ms_token,),
-                    self.headers_tiktok,
+                    (
+                        self.headers_tiktok,
+                        self.headers_download_tiktok,
+                    ),
                     self.cookie_dict_tiktok,
                     self.cookie_str_tiktok,
                 )
@@ -660,30 +674,36 @@ async def update_params_offline(self) -> None:
     async def __update_cookie(
         self,
         parameters: tuple[dict, ...],
-        headers: dict,
-        cookie: dict,
-        cache: str,
+        headers: tuple[dict, ...],
+        cookie_dict: dict,
+        cookie_str: str,
     ) -> None:
-        if cookie:
-            await self.__add_cookie(
-                parameters,
-                cookie,
-            )
-            headers["Cookie"] = cookie_dict_to_str(cookie)
-        elif cache:
-            headers["Cookie"] = await self.__add_cookie(parameters, cache)
+        cookie = self.__add_cookie(
+            parameters,
+            cookie_dict or cookie_str,
+        )
+        if not isinstance(cookie, str):
+            cookie = cookie_dict_to_str(cookie_dict)
+        for i in headers:
+            i["Cookie"] = cookie
 
     def set_headers_cookie(
         self,
     ) -> None:
         if self.cookie_dict:
-            self.headers["Cookie"] = cookie_dict_to_str(self.cookie_dict)
+            cookie = cookie_dict_to_str(self.cookie_dict)
+            self.headers["Cookie"] = cookie
+            self.headers_download["Cookie"] = cookie
         elif self.cookie_str:
             self.headers["Cookie"] = self.cookie_str
+            self.headers_download["Cookie"] = self.cookie_str
         if self.cookie_dict_tiktok:
-            self.headers_tiktok["Cookie"] = cookie_dict_to_str(self.cookie_dict_tiktok)
+            cookie = cookie_dict_to_str(self.cookie_dict_tiktok)
+            self.headers_tiktok["Cookie"] = cookie
+            self.headers_download_tiktok["Cookie"] = cookie
         elif self.cookie_str_tiktok:
             self.headers_tiktok["Cookie"] = self.cookie_str_tiktok
+            self.headers_download_tiktok["Cookie"] = self.cookie_str_tiktok
 
     def set_download_headers(self) -> None:
         self.__update_download_headers()