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

Add support for a default in Env

Browse source
This commit is contained in:
Laurent Bachelier 2014-08-01 15:57:38 +02:00
commit 94deb53352
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
weboob/tools/browser2/filters/standard.py
View file

@ -148,12 +148,15 @@ class Env(_Filter):
method on ItemElement. method on ItemElement.
""" """
def __init__(self, name): def __init__(self, name, default=_NO_DEFAULT):
super(Env, self).__init__() super(Env, self).__init__(default)
self.name = name self.name = name
def __call__(self, item): def __call__(self, item):
try:
return item.env[self.name] return item.env[self.name]
except KeyError:
return self.default_or_raise(ParseError('Environment variable %s not found' % self.name))
class TableCell(_Filter): class TableCell(_Filter):