refactor(app): 重构数据处理逻辑

source/application/app.py

@@ -199,7 +199,7 @@ class XHS:
         container: dict,
         data: Namespace,
     ):
-        container["下载地址"] = self.video.get_video_link(
+        container["下载地址"] = self.video.deal_video_link(
             data,
         )
         container["动图地址"] = [

source/application/image.py

@@ -10,10 +10,16 @@ class Image:
     def get_image_link(cls, data: Namespace, format_: str) -> tuple[list, list]:
         images = data.safe_extract("imageList", [])
         live_link = cls.__get_live_link(images)
-        token_list = [
-            cls.__extract_image_token(Namespace.object_extract(i, "urlDefault"))
-            for i in images
-        ]
+        if not any(
+            token_list := [
+                cls.__extract_image_token(Namespace.object_extract(i, "urlDefault"))
+                for i in images
+            ]
+        ):
+            token_list = [
+                cls.__extract_image_token(Namespace.object_extract(i, "url"))
+                for i in images
+            ]
         match format_:
             case "png" | "webp" | "jpeg" | "heic" | "avif":
                 return [

source/application/video.py

@@ -11,6 +11,14 @@ class Video:
         "originVideoKey",
     )
 
+    @classmethod
+    def deal_video_link(
+        cls,
+        data: Namespace,
+        preference="resolution",
+    ):
+        return cls.generate_video_link(data) or cls.get_video_link(data, preference)
+
     @classmethod
     def generate_video_link(cls, data: Namespace) -> list:
         return (

source/expansion/converter.py

@@ -8,12 +8,17 @@ __all__ = ["Converter"]
 
 class Converter:
     INITIAL_STATE = "//script/text()"
-    KEYS_LINK = (
+    PC_KEYS_LINK = (
         "note",
         "noteDetailMap",
         "[-1]",
         "note",
     )
+    PHONE_KEYS_LINK = (
+        "noteData",
+        "data",
+        "noteData",
+    )
 
     def run(self, content: str) -> dict:
         return self._filter_object(self._convert_object(self._extract_object(content)))
@@ -31,7 +36,11 @@ class Converter:
 
     @classmethod
     def _filter_object(cls, data: dict) -> dict:
-        return cls.deep_get(data, cls.KEYS_LINK) or {}
+        return (
+            cls.deep_get(data, cls.PHONE_KEYS_LINK)
+            or cls.deep_get(data, cls.PC_KEYS_LINK)
+            or {}
+        )
 
     @classmethod
     def deep_get(cls, data: dict, keys: list | tuple, default=None):
@@ -58,7 +67,11 @@ class Converter:
     @staticmethod
     def get_script(scripts: list) -> str:
         scripts.reverse()
-        for script in scripts:
-            if script.startswith("window.__INITIAL_STATE__"):
-                return script
-        return ""
+        return next(
+            (
+                script
+                for script in scripts
+                if script.startswith("window.__INITIAL_STATE__")
+            ),
+            "",
+        )