Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2026-01-06 07:53:40 +01:00
Code Issues Projects Releases Wiki Activity

Move the filtering logic of '_tags_for' function into a dedicated 'filter_tags' function.

Browse source
This commit is contained in:
Fabrice Laporte 2014-04-22 12:59:08 +02:00 committed by Adrian Sampson
parent 3d30708839
commit 499a7b868e
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
beetsplug/lastgenre/__init__.py
View file

@ -66,25 +66,32 @@ def _tags_for(obj):
log.debug(u'last.fm error: %s' % unicode(exc))
return []
tags = []
res = filter_tags(res)
return res
def filter_tags(tags):
"""Filter tags based on their weights and whitelist content
"""
res = []
min_weight = config['lastgenre']['min_weight'].get(int)
count = config['lastgenre']['count'].get(int)
dbg = []
for el in res:
for el in tags:
weight = int(el.weight or 0)
tag = el.item.get_name().lower()
if _is_allowed(tag):
if min_weight > -1 and min_weight > weight and len(tags) > 0:
return tags
tags.append(tag)
if min_weight > -1 and min_weight > weight and len(res) > 0:
return res
res.append(tag)
dbg.append(u'{0} [{1}]'.format(tag, weight))
if len(tags) == count:
if len(res) == count:
break
log.debug(u'lastfm.tag (min. {0}): {1}'.format(
min_weight, u', '.join(dbg)
))
return tags
return res
def _is_allowed(genre):