feat(subtitle): add support to parse idx files

mnamer/const.py

@@ -62,7 +62,7 @@
 
 IS_DEBUG = gettrace() is not None
 
-SUBTITLE_CONTAINERS = [".srt", ".idx", ".sub"]
+SUBTITLE_CONTAINERS = [".srt", ".idx"]
 
 SYSTEM = {
     "date": dt.date.today(),

mnamer/metadata.py

@@ -45,6 +45,7 @@ class Metadata:
     quality: Optional[str] = None
     synopsis: Optional[str] = None
     media: Union[MediaType, str, None] = None
+    forced_sub: Optional[bool] = None
 
     def __setattr__(self, key: str, value: Any):
         converter_map: Dict[str, Callable] = {
@@ -55,8 +56,7 @@ def __setattr__(self, key: str, value: Any):
             "media": MediaType,
             "quality": str.lower,
             "synopsis": str.capitalize,
-        }
-        converter: Optional[Callable] = converter_map.get(key)
+        }.get(key)
         if value is not None and converter:
             value = converter(value)
         super().__setattr__(key, value)
@@ -70,7 +70,7 @@ def __str__(self) -> str:
     @property
     def extension(self):
         if is_subtitle(self.container) and self.language_sub:
-            return f".{self.language_sub.a2}{self.container}"
+            return f".{self.language_sub.a2}{'-forced' if self.forced_sub else ''}{self.container}"
         else:
             return self.container
 

mnamer/target.py

@@ -1,12 +1,11 @@
-from __future__ import annotations
-
-import datetime as dt
+from datetime import date
 from os import path
-from pathlib import Path
+from pathlib import Path, PurePath
+import re
 from shutil import move
-from typing import Any, ClassVar, Dict, List, Optional, Type
+from typing import Dict, List, Optional, Set, Union
 
-from guessit import guessit  # type: ignore
+from guessit import guessit
 
 from mnamer.exceptions import MnamerException
 from mnamer.language import Language
@@ -19,6 +18,7 @@
     filename_replace,
     filter_blacklist,
     filter_containers,
+    is_subtitle,
     str_replace,
     str_sanitize,
     str_scenify,
@@ -112,10 +112,22 @@ def destination(self) -> Path:
         if self._settings.lower:
             filename = filename.lower()
         filename = str_sanitize(filename)
-        directory = Path(dir_head, dir_tail)
-        return Path(directory, filename)
+        directory = self._make_path(dir_head, dir_tail)
+        return self._make_path(directory, filename)
+
+    def _make_path(self, *obj: Union[str, Path, PurePath]) -> Union[PurePath, Path]:
+        # Calling PurePath will create a PurePoxisPath or PureWindowsPath based
+        # on the system platform. This will create one based on the type of the
+        # source path class type instead.
+        return type(self.source)(*obj)
 
     def _parse(self, file_path: Path):
+        def _parse_idx_file(file_path: Path):
+            pattern = re.compile("^id:\s?([a-zA-Z]{,3})")
+            for i, line in enumerate(open(file_path)):
+                for match in re.finditer(pattern, line):
+                    return match.group(1)
+
         path_data: Dict[str, Any] = {}
         options = {"type": self._settings.media}
         raw_data = dict(guessit(str(file_path), options))
@@ -170,6 +182,16 @@ def _parse(self, file_path: Path):
             self.metadata.language_sub = path_data.get("subtitle_language")
         except MnamerException:
             pass
+        if not self.metadata.language_sub and self.metadata.container == ".idx":
+            try:
+                self.metadata.language_sub = _parse_idx_file(file_path)
+            except MnamerException:
+                pass
+        if is_subtitle(self.metadata.container) and re.search(
+            "\Wforced\W", str(file_path)
+        ):
+            self.metadata.forced_sub = True
+
         if self.metadata.media is MediaType.MOVIE:
             self.metadata.name = path_data.get("title")
             self.metadata.year = path_data.get("year")
@@ -236,6 +258,13 @@ def relocate(self) -> None:
         """Performs the action of renaming and/or moving a file."""
         destination_path = Path(self.destination).resolve()
         destination_path.parent.mkdir(parents=True, exist_ok=True)
+
+        if self.metadata.container == ".idx":
+            relevant_sub = self.source.with_suffix(".sub")
+            dest_relevant_sub = destination_path.with_suffix(".sub")
+            if relevant_sub.exists():
+                move(str(relevant_sub), str(dest_relevant_sub))
+
         try:
             move(str(self.source), destination_path)
         except OSError:  # pragma: no cover