# -*- coding: iso-8859-1 -*- import ipipe, os, webbrowser, urllib import wx import wx.grid, wx.html try: sorted except NameError: from ipipe import sorted __all__ = ["igrid"] class IGridRenderer(wx.grid.PyGridCellRenderer): """ This is a custom renderer for our IGridGrid """ def __init__(self, table): self.maxchars = 200 self.table = table self.colormap = ( ( 0, 0, 0), (174, 0, 0), ( 0, 174, 0), (174, 174, 0), ( 0, 0, 174), (174, 0, 174), ( 0, 174, 174), ( 64, 64, 64) ) wx.grid.PyGridCellRenderer.__init__(self) def _getvalue(self, row, col): try: value = self.table.displayattrs[col].value(self.table.items[row]) (align, width, text) = ipipe.xformat(value, "cell", self.maxchars) except Exception, exc: (align, width, text) = ipipe.xformat(exc, "cell", self.maxchars) return (align, text) def GetBestSize(self, grid, attr, dc, row, col): text = grid.GetCellValue(row, col) (align, text) = self._getvalue(row, col) dc.SetFont(attr.GetFont()) (w, h) = dc.GetTextExtent(str(text)) return wx.Size(min(w+2, 600), h+2) # add border def Draw(self, grid, attr, dc, rect, row, col, isSelected): """ Takes care of drawing everything in the cell; aligns the text """ text = grid.GetCellValue(row, col) (align, text) = self._getvalue(row, col) if isSelected: bg = grid.GetSelectionBackground() else: bg = ["white", (240, 240, 240)][row%2] dc.SetTextBackground(bg) dc.SetBrush(wx.Brush(bg, wx.SOLID)) dc.SetPen(wx.TRANSPARENT_PEN) dc.SetFont(attr.GetFont()) dc.DrawRectangleRect(rect) dc.SetClippingRect(rect) # Format the text if align == -1: # left alignment (width, height) = dc.GetTextExtent(str(text)) x = rect[0]+1 y = rect[1]+0.5*(rect[3]-height) for (style, part) in text: if isSelected: fg = grid.GetSelectionForeground() else: fg = self.colormap[style.fg] dc.SetTextForeground(fg) (w, h) = dc.GetTextExtent(part) dc.DrawText(part, x, y) x += w elif align == 0: # center alignment (width, height) = dc.GetTextExtent(str(text)) x = rect[0]+0.5*(rect[2]-width) y = rect[1]+0.5*(rect[3]-height) for (style, part) in text: if isSelected: fg = grid.GetSelectionForeground() else: fg = self.colormap[style.fg] dc.SetTextForeground(fg) (w, h) = dc.GetTextExtent(part) dc.DrawText(part, x, y) x += w else: # right alignment (width, height) = dc.GetTextExtent(str(text)) x = rect[0]+rect[2]-1 y = rect[1]+0.5*(rect[3]-height) for (style, part) in reversed(text): (w, h) = dc.GetTextExtent(part) x -= w if isSelected: fg = grid.GetSelectionForeground() else: fg = self.colormap[style.fg] dc.SetTextForeground(fg) dc.DrawText(part, x, y) dc.DestroyClippingRegion() def Clone(self): return IGridRenderer(self.table) class IGridTable(wx.grid.PyGridTableBase): # The data table for the ``IGridGrid``. Some dirty tricks were used here: # ``GetValue()`` does not get any values (or at least it does not return # anything, accessing the values is done by the renderer) # but rather tries to fetch the objects which were requested into the table. # General behaviour is: Fetch the first X objects. If the user scrolls down # to the last object another bunch of X objects is fetched (if possible) def __init__(self, input, fontsize, *attrs): wx.grid.PyGridTableBase.__init__(self) self.input = input self.iterator = ipipe.xiter(input) self.items = [] self.hiddenattrs = [] self.attrs = attrs self.displayattrs = [] self.fetch(1) self.sizing = False self.fontsize = fontsize def GetAttr(self, *args): attr = wx.grid.GridCellAttr() attr.SetFont(wx.Font(self.fontsize, wx.TELETYPE, wx.NORMAL, wx.NORMAL)) return attr def GetNumberRows(self): return len(self.items) def GetNumberCols(self): return len(self.displayattrs) def GetColLabelValue(self, col): if col < len(self.displayattrs): return self.displayattrs[col].name() else: return "" def GetRowLabelValue(self, row): return str(row) def IsEmptyCell(self, row, col): return False def fetch(self, count): # Try to fill ``self.items`` with at least ``count`` objects. have = len(self.items) work = False while self.iterator is not None and have < count: try: item = self.iterator.next() except StopIteration: self.iterator = None break except (KeyboardInterrupt, SystemExit): raise except Exception, exc: have += 1 self.items.append(exc) work = True self.iterator = None break else: have += 1 self.items.append(item) work = True if work: self.calcdisplayattrs() def calcdisplayattrs(self): # Calculate which attributes are available from the objects that are # currently visible on screen (and store it in ``self.displayattrs``) attrs = set() self.displayattrs = [] if self.attrs: # If the browser object specifies a fixed list of attributes, # simply use it (removing hidden attributes). for attr in self.attrs: attr = ipipe.upgradexattr(attr) if attr not in attrs and attr not in self.hiddenattrs: self.displayattrs.append(attr) attrs.add(attr) else: endy = len(self.items) for i in xrange(endy): for attr in ipipe.xattrs(self.items[i]): attr = ipipe.upgradexattr(attr) if attr not in attrs and attr not in self.hiddenattrs: self.displayattrs.append(attr) attrs.add(attr) def GetValue(self, row, col): # some kind of dummy-function: does not return anything but ""; # (The value isn't use anyway) # its main task is to trigger the fetch of new objects had_cols = self.displayattrs[:] had_rows = len(self.items) if row == had_rows - 1 and self.iterator is not None and not self.sizing: self.fetch(row + 20) have_rows = len(self.items) have_cols = len(self.displayattrs) if have_rows > had_rows: msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED, have_rows - had_rows) self.GetView().ProcessTableMessage(msg) self.sizing = True self.GetView().AutoSizeColumns(False) self.sizing = False if row >= have_rows: return "" if self.displayattrs != had_cols: msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_NOTIFY_COLS_APPENDED, have_cols - len(had_cols)) self.GetView().ProcessTableMessage(msg) return "" def SetValue(self, row, col, value): pass class IGridGrid(wx.grid.Grid): # The actual grid # all methods for selecting/sorting/picking/... data are implemented here def __init__(self, panel, input, *attrs): wx.grid.Grid.__init__(self, panel) fontsize = 9 self.input = input self.table = IGridTable(self.input, fontsize, *attrs) self.SetTable(self.table, True) self.SetSelectionMode(wx.grid.Grid.wxGridSelectRows) self.SetDefaultRenderer(IGridRenderer(self.table)) self.EnableEditing(False) self.Bind(wx.EVT_KEY_DOWN, self.key_pressed) self.Bind(wx.grid.EVT_GRID_CELL_LEFT_DCLICK, self.cell_doubleclicked) self.Bind(wx.grid.EVT_GRID_LABEL_LEFT_DCLICK, self.label_doubleclicked) self.Bind(wx.grid.EVT_GRID_LABEL_LEFT_CLICK, self.on_label_leftclick) self.Bind(wx.grid.EVT_GRID_RANGE_SELECT, self._on_selected_range) self.Bind(wx.grid.EVT_GRID_SELECT_CELL, self._on_selected_cell) self.current_selection = set() self.maxchars = 200 def on_label_leftclick(self, event): event.Skip() def error_output(self, text): wx.Bell() frame = self.GetParent().GetParent().GetParent() frame.SetStatusText(text) def _on_selected_range(self, event): # Internal update to the selection tracking lists if event.Selecting(): # adding to the list... self.current_selection.update(xrange(event.GetTopRow(), event.GetBottomRow()+1)) else: # removal from list for index in xrange( event.GetTopRow(), event.GetBottomRow()+1): self.current_selection.discard(index) event.Skip() def _on_selected_cell(self, event): # Internal update to the selection tracking list self.current_selection = set([event.GetRow()]) event.Skip() def sort(self, key, reverse=False): """ Sort the current list of items using the key function ``key``. If ``reverse`` is true the sort order is reversed. """ row = self.GetGridCursorRow() col = self.GetGridCursorCol() curitem = self.table.items[row] # Remember where the cursor is now # Sort items def realkey(item): return key(item) try: self.table.items = ipipe.deque(sorted(self.table.items, key=realkey, reverse=reverse)) except TypeError, exc: self.error_output("Exception encountered: %s" % exc) return # Find out where the object under the cursor went for (i, item) in enumerate(self.table.items): if item is curitem: self.SetGridCursor(i,col) self.MakeCellVisible(i,col) self.Refresh() def sortattrasc(self): """ Sort in ascending order; sorting criteria is the current attribute """ col = self.GetGridCursorCol() attr = self.table.displayattrs[col] frame = self.GetParent().GetParent().GetParent() if attr is ipipe.noitem: self.error_output("no column under cursor") return frame.SetStatusText("sort by %s (ascending)" % attr.name()) def key(item): try: return attr.value(item) except (KeyboardInterrupt, SystemExit): raise except Exception: return None self.sort(key) def sortattrdesc(self): """ Sort in descending order; sorting criteria is the current attribute """ col = self.GetGridCursorCol() attr = self.table.displayattrs[col] frame = self.GetParent().GetParent().GetParent() if attr is ipipe.noitem: self.error_output("no column under cursor") return frame.SetStatusText("sort by %s (descending)" % attr.name()) def key(item): try: return attr.value(item) except (KeyboardInterrupt, SystemExit): raise except Exception: return None self.sort(key, reverse=True) def label_doubleclicked(self, event): row = event.GetRow() col = event.GetCol() if col == -1: self.enter(row) def _getvalue(self, row, col): """ Gets the text which is displayed at ``(row, col)`` """ try: value = self.table.displayattrs[col].value(self.table.items[row]) (align, width, text) = ipipe.xformat(value, "cell", self.maxchars) except IndexError: raise IndexError except Exception, exc: (align, width, text) = ipipe.xformat(exc, "cell", self.maxchars) return text def search(self, searchtext, startrow=0, startcol=0, search_forward=True): """ search for ``searchtext``, starting in ``(startrow, startcol)``; if ``search_forward`` is true the direction is "forward" """ row = startrow searchtext = searchtext.lower() if search_forward: while True: for col in xrange(startcol, self.table.GetNumberCols()): try: foo = self.table.GetValue(row, col) text = self._getvalue(row, col) if searchtext in text.string().lower(): self.SetGridCursor(row, col) self.MakeCellVisible(row, col) return except IndexError: return startcol = 0 row += 1 else: while True: for col in xrange(startcol, -1, -1): try: foo = self.table.GetValue(row, col) text = self._getvalue(row, col) if searchtext in text.string().lower(): self.SetGridCursor(row, col) self.MakeCellVisible(row, col) return except IndexError: return startcol = self.table.GetNumberCols()-1 row -= 1 def key_pressed(self, event): """ Maps pressed keys to functions """ frame = self.GetParent().GetParent().GetParent() frame.SetStatusText("") sh = event.ShiftDown() ctrl = event.ControlDown() keycode = event.GetKeyCode() if keycode == ord("P"): row = self.GetGridCursorRow() if event.ShiftDown(): col = self.GetGridCursorCol() self.pickattr(row, col) else: self.pick(row) elif keycode == ord("M"): if ctrl: col = self.GetGridCursorCol() self.pickrowsattr(sorted(self.current_selection), col) else: self.pickrows(sorted(self.current_selection)) elif keycode in (wx.WXK_BACK, wx.WXK_DELETE, ord("X")) and not (ctrl or sh): self.delete_current_notebook() elif keycode == ord("E") and not (ctrl or sh): row = self.GetGridCursorRow() self.enter(row) elif keycode == ord("E") and sh and not ctrl: row = self.GetGridCursorRow() col = self.GetGridCursorCol() self.enterattr(row, col) elif keycode == ord("E") and ctrl: row = self.GetGridCursorRow() self.SetGridCursor(row, self.GetNumberCols()-1) elif keycode == wx.WXK_HOME or (keycode == ord("A") and ctrl): row = self.GetGridCursorRow() self.SetGridCursor(row, 0) elif keycode == ord("C") and sh: col = self.GetGridCursorCol() attr = self.table.displayattrs[col] returnobj = [] for i in xrange(self.GetNumberRows()): returnobj.append(self.table.displayattrs[col].value(self.table.items[i])) self.quit(returnobj) elif keycode in (wx.WXK_ESCAPE, ord("Q")) and not (ctrl or sh): self.quit() elif keycode == ord("<"): row = self.GetGridCursorRow() col = self.GetGridCursorCol() if not event.ShiftDown(): newcol = col - 1 if newcol >= 0: self.SetGridCursor(row, col - 1) else: newcol = col + 1 if newcol < self.GetNumberCols(): self.SetGridCursor(row, col + 1) elif keycode == ord("D"): col = self.GetGridCursorCol() row = self.GetGridCursorRow() if not sh: self.detail(row, col) else: self.detail_attr(row, col) elif keycode == ord("F") and ctrl: frame.enter_searchtext(event) elif keycode == wx.WXK_F3: if sh: frame.find_previous(event) else: frame.find_next(event) elif keycode == ord("V"): if sh: self.sortattrdesc() else: self.sortattrasc() else: event.Skip() def delete_current_notebook(self): """ deletes the current notebook tab """ panel = self.GetParent() nb = panel.GetParent() current = nb.GetSelection() count = nb.GetPageCount() if count > 1: for i in xrange(count-1, current-1, -1): nb.DeletePage(i) nb.GetCurrentPage().grid.SetFocus() else: frame = nb.GetParent() frame.SetStatusText("This is the last level!") def _doenter(self, value, *attrs): """ "enter" a special item resulting in a new notebook tab """ panel = self.GetParent() nb = panel.GetParent() frame = nb.GetParent() current = nb.GetSelection() count = nb.GetPageCount() try: # if we want to enter something non-iterable, e.g. a function if current + 1 == count and value is not self.input: # we have an event in the last tab frame._add_notebook(value, *attrs) elif value != self.input: # we have to delete all tabs newer than [panel] first for i in xrange(count-1, current, -1): # some tabs don't close if we don't close in *reverse* order nb.DeletePage(i) frame._add_notebook(value) except TypeError, exc: if exc.__class__.__module__ == "exceptions": msg = "%s: %s" % (exc.__class__.__name__, exc) else: msg = "%s.%s: %s" % (exc.__class__.__module__, exc.__class__.__name__, exc) frame.SetStatusText(msg) def enterattr(self, row, col): try: attr = self.table.displayattrs[col] value = attr.value(self.table.items[row]) except Exception, exc: self.error_output(str(exc)) else: self._doenter(value) def enter(self, row): try: value = self.table.items[row] except Exception, exc: self.error_output(str(exc)) else: self._doenter(value) def detail(self, row, col): """ shows a detail-view of the current cell """ try: attr = self.table.displayattrs[col] item = self.table.items[row] except Exception, exc: self.error_output(str(exc)) else: attrs = [ipipe.AttributeDetail(item, attr) for attr in ipipe.xattrs(item, "detail")] self._doenter(attrs) def detail_attr(self, row, col): try: attr = self.table.displayattrs[col] item = attr.value(self.table.items[row]) except Exception, exc: self.error_output(str(exc)) else: attrs = [ipipe.AttributeDetail(item, attr) for attr in ipipe.xattrs(item, "detail")] self._doenter(attrs) def quit(self, returnobj=None): """ quit """ frame = self.GetParent().GetParent().GetParent() if frame.helpdialog: frame.helpdialog.Destroy() frame.parent.returnobj = returnobj frame.Close() frame.Destroy() def cell_doubleclicked(self, event): self.enterattr(event.GetRow(), event.GetCol()) def pick(self, row): """ pick a single row and return to the IPython prompt """ try: value = self.table.items[row] except Exception, exc: self.error_output(str(exc)) else: self.quit(value) def pickrows(self, rows): """ pick multiple rows and return to the IPython prompt """ try: value = [self.table.items[row] for row in rows] except Exception, exc: self.error_output(str(exc)) else: self.quit(value) def pickrowsattr(self, rows, col): """" pick one column from multiple rows """ values = [] try: attr = self.table.displayattrs[col] for row in rows: try: values.append(attr.value(self.table.items[row])) except (SystemExit, KeyboardInterrupt): raise except Exception: raise #pass except Exception, exc: self.error_output(str(exc)) else: self.quit(values) def pickattr(self, row, col): try: attr = self.table.displayattrs[col] value = attr.value(self.table.items[row]) except Exception, exc: self.error_output(str(exc)) else: self.quit(value) class IGridPanel(wx.Panel): # Each IGridPanel contains an IGridGrid def __init__(self, parent, input, *attrs): wx.Panel.__init__(self, parent, -1) self.grid = IGridGrid(self, input, *attrs) sizer = wx.BoxSizer(wx.VERTICAL) sizer.Add(self.grid, proportion=1, flag=wx.EXPAND | wx.ALL, border=10) self.SetSizer(sizer) sizer.Fit(self) sizer.SetSizeHints(self) class IGridHTMLHelp(wx.Frame): def __init__(self, parent, title, filename, size): wx.Frame.__init__(self, parent, -1, title, size=size) html = wx.html.HtmlWindow(self) if "gtk2" in wx.PlatformInfo: html.SetStandardFonts() html.LoadFile(filename) class IGridFrame(wx.Frame): maxtitlelen = 30 def __init__(self, parent, input): wx.Frame.__init__(self, None, title="IGrid", size=(640, 480)) self.menubar = wx.MenuBar() self.menucounter = 100 self.m_help = wx.Menu() self.m_search = wx.Menu() self.m_sort = wx.Menu() self.notebook = wx.Notebook(self, -1, style=0) self.statusbar = self.CreateStatusBar(1, wx.ST_SIZEGRIP) self.parent = parent self._add_notebook(input) self.Bind(wx.EVT_CLOSE, self.OnCloseWindow) self.makemenu(self.m_sort, "&Sort (asc)", "Sort ascending", self.sortasc) self.makemenu(self.m_sort, "Sort (&desc)", "Sort descending", self.sortdesc) self.makemenu(self.m_help, "&Help", "Help", self.display_help) self.makemenu(self.m_help, "&Show help in browser", "Show help in browser", self.display_help_in_browser) self.makemenu(self.m_search, "&Find text", "Find text", self.enter_searchtext) self.makemenu(self.m_search, "Find by &expression", "Find by expression", self.enter_searchexpression) self.makemenu(self.m_search, "Find &next", "Find next", self.find_next) self.makemenu(self.m_search, "Find &previous", "Find previous", self.find_previous) self.menubar.Append(self.m_search, "&Find") self.menubar.Append(self.m_sort, "&Sort") self.menubar.Append(self.m_help, "&Help") self.SetMenuBar(self.menubar) self.searchtext = "" self.helpdialog = None def sortasc(self, event): grid = self.notebook.GetPage(self.notebook.GetSelection()).grid grid.sortattrasc() def sortdesc(self, event): grid = self.notebook.GetPage(self.notebook.GetSelection()).grid grid.sortattrdesc() def find_previous(self, event): """ find previous occurrences """ if self.searchtext: grid = self.notebook.GetPage(self.notebook.GetSelection()).grid row = grid.GetGridCursorRow() col = grid.GetGridCursorCol() if col-1 >= 0: grid.search(self.searchtext, row, col-1, False) else: grid.search(self.searchtext, row-1, grid.table.GetNumberCols()-1, False) else: self.enter_searchtext(event) def find_next(self, event): """ find the next occurrence """ if self.searchtext: grid = self.notebook.GetPage(self.notebook.GetSelection()).grid row = grid.GetGridCursorRow() col = grid.GetGridCursorCol() if col+1 < grid.table.GetNumberCols(): grid.search(self.searchtext, row, col+1) else: grid.search(self.searchtext, row+1, 0) else: self.enter_searchtext(event) def display_help(self, event): """ Display a help dialog """ if self.helpdialog: self.helpdialog.Destroy() filename = os.path.join(os.path.dirname(__file__), "igrid_help.html") self.helpdialog = IGridHTMLHelp(None, title="Help", filename=filename, size=wx.Size(600,400)) self.helpdialog.Show() def display_help_in_browser(self, event): """ Show the help-HTML in a browser (as a ``HtmlWindow`` does not understand CSS this looks better) """ filename = urllib.pathname2url(os.path.abspath(os.path.join(os.path.dirname(__file__), "igrid_help.html"))) if not filename.startswith("file"): filename = "file:" + filename webbrowser.open(filename, new=1, autoraise=True) def enter_searchexpression(self, event): pass def makemenu(self, menu, label, help, cmd): menu.Append(self.menucounter, label, help) self.Bind(wx.EVT_MENU, cmd, id=self.menucounter) self.menucounter += 1 def _add_notebook(self, input, *attrs): # Adds another notebook which has the starting object ``input`` panel = IGridPanel(self.notebook, input, *attrs) text = str(ipipe.xformat(input, "header", self.maxtitlelen)[2]) if len(text) >= self.maxtitlelen: text = text[:self.maxtitlelen].rstrip(".") + "..." self.notebook.AddPage(panel, text, True) panel.grid.SetFocus() self.Layout() def OnCloseWindow(self, event): self.Destroy() def enter_searchtext(self, event): # Displays a dialog asking for the searchtext dlg = wx.TextEntryDialog(self, "Find:", "Find in list") if dlg.ShowModal() == wx.ID_OK: self.searchtext = dlg.GetValue() self.notebook.GetPage(self.notebook.GetSelection()).grid.search(self.searchtext, 0, 0) dlg.Destroy() class igrid(ipipe.Display): """ This is a wx-based display object that can be used instead of ``ibrowse`` (which is curses-based) or ``idump`` (which simply does a print). """ def display(self): self.returnobj = None app = wx.App() self.frame = IGridFrame(self, self.input) self.frame.Show() app.SetTopWindow(self.frame) self.frame.Raise() app.MainLoop() return self.returnobj