mirror of
https://github.com/Mikaela/Limnoria.git
synced 2024-11-23 02:49:27 +01:00
utils/str.py: Use classes Multiple* instead of lambda functions.
As in Python 3, functions are bound to a class when assigned to an attribute, lambda functions get two parameters, unless they are decorated with classmethod(). And there is no reason to ask plugin developpers use this decorator, while we avoid them doing this (more over, it is not obvious to debug if they forgot it).
This commit is contained in:
parent
17eb6c497f
commit
602f1c7ecb
@ -218,19 +218,31 @@ def perlVariableSubstitute(vars, text):
|
||||
return '$' + unbraced
|
||||
return _perlVarSubstituteRe.sub(replacer, text)
|
||||
|
||||
def multipleReplacer(dict_):
|
||||
"""Return a function that replaces all dict keys by the associated
|
||||
class MultipleReplacer:
|
||||
"""Return a callable that replaces all dict keys by the associated
|
||||
value. More efficient than multiple .replace()."""
|
||||
dict_ = {re.escape(key): val for key,val in dict_.items()}
|
||||
matcher = re.compile('|'.join(dict_.keys()))
|
||||
return lambda x:matcher.sub(lambda m: dict_[m.group(0)], x)
|
||||
|
||||
def multipleRemover(list_):
|
||||
"""Return a function that removes all words in the list. A bit more
|
||||
# We use an object instead of a lambda function because it avoids the
|
||||
# need for using the staticmethod() on the lambda function if assigning
|
||||
# it to a class in Python 3.
|
||||
def __init__(self, dict_):
|
||||
self._dict = dict_
|
||||
dict_ = {re.escape(key): val for key,val in dict_.items()}
|
||||
self._matcher = re.compile('|'.join(dict_.keys()))
|
||||
def __call__(self, s):
|
||||
return self._matcher.sub(lambda m: self._dict[m.group(0)], s)
|
||||
def multipleReplacer(dict_):
|
||||
return MultipleReplacer(dict_)
|
||||
|
||||
class MultipleRemover:
|
||||
"""Return a callable that removes all words in the list. A bit more
|
||||
efficient than multipleReplacer"""
|
||||
list_ = [re.escape(x) for x in list_]
|
||||
matcher = re.compile('|'.join(list_))
|
||||
return lambda x:matcher.sub(lambda m: '', x)
|
||||
# See comment of MultipleReplacer
|
||||
def __init__(self, list_):
|
||||
list_ = [re.escape(x) for x in list_]
|
||||
self._matcher = re.compile('|'.join(list_))
|
||||
def __call__(self, s):
|
||||
return self._matcher.sub(lambda m: '', s)
|
||||
|
||||
|
||||
|
||||
|
@ -310,11 +310,11 @@ class StrTest(SupyTestCase):
|
||||
self.assertEqual(f('bar'), 'foobar')
|
||||
|
||||
def testMultipleReplacer(self):
|
||||
replacer = utils.str.multipleReplacer({'foo': 'bar', 'a': 'b'})
|
||||
replacer = utils.str.MultipleReplacer({'foo': 'bar', 'a': 'b'})
|
||||
self.assertEqual(replacer('hi foo hi'), 'hi bar hi')
|
||||
|
||||
def testMultipleRemover(self):
|
||||
remover = utils.str.multipleRemover(['foo', 'bar'])
|
||||
remover = utils.str.MultipleRemover(['foo', 'bar'])
|
||||
self.assertEqual(remover('testfoobarbaz'), 'testbaz')
|
||||
|
||||
def testPReToReplacerDifferentSeparator(self):
|
||||
|
Loading…
Reference in New Issue
Block a user