From 94deb53352fb9b398638688e329ee5dbc2f1582f Mon Sep 17 00:00:00 2001 From: Laurent Bachelier Date: Fri, 1 Aug 2014 15:57:38 +0200 Subject: [PATCH] Add support for a default in Env --- weboob/tools/browser2/filters/standard.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/weboob/tools/browser2/filters/standard.py b/weboob/tools/browser2/filters/standard.py index f4fcc44e..c7c4be6a 100644 --- a/weboob/tools/browser2/filters/standard.py +++ b/weboob/tools/browser2/filters/standard.py @@ -148,12 +148,15 @@ class Env(_Filter): method on ItemElement. """ - def __init__(self, name): - super(Env, self).__init__() + def __init__(self, name, default=_NO_DEFAULT): + super(Env, self).__init__(default) self.name = name def __call__(self, item): - return item.env[self.name] + try: + return item.env[self.name] + except KeyError: + return self.default_or_raise(ParseError('Environment variable %s not found' % self.name)) class TableCell(_Filter):