Add size to quality guessing

And cleanup searcher
This commit is contained in:
mano3m
2014-04-21 00:26:17 +02:00
parent e51ddd7a50
commit bfe8aa5f5f
3 changed files with 35 additions and 27 deletions
+19 -1
View File
@@ -177,7 +177,7 @@ class QualityPlugin(Plugin):
return False
def guess(self, files, extra = None):
def guess(self, files, extra = None, size = None):
if not extra: extra = {}
# Create hash for cache
@@ -205,6 +205,11 @@ class QualityPlugin(Plugin):
self.calcScore(score, quality, contains_score, threedscore)
# Evaluate score based on size
for quality in qualities:
size_score = self.guessSizeScore(quality, size = size)
self.calcScore(score, quality, size_score)
# Try again with loose testing
for quality in qualities:
loose_score = self.guessLooseScore(quality, extra = extra)
@@ -308,6 +313,19 @@ class QualityPlugin(Plugin):
return score
def guessSizeScore(self, quality, size = None):
score = 0
if size:
if tryInt(size) >= tryInt(quality['size_min']) and tryInt(size) < tryInt(quality['size_max']):
log.info2('Found %s via release size: %s MB < %s MB < %s MB', (quality['identifier'], quality['size_min'], size, quality['size_max']))
score += 5
return score
def calcScore(self, score, quality, add_score, threedscore = (0, None)):
score[quality['identifier']]['score'] += add_score