Make sure fix_webvtt_separator is passed bytes or str

devine-dl · Dec 2, 2023 · b8cf7a2 · b8cf7a2
1 parent ada1fee
commit b8cf7a2
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/devine/core/tracks/subtitle.py b/devine/core/tracks/subtitle.py
@@ -191,10 +191,13 @@ def fix_webvtt_separator(data: Union[str, bytes]):
         """
         if isinstance(data, bytes):
             data = data.decode("utf8")
-        text = data.replace("WEBVTT", "\n\nWEBVTT"). \
-                    replace("\r", ""). \
-                    replace("\n\n\n", "\n \n\n"). \
-                    replace("\n\n<", "\n<")
+        elif not isinstance(data, str):
+            raise ValueError(f"Expecting data to be a str, not {data!r}")
+
+        text = data.replace("WEBVTT", "\n\nWEBVTT").\
+            replace("\r", "").\
+            replace("\n\n\n", "\n \n\n").\
+            replace("\n\n<", "\n<")
 
         return text