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

rewrite formatters management in repl

Browse source 
Commands can use specific default formatters, user can redefine formatters
per commands or globally.
This commit is contained in:
Romain Bignon 2010-10-14 20:09:32 +02:00
commit a86a6b2669
3 changed files with 119 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

21
weboob/tools/application/formatters/table.py
View file

@ -21,17 +21,17 @@ from prettytable import PrettyTable
from .iformatter import IFormatter
__all__ = ['TableFormatter']
__all__ = ['TableFormatter', 'HTMLTableFormatter']
class TableFormatter(IFormatter):
column_headers = None
queue = []
header = None
HTML = False
def __init__(self, display_keys=True, return_only=False, result_funcname='get_string'):
def __init__(self, display_keys=True, return_only=False):
IFormatter.__init__(self, display_keys=display_keys, return_only=return_only)
self.result_funcname = result_funcname
def after_format(self, formatted):
if self.column_headers is None:
@ -43,17 +43,21 @@ class TableFormatter(IFormatter):
return None
s = ''
if self.display_header and self.header:
if self.result_funcname == 'get_string':
s += self.header
elif self.result_funcname == 'get_html_string':
if self.HTML:
s+= '<p>%s</p>' % self.header
else:
s += self.header
s += "\n"
table = PrettyTable(list(self.column_headers))
for column_header in self.column_headers:
table.set_field_align(column_header, 'l')
for line in self.queue:
table.add_row(line)
s += getattr(table, self.result_funcname)()
if self.HTML:
s += table.get_html_string()
else:
s += table.get_string()
self.queue = []
if self.return_only:
return s
@ -66,3 +70,6 @@ class TableFormatter(IFormatter):
def set_header(self, string):
self.header = string
class HTMLTableFormatter(TableFormatter):
HTML = True