Strip bt sites tag from group name

Related divijbindlish#15 Signed-off-by: Roi Dayan <[email protected]>
roidayan · Aug 13, 2017 · 7a0fae9 · 7a0fae9
1 parent 6e0032e
commit 7a0fae9
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 4 deletions.
diff --git a/PTN/parse.py b/PTN/parse.py
@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 
 import re
-from .patterns import patterns, types
+from .patterns import patterns, types, bt_sites
 
 
 class PTN(object):
@@ -151,6 +151,16 @@ def parse(self, name):
                 group = group[:-(len(container)+1)]
                 self.parts['group'] = group
 
+        # clean group name from having bt site name
+        if 'group' in self.parts:
+
+            group = self.parts['group']
+            sites = '|'.join(bt_sites)
+            pat = '\[(' + sites + ')\]$'
+            group = re.sub(pat, '', group, flags=re.I)
+            if group:
+                self.parts['group'] = group
+
         if len(clean) != 0:
             if len(clean) == 1:
                 clean = clean[0]

diff --git a/PTN/patterns.py b/PTN/patterns.py
@@ -44,3 +44,5 @@
     'unrated': 'boolean',
     '3d': 'boolean'
 }
+
+bt_sites = ['eztv', 'ettv', 'rarbg', 'rartv', 'ETRG']
diff --git a/tests/files/output.json b/tests/files/output.json
@@ -5,7 +5,8 @@
     "resolution": "720p",
     "quality": "HDTV",
     "codec": "x264",
-    "title": "The Walking Dead"
+    "title": "The Walking Dead",
+    "group": "ASAP"
   },
   {
     "year": 2014,
@@ -50,7 +51,8 @@
     "codec": "XViD",
     "extended": true,
     "excess": "",
-    "title": "Hercules"
+    "title": "Hercules",
+    "group": "juggs"
   },
   {
     "year": 2014,
@@ -119,7 +121,8 @@
     "quality": "WEB-DL",
     "codec": "H.264",
     "audio": "DD5.1",
-    "title": "The Walking Dead"
+    "title": "The Walking Dead",
+    "group": "Cyphanix"
   },
   {
     "year": 2012,