fix(streaming_service): add iT keyword support for iTunes (#669)

guessit-io · Feb 3, 2021 · 51e0021 · 51e0021
1 parent c635022
commit 51e0021
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/guessit/config/options.json b/guessit/config/options.json
@@ -490,7 +490,10 @@
       "Hulu": "HULU",
       "Investigation Discovery": "ID",
       "IFC": "IFC",
-      "iTunes": "iTunes",
+      "iTunes": [
+        "iTunes",
+        {"pattern": "iT", "ignore_case": false}
+      ],
       "ITV": "ITV",
       "Knowledge Network": "KNOW",
       "Lifetime": "LIFE",

diff --git a/guessit/rules/properties/streaming_service.py b/guessit/rules/properties/streaming_service.py
@@ -30,10 +30,17 @@ def streaming_service(config):  # pylint: disable=too-many-statements,unused-arg
     for value, items in config.items():
         patterns = items if isinstance(items, list) else [items]
         for pattern in patterns:
-            if pattern.startswith(regex_prefix):
-                rebulk.regex(pattern[len(regex_prefix):], value=value)
+            if isinstance(pattern, dict):
+                pattern_value = pattern.pop('pattern')
+                kwargs = pattern
+                pattern = pattern_value
             else:
-                rebulk.string(pattern, value=value)
+                kwargs = {}
+            regex = kwargs.pop('regex', False)
+            if regex or pattern.startswith(regex_prefix):
+                rebulk.regex(pattern[len(regex_prefix):], value=value, **kwargs)
+            else:
+                rebulk.string(pattern, value=value, **kwargs)
 
     rebulk.rules(ValidateStreamingService)
 

diff --git a/guessit/test/streaming_services.yaml b/guessit/test/streaming_services.yaml
@@ -607,6 +607,7 @@
   release_group: CasStudio
   type: episode
 
+? Suits.S07E01.1080p.iT.WEB-DL.DD5.1.H.264-VLAD.mkv
 ? Suits.S07E01.1080p.iTunes.WEB-DL.DD5.1.H.264-VLAD.mkv
 : title: Suits
   season: 7