mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 08:39:17 +01:00
f005ec2de0
The initial idea for this refactor was motivated by the need to make PluginQuery.match() have the same method signature as the match() methods on other queries. That is, it needed to take an *item*, not the pattern and value. (The pattern is supplied when the query is constructed.) So it made sense to move the value-to-pattern code to a class method. But then I realized that all the other FieldQuery subclasses needed to do essentially the same thing. So I eliminated PluginQuery altogether and refactored FieldQuery to subsume its functionality. I then changed all the other FieldQuery subclasses to conform to the same pattern. This has the side effect of allowing different kinds of queries (even non-field queries) down the road.
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
# This file is part of beets.
|
|
# Copyright 2013, Philippe Mongeau.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
|
# a copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
# permit persons to whom the Software is furnished to do so, subject to
|
|
# the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be
|
|
# included in all copies or substantial portions of the Software.
|
|
|
|
"""Provides a fuzzy matching query.
|
|
"""
|
|
|
|
from beets.plugins import BeetsPlugin
|
|
from beets.library import FieldQuery
|
|
import beets
|
|
import difflib
|
|
|
|
|
|
class FuzzyQuery(FieldQuery):
|
|
@classmethod
|
|
def value_match(self, pattern, val):
|
|
# smartcase
|
|
if pattern.islower():
|
|
val = val.lower()
|
|
queryMatcher = difflib.SequenceMatcher(None, pattern, val)
|
|
threshold = beets.config['fuzzy']['threshold'].as_number()
|
|
return queryMatcher.quick_ratio() >= threshold
|
|
|
|
|
|
class FuzzyPlugin(BeetsPlugin):
|
|
def __init__(self):
|
|
super(FuzzyPlugin, self).__init__()
|
|
self.config.add({
|
|
'prefix': '~',
|
|
'threshold': 0.7,
|
|
})
|
|
super(FuzzyPlugin, self).__init__(self)
|
|
|
|
def queries(self):
|
|
prefix = beets.config['fuzzy']['prefix'].get(basestring)
|
|
return {prefix: FuzzyQuery}
|