import logging logger = logging.getLogger("panwid.datatable") import urwid intersperse = lambda e,l: sum([[x, e] for x in l],[])[:-1] class DataTableCell(urwid.WidgetWrap): signals = ["click", "select"] ATTR = "table_cell" PADDING_ATTR = "table_row_padding" def __init__(self, table, column, value=None, value_attr=None, cell_selection=False, padding=0, *args, **kwargs): self.table = table self.attr = self.ATTR self.attr_focused = "%s focused" %(self.attr) self.attr_column_focused = "%s column_focused" %(self.attr) self.attr_highlight = "%s highlight" %(self.attr) self.attr_highlight_focused = "%s focused" %(self.attr_highlight) self.attr_highlight_column_focused = "%s column_focused" %(self.attr_highlight) self.column = column self.value = value self.value_attr = value_attr self.cell_selection = cell_selection if column.padding: self.padding = column.padding else: self.padding = padding self.update_contents() self.padding = urwid.Padding( self.contents, left=self.padding, right=self.padding ) self.normal_attr_map = {} self.highlight_attr_map = {} self.normal_focus_map = {} self.highlight_focus_map = {} self.highlight_column_focus_map = {} self.set_attr_maps() self.highlight_attr_map.update(self.table.highlight_map) self.highlight_focus_map.update(self.table.highlight_focus_map) self.attrmap = urwid.AttrMap( self.padding, attr_map = self.normal_attr_map, focus_map = self.normal_focus_map ) super(DataTableCell, self).__init__(self.attrmap) def update_contents(self): pass def set_attr_maps(self): self.normal_attr_map[None] = self.attr self.highlight_attr_map [None] = self.attr_highlight self.normal_focus_map[None] = self.attr_focused self.highlight_focus_map[None] = self.attr_highlight_focused if self.value_attr: self.normal_attr_map.update({None: self.value_attr}) self.normal_focus_map.update({None: "%s focused" %(self.value_attr)}) self.highlight_attr_map.update({None: "%s highlight" %(self.value_attr)}) if self.cell_selection: self.highlight_focus_map.update({None: "%s highlight column_focused" %(self.value_attr)}) else: self.highlight_focus_map.update({None: "%s highlight focused" %(self.value_attr)}) def highlight(self): self.attrmap.set_attr_map(self.highlight_attr_map) self.attrmap.set_focus_map(self.highlight_focus_map) def unhighlight(self): self.attrmap.set_attr_map(self.normal_attr_map) self.attrmap.set_focus_map(self.normal_focus_map) def enable_selection(self): self.cell_selection = True def disable_selection(self): self.cell_selection = False def selectable(self): return self.cell_selection def keypress(self, size, key): try: key = super(DataTableCell, self).keypress(size, key) except AttributeError: pass return key # return super(DataTableCell, self).keypress(size, key) def _format(self, v): return self.column._format(v) def set_attr_map(self, attr_map): self.attrmap.set_attr_map(attr_map) def mouse_event(self, size, event, button, col, row, focus): if event == 'mouse press': urwid.emit_signal(self, "click") def set_attr(self, attr): attr_map = self.attrmap.get_attr_map() attr_map[None] = attr self.attr.set_attr_map(attr_map) focus_map = self.attr.get_focus_map() focus_map[None] = "%s focused" %(attr) self.attrmap.set_focus_map(focus_map) def clear_attr(self, attr): attr_map = self.attrmap.get_attr_map() attr_map = self.normal_attr_map # attr_map[None] = self.attr self.attr.set_attr_map(attr_map) focus_map = self.normal_focus_map #.attr.get_focus_map() # focus_map[None] = self.attr_focused self.attrmap.set_focus_map(focus_map) class DataTableBodyCell(DataTableCell): ATTR = "table_row_body" PADDING_ATTR = "table_row_body_padding" def update_contents(self): self.contents = self._format(self.value) class DataTableHeaderCell(DataTableCell): ATTR = "table_row_header" PADDING_ATTR = "table_row_header_padding" ASCENDING_SORT_MARKER = u"\N{UPWARDS ARROW}" DESCENDING_SORT_MARKER = u"\N{DOWNWARDS ARROW}" # def __init__(self, table, column, sort=None, sort_icon=None, *args, **kwargs): def update_contents(self): self.sort_icon = self.column.sort_icon if self.column.sort_icon else self.table.sort_icons self.columns = urwid.Columns([ ('weight', 1, self.column.label if isinstance(self.column.label, urwid.Widget) else urwid.Text(self.column.label, align=self.column.align) ) ]) if self.sort_icon: if self.column.align == "right": self.columns.contents.insert(0, (urwid.Text(""), self.columns.options("given", 1)) ) else: self.columns.contents.append( (urwid.Text(""), self.columns.options("given", 1)) ) self.contents = self.columns self.update_sort(self.table.sort_by) def set_attr_maps(self): self.normal_attr_map[None] = self.attr self.highlight_attr_map [None] = self.attr_highlight # if self.cell_selection: self.normal_focus_map[None] = self.attr_column_focused self.highlight_focus_map[None] = self.attr_highlight_column_focused def _format(self, v): return self.column.format(v) def selectable(self): return self.table.ui_sort def keypress(self, size, key): if key != "enter": return key urwid.emit_signal(self, "select", self) def mouse_event(self, size, event, button, col, row, focus): if event == 'mouse press': urwid.emit_signal(self, "click", self) def update_sort(self, sort): if not self.sort_icon: return index = 0 if self.column.align=="right" else 1 if sort and sort[0] == self.column.name: direction = self.DESCENDING_SORT_MARKER if sort[1] else self.ASCENDING_SORT_MARKER self.columns.contents[index][0].set_text(direction) else: self.columns.contents[index][0].set_text("") class DataTableFooterCell(DataTableCell): ATTR = "table_row_footer" PADDING_ATTR = "table_row_footer_padding" def update_contents(self): if self.column.footer_fn and len(self.table.df): # self.table.df.log_dump() if self.column.footer_arg == "values": footer_arg = self.table.df[self.column.name].to_list() elif self.column.footer_arg == "rows": footer_arg = self.table.df.iterrows() elif self.column.footer_arg == "table": footer_arg = self.table.df else: raise Exception self.value = self.column.footer_fn(self.column, footer_arg) self.contents = self._format(self.value) else: self.contents = urwid.Text("")