nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add replace option to CleanText

Browse source
This commit is contained in:
Florent 2014-04-01 10:46:08 +02:00
commit 6138209a2f
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
weboob/tools/browser2/filters.py
View file

@ -170,16 +170,18 @@ class CleanText(Filter):
string. string.
Second, it replaces all symbols given in second argument. Second, it replaces all symbols given in second argument.
""" """
def __init__(self, selector, symbols='', **kwargs): def __init__(self, selector, symbols='', replace=[], **kwargs):
super(CleanText, self).__init__(selector, **kwargs) super(CleanText, self).__init__(selector, **kwargs)
self.symbols = symbols self.symbols = symbols
self.toreplace = replace
def filter(self, txt): def filter(self, txt):
if isinstance(txt, (tuple,list)): if isinstance(txt, (tuple,list)):
txt = ' '.join(map(self.clean, txt)) txt = ' '.join(map(self.clean, txt))
txt = self.clean(txt) txt = self.clean(txt)
return self.remove(txt, self.symbols) txt = self.remove(txt, self.symbols)
return self.replace(txt, self.toreplace)
@classmethod @classmethod
def clean(cls, txt): def clean(cls, txt):
@ -195,6 +197,12 @@ class CleanText(Filter):
txt = txt.replace(symbol, '') txt = txt.replace(symbol, '')
return txt return txt
@classmethod
def replace(cls, txt, replace):
for before, after in replace:
txt = txt.replace(before, after)
return txt
class Lower(CleanText): class Lower(CleanText):
def filter(self, txt): def filter(self, txt):