Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2025-01-03 16:52:34 +01:00
Code Issues Projects Releases Wiki Activity

Add utils.str.multipleReplacer.

Browse Source
This commit is contained in:
Valentin Lorentz 2012-08-03 09:39:31 +02:00
parent 59b9ba2cf8
commit fffccb4600
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/utils/str.py
View File

@ -218,6 +218,13 @@ def perlVariableSubstitute(vars, text):
return '$' + unbraced return '$' + unbraced
return _perlVarSubstituteRe.sub(replacer, text) return _perlVarSubstituteRe.sub(replacer, text)
def multipleReplacer(dict_):
"""Return a function that replaces all dict keys by the associated
value."""
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 commaAndify(seq, comma=',', And='and'): def commaAndify(seq, comma=',', And='and'):
"""Given a a sequence, returns an English clause for that sequence. """Given a a sequence, returns an English clause for that sequence.

4
test/test_utils.py
View File

@ -309,6 +309,10 @@ class StrTest(SupyTestCase):
f = PRTR('s/^/foo/') f = PRTR('s/^/foo/')
self.assertEqual(f('bar'), 'foobar') self.assertEqual(f('bar'), 'foobar')
def testmultipleReplacer(self):
replacer = utils.str.multipleReplacer({'foo': 'bar', 'a': 'b'})
self.assertEqual(replacer('hi foo hi'), 'hi bar hi')
def testPReToReplacerDifferentSeparator(self): def testPReToReplacerDifferentSeparator(self):
f = utils.str.perlReToReplacer('s#foo#bar#') f = utils.str.perlReToReplacer('s#foo#bar#')
self.assertEqual(f('foobarbaz'), 'barbarbaz') self.assertEqual(f('foobarbaz'), 'barbarbaz')