refactor

Author: Touyu-Pan

src/ffmpeg_media_type/utils/ffmpeg.py

@@ -249,25 +249,17 @@ def animated_webp_support(func: Callable[[str], FFProbeInfo]) -> Callable[[str],
     def wrapper(uri: str) -> FFProbeInfo:
         probe_info = func(uri)
         if probe_info.streams[0].height == 0 and probe_info.streams[0].width == 0 and probe_info.format.format_name == "webp_pipe":
-            if not uri.startswith("http"):
-                webpmux_command = ["webpmux", "-get", "frame", "1", uri, "-o", uri]
-                call(webpmux_command)
-                return func(uri)
-
-            parsed_uri = urlparse(uri)
-
-            path = parsed_uri.path
-            file = os.path.basename(path)
-            _, file_ext = os.path.splitext(file)
-
-            response = requests.get(uri)
-            response.raise_for_status()
-
-            with tempfile.NamedTemporaryFile(mode="wb", suffix=file_ext, delete=True) as f:
-                f.write(response.content)
-                webpmux_command = ["webpmux", "-get", "frame", "1", f.name, "-o", f.name]
-                call(webpmux_command)
-                return func(f.name)
+            if uri.startswith("http"):
+                response = requests.get(uri)
+                response.raise_for_status()
+                with tempfile.NamedTemporaryFile(mode="wb", suffix=".webp", delete=False) as f:
+                    f.write(response.content)
+                    uri = f.name
+
+            assert os.path.exists(uri)
+            webpmux_command = ["webpmux", "-get", "frame", "1", uri, "-o", uri]
+            call(webpmux_command)
+            return func(uri)
         return probe_info
 
     return wrapper