Handle more exceptions in title tag normalization code - 57

desbma · Mar 8, 2018 · 6b0a257 · 6b0a257
1 parent fd6a655
commit 6b0a257
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/amg/tag.py b/amg/tag.py
@@ -271,7 +271,9 @@ class SimpleSuffixCleaner(TitleCleanerBase):
   def cleanup(self, title, suffix):
     """ See TitleCleanerBase.cleanup. """
     if self.endslike(title, suffix):
-      title = self.rclean(self.rmsuffix(title, suffix))
+      new_title = self.rclean(self.rmsuffix(title, suffix))
+      if new_title.lower() != "the":
+        title = new_title
     return title
 
 

diff --git a/tests/test_tag.py b/tests/test_tag.py
@@ -445,7 +445,11 @@ def test_normalize_title_tag(self):
                   ("ARMORED DAWN - Barbarians In Black // Album Trailer (2018) //  AFM Records",
                    "Armored Dawn",
                    "Barbarians in Black",
-                   "Barbarians in Black"))
+                   "Barbarians in Black"),
+                  ("Conjurer - The Mire (Official Video)",
+                   "Conjurer",
+                   "Mire",
+                   "The Mire"))
 
     for source, artist, album, expected_result in references:
       with self.subTest(source=source, expected_result=expected_result, artist=artist, album=album):