YunoHost/moulinette
1
0
Fork 0
mirror of https://github.com/YunoHost/moulinette.git synced 2024-09-03 20:06:31 +02:00
Code Issues Projects Releases Packages Wiki Activity

[enh] allow Table/TableDict to have a row_function

Browse source
This commit is contained in:
Laurent Peuch 2021-06-22 04:15:55 +02:00
parent 590b67639e
commit 41539a2769
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
moulinette/__init__.py
View file

@ -65,10 +65,11 @@ console.format_exception = _format_exception
class Table:
def __init__(self, data, columns=None, title=None):
def __init__(self, data, columns=None, title=None, row_function=None):
self.data = data
self.columns = columns
self.title = title
self.row_function = row_function
def print(self):
if not self.data:
@ -106,7 +107,11 @@ class Table:
for column in self.columns:
values.append(row.get(column, ""))
table.add_row(*values)
if self.row_function is not None:
# this function is responsible for adding the row
self.row_function(table=table, columns=self.columns, row=row, values=values)
else:
table.add_row(*values)
class TableForDict(Table):
@ -139,8 +144,12 @@ class TableForDict(Table):
if column != self.key:
row_values.append(str(values.get(column, "")))
table.add_row(*row_values)
if self.row_function is not None:
# this function is responsible for adding the row
self.row_function(table=table, columns=self.columns, row=(key, values),
values=row_values)
else:
table.add_row(*row_values)
# Package functions