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

formatters accept no backend name

Browse source
This commit is contained in:
Christophe Benz 2010-07-06 14:42:45 +02:00
commit 41eea192f4
2 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
weboob/tools/application/console.py
View file

@ -203,9 +203,10 @@ class ConsoleApplication(BaseApplication):
def set_header(self, string):
self.formatter.set_header(string)
def format(self, result, backend_name):
def format(self, result, backend_name=None):
try:
self.formatter.format(result, backend_name, selected_fields=self.selected_fields, condition=self.condition)
self.formatter.format(obj=result, backend_name=backend_name,
selected_fields=self.selected_fields, condition=self.condition)
except ResultsConditionException, e:
logging.error(e)

11
weboob/tools/application/formatters/iformatter.py
View file

@ -33,7 +33,7 @@ class IFormatter(object):
def flush(self):
raise NotImplementedError()
def format(self, obj, backend_name, selected_fields=None, condition=None):
def format(self, obj, backend_name=None, selected_fields=None, condition=None):
"""
Format an object to be human-readable.
An object has fields which can be selected, and the objects
@ -46,7 +46,10 @@ class IFormatter(object):
@param condition [Condition] condition to objects to display
@return a string of the formatted object
"""
item = self.to_dict(obj, backend_name, condition, selected_fields)
if isinstance(obj, dict):
item = obj
else:
item = self.to_dict(obj, backend_name, condition, selected_fields)
if item is None:
return None
formatted = self.format_dict(item=item)
@ -76,7 +79,7 @@ class IFormatter(object):
def set_header(self, string):
raise NotImplementedError()
def to_dict(self, obj, backend_name, condition=None, selected_fields=None):
def to_dict(self, obj, backend_name=None, condition=None, selected_fields=None):
def iter_select_and_decorate(d):
if hasattr(obj, '__id__'):
id_attr = getattr(obj, '__id__')
@ -88,7 +91,7 @@ class IFormatter(object):
for k, v in d:
if selected_fields is not None and k not in selected_fields:
continue
if k in id_fields:
if k in id_fields and backend_name is not None:
v = u'%s@%s' % (unicode(v), backend_name)
yield k, v
fields_iterator = obj.iter_fields() if hasattr(obj, 'iter_fields') else self.iter_fields(obj)