From 007597239fa8d791a811cb23f4646ddc769fe0cf Mon Sep 17 00:00:00 2001 From: clinton-hall Date: Fri, 14 Jun 2013 15:06:59 +0930 Subject: [PATCH 1/2] add categories --- couchpotato/core/plugins/renamer/main.py | 4 ++++ couchpotato/core/plugins/searcher/main.py | 14 ++++++++++---- couchpotato/core/settings/model.py | 17 +++++++++++++++++ 3 files changed, 31 insertions(+), 4 deletions(-) diff --git a/couchpotato/core/plugins/renamer/main.py b/couchpotato/core/plugins/renamer/main.py index 7ab9f450..cf79b34f 100644 --- a/couchpotato/core/plugins/renamer/main.py +++ b/couchpotato/core/plugins/renamer/main.py @@ -143,6 +143,10 @@ class Renamer(Plugin): remove_releases = [] movie_title = getTitle(group['library']) + try: + destination = group['category']['path'] + except: + destination = self.conf('to') # Add _UNKNOWN_ if no library item is connected if not group['library'] or not movie_title: diff --git a/couchpotato/core/plugins/searcher/main.py b/couchpotato/core/plugins/searcher/main.py index 23e8ea58..ab309845 100644 --- a/couchpotato/core/plugins/searcher/main.py +++ b/couchpotato/core/plugins/searcher/main.py @@ -385,24 +385,30 @@ class Searcher(Plugin): nzb_words = re.split('\W+', nzb_name) # Make sure it has required words - required_words = splitString(self.conf('required_words').lower()) + try: + required_words = splitString(movie['category']['required'].lower()) + except: + required_words = splitString(self.conf('required_words').lower()) req_match = 0 for req_set in required_words: req = splitString(req_set, '&') req_match += len(list(set(nzb_words) & set(req))) == len(req) - if self.conf('required_words') and req_match == 0: + if len(required_words) > 0 and req_match == 0: log.info2('Wrong: Required word missing: %s', nzb['name']) return False # Ignore releases - ignored_words = splitString(self.conf('ignored_words').lower()) + try: + ignored_words = splitString(movie['category']['ignored'].lower()) + except: + ignored_words = splitString(self.conf('ignored_words').lower()) ignored_match = 0 for ignored_set in ignored_words: ignored = splitString(ignored_set, '&') ignored_match += len(list(set(nzb_words) & set(ignored))) == len(ignored) - if self.conf('ignored_words') and ignored_match: + if len(ignored_words) > 0 and ignored_match: log.info2("Wrong: '%s' contains 'ignored words'", (nzb['name'])) return False diff --git a/couchpotato/core/settings/model.py b/couchpotato/core/settings/model.py index 6c8deb1e..3ec78f44 100644 --- a/couchpotato/core/settings/model.py +++ b/couchpotato/core/settings/model.py @@ -50,6 +50,7 @@ class Movie(Entity): library = ManyToOne('Library', cascade = 'delete, delete-orphan', single_parent = True) status = ManyToOne('Status') profile = ManyToOne('Profile') + category = ManyToOne('Category') releases = OneToMany('Release', cascade = 'all, delete-orphan') files = ManyToMany('File', cascade = 'all, delete-orphan', single_parent = True) @@ -174,6 +175,22 @@ class Profile(Entity): return orig_dict +class Category(Entity): + """""" + using_options(order_by = 'order') + + label = Field(Unicode(50)) + order = Field(Integer, default = 0, index = True) + core = Field(Boolean, default = False) + hide = Field(Boolean, default = False) + + movie = OneToMany('Movie') + path = Field(Unicode(255)) + required = Field(Unicode(255)) + preferred = Field(Unicode(255)) + ignored = Field(Unicode(255)) + + class ProfileType(Entity): """""" using_options(order_by = 'order') From 60034f2c963d4e72cf0d94b39165631f0ac94a08 Mon Sep 17 00:00:00 2001 From: clinton-hall Date: Fri, 14 Jun 2013 21:56:26 +0930 Subject: [PATCH 2/2] add category preffered words and partial ignore. --- couchpotato/core/plugins/score/main.py | 14 ++++++--- couchpotato/core/plugins/score/scores.py | 40 ++++++++++++++++++++++++ 2 files changed, 50 insertions(+), 4 deletions(-) diff --git a/couchpotato/core/plugins/score/main.py b/couchpotato/core/plugins/score/main.py index f853be95..28baae7e 100644 --- a/couchpotato/core/plugins/score/main.py +++ b/couchpotato/core/plugins/score/main.py @@ -3,8 +3,8 @@ from couchpotato.core.helpers.encoding import toUnicode from couchpotato.core.helpers.variable import getTitle from couchpotato.core.logger import CPLog from couchpotato.core.plugins.base import Plugin -from couchpotato.core.plugins.score.scores import nameScore, nameRatioScore, \ - sizeScore, providerScore, duplicateScore, partialIgnoredScore, namePositionScore, \ +from couchpotato.core.plugins.score.scores import nameScore, CatnameScore, nameRatioScore, \ + sizeScore, providerScore, duplicateScore, partialIgnoredScore, CatpartialIgnoredScore, namePositionScore, \ halfMultipartScore log = CPLog(__name__) @@ -18,7 +18,10 @@ class Score(Plugin): def calculate(self, nzb, movie): ''' Calculate the score of a NZB, used for sorting later ''' - score = nameScore(toUnicode(nzb['name']), movie['library']['year']) + if movie and movie['category'] and movie['category']['preferred']: + score = CatnameScore(toUnicode(nzb['name']), movie['library']['year'], movie['category']['preferred']) + else: + score = nameScore(toUnicode(nzb['name']), movie['library']['year']) for movie_title in movie['library']['titles']: score += nameRatioScore(toUnicode(nzb['name']), toUnicode(movie_title['title'])) @@ -41,7 +44,10 @@ class Score(Plugin): score += duplicateScore(nzb['name'], getTitle(movie['library'])) # Partial ignored words - score += partialIgnoredScore(nzb['name'], getTitle(movie['library'])) + if movie and movie['category'] and movie['category']['ignored']: + score = CatpartialIgnoredScore(nzb['name'], getTitle(movie['library']), movie['category']['ignored']) + else: + score += partialIgnoredScore(nzb['name'], getTitle(movie['library'])) # Ignore single downloads from multipart score += halfMultipartScore(nzb['name']) diff --git a/couchpotato/core/plugins/score/scores.py b/couchpotato/core/plugins/score/scores.py index c9345fc0..5feeff39 100644 --- a/couchpotato/core/plugins/score/scores.py +++ b/couchpotato/core/plugins/score/scores.py @@ -49,6 +49,32 @@ def nameScore(name, year): return score +def CatnameScore(name, year, preferred): + ''' Calculate score for words in the NZB name ''' + + score = 0 + name = name.lower() + + # give points for the cool stuff + for value in name_scores: + v = value.split(':') + add = int(v.pop()) + if v.pop() in name: + score = score + add + + # points if the year is correct + if str(year) in name: + score = score + 5 + + # Contains preferred word + nzb_words = re.split('\W+', simplifyString(name)) + preferred_words = [x.strip() for x in preferred.split(',')] + for word in preferred_words: + if word.strip() and word.strip().lower() in nzb_words: + score = score + 100 + + return score + def nameRatioScore(nzb_name, movie_name): nzb_words = re.split('\W+', fireEvent('scanner.create_file_identifier', nzb_name, single = True)) @@ -150,6 +176,20 @@ def partialIgnoredScore(nzb_name, movie_name): return score +def CatpartialIgnoredScore(nzb_name, movie_name, ignored): + + nzb_name = nzb_name.lower() + movie_name = movie_name.lower() + + ignored_words = [x.strip().lower() for x in ignored.split(',')] + + score = 0 + for ignored_word in ignored_words: + if ignored_word in nzb_name and ignored_word not in movie_name: + score -= 5 + + return score + def halfMultipartScore(nzb_name): wrong_found = 0