refactor(recognize enhance): disney -> DSNP

hsuyelin · Sep 8, 2023 · 81190bc · 81190bc
1 parent 692f5ff
commit 81190bc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/app/media/meta/metavideov2.py b/app/media/meta/metavideov2.py
@@ -289,12 +289,15 @@ def __init_resource_type(self):
         if not StringUtils.is_string_and_not_empty(streaming_services):
             return
 
-        streaming_services = "AMZN" if streaming_services.lower() == "amazon prime" else streaming_services
         if StringUtils.is_string_and_not_empty(self.resource_type):
             self.resource_type += f" {streaming_services}"
         else:
             self.resource_type = streaming_services
 
+        self.resource_type = re.sub(r'\+', '', self.resource_type)
+        self.resource_type = re.sub(r'(?i)amazon prime', 'AMZN', self.resource_type)
+        self.resource_type = re.sub(r'(?i)disney', 'DSNP', self.resource_type)
+
     def __init_resource_effect(self):
         title_resource_effects = self._media_item_title.other.other
         subltitle_resource_effects = self._media_item_subtitle.other.other