##// END OF EJS Templates
Do not generate output for empty figures in Qt console....
Do not generate output for empty figures in Qt console. Before, calling figure() would produce a big blank area. This ensures output is only returned if the figure has content. Also added basic testing for pylabtools, for which we had none.

File last commit:

r2267:928c921b
r3731:57d3cb1a
Show More
igrid.py
1126 lines | 40.8 KiB | text/x-python | PythonLexer
walter.doerwald
Fix linefeeds.
r538 # -*- coding: iso-8859-1 -*-
import ipipe, os, webbrowser, urllib
Brian Granger
ipapi.py => core/ipapi.py and imports updated.
r2027 from IPython.core import ipapi
walter.doerwald
Fix linefeeds.
r538 import wx
import wx.grid, wx.html
try:
sorted
except NameError:
from ipipe import sorted
vivainio
check for set (python 2.3 compatibility). Closes #209
r942 try:
set
except:
from sets import Set as set
walter.doerwald
Fix linefeeds.
r538
__all__ = ["igrid"]
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 help = """
<?xml version='1.0' encoding='iso-8859-1'?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=iso-8859-1" />
<link rel="stylesheet" href="igrid_help.css" type="text/css" />
<title>igrid help</title>
</head>
<body>
<h1>igrid help</h1>
<h2>Commands</h2>
<h3>pick (P)</h3>
<p>Pick the whole row (object is available as "_")</p>
<h3>pickattr (Shift-P)</h3>
<p>Pick the attribute under the cursor</p>
<h3>pickallattrs (Shift-C)</h3>
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 <p>Pick the complete column under the cursor (i.e. the attribute under the
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 cursor) from all currently fetched objects. These attributes will be returned
as a list.</p>
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 <h3>pickinput (I)</h3>
<p>Pick the current row as next input line in IPython. Additionally the row is stored as "_"</p>
<h3>pickinputattr (Shift-I)</h3>
<p>Pick the attribute under the cursor as next input line in IPython. Additionally the row is stored as "_"</p>
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 <h3>enter (E)</h3>
<p>Enter the object under the cursor. (what this mean depends on the object
itself, i.e. how it implements iteration). This opens a new browser 'level'.</p>
<h3>enterattr (Shift-E)</h3>
<p>Enter the attribute under the cursor.</p>
<h3>detail (D)</h3>
<p>Show a detail view of the object under the cursor. This shows the name,
type, doc string and value of the object attributes (and it might show more
attributes than in the list view, depending on the object).</p>
<h3>detailattr (Shift-D)</h3>
<p>Show a detail view of the attribute under the cursor.</p>
<h3>pickrows (M)</h3>
<p>Pick multiple selected rows (M)</p>
<h3>pickrowsattr (CTRL-M)</h3>
<p>From multiple selected rows pick the cells matching the attribute the cursor is in (CTRL-M)</p>
<h3>find (CTRL-F)</h3>
<p>Find text</p>
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 <h3>find_expression (CTRL-Shift-F)</h3>
<p>Find entries matching an expression</p>
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 <h3>find_next (F3)</h3>
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 <p>Find next occurrence</p>
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641
<h3>find_previous (Shift-F3)</h3>
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 <p>Find previous occurrence</p>
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641
<h3>sortattrasc (V)</h3>
<p>Sort the objects (in ascending order) using the attribute under the cursor as the sort key.</p>
<h3>sortattrdesc (Shift-V)</h3>
<p>Sort the objects (in descending order) using the attribute under the cursor as the sort key.</p>
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 <h3>refresh_once (R, F5)</h3>
<p>Refreshes the display by restarting the iterator</p>
<h3>refresh_every_second</h3>
<p>Refreshes the display by restarting the iterator every second until stopped by stop_refresh.</p>
<h3>refresh_interval</h3>
<p>Refreshes the display by restarting the iterator every X ms (X is a custom interval set by the user) until stopped by stop_refresh.</p>
<h3>stop_refresh</h3>
<p>Stops all refresh timers.</p>
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 <h3>leave (Backspace, DEL, X)</h3>
<p>Close current tab (and all the tabs to the right of the current one).</h3>
<h3>quit (ESC,Q)</h3>
<p>Quit igrid and return to the IPython prompt.</p>
<h2>Navigation</h2>
<h3>Jump to the last column of the current row (END, CTRL-E, CTRL-Right)</h3>
<h3>Jump to the first column of the current row (HOME, CTRL-A, CTRL-Left)</h3>
<h3>Move the cursor one column to the left (&lt;)</h3>
<h3>Move the cursor one column to the right (&gt;)</h3>
<h3>Jump to the first row in the current column (CTRL-Up)</h3>
<h3>Jump to the last row in the current column (CTRL-Down)</h3>
</body>
</html>
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 """
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Make a few igrid attributes private....
r542 value = self.table._displayattrs[col].value(self.table.items[row])
walter.doerwald
Fix linefeeds.
r538 (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 = []
walter.doerwald
Make a few igrid attributes private....
r542 self.attrs = [ipipe.upgradexattr(attr) for attr in attrs]
self._displayattrs = self.attrs[:]
self._displayattrset = set(self.attrs)
walter.doerwald
Fix linefeeds.
r538 self.fontsize = fontsize
walter.doerwald
Make a few igrid attributes private....
r542 self._fetch(1)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.timer = wx.Timer()
self.timer.Bind(wx.EVT_TIMER, self.refresh_content)
walter.doerwald
Fix linefeeds.
r538
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):
walter.doerwald
Make a few igrid attributes private....
r542 return len(self._displayattrs)
walter.doerwald
Fix linefeeds.
r538
def GetColLabelValue(self, col):
walter.doerwald
Make a few igrid attributes private....
r542 if col < len(self._displayattrs):
return self._displayattrs[col].name()
walter.doerwald
Fix linefeeds.
r538 else:
return ""
def GetRowLabelValue(self, row):
return str(row)
def IsEmptyCell(self, row, col):
return False
walter.doerwald
Make a few igrid attributes private....
r542 def _append(self, item):
self.items.append(item)
# Nothing to do if the set of attributes has been fixed by the user
if not self.attrs:
for attr in ipipe.xattrs(item):
attr = ipipe.upgradexattr(attr)
if attr not in self._displayattrset:
self._displayattrs.append(attr)
self._displayattrset.add(attr)
def _fetch(self, count):
walter.doerwald
Fix linefeeds.
r538 # Try to fill ``self.items`` with at least ``count`` objects.
have = len(self.items)
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
walter.doerwald
Fix bug that surfaced when the igrid input raised an...
r627 self._append(exc)
walter.doerwald
Fix linefeeds.
r538 self.iterator = None
break
else:
have += 1
walter.doerwald
Make a few igrid attributes private....
r542 self._append(item)
walter.doerwald
Fix linefeeds.
r538
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
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 sizing_needed = False
had_cols = len(self._displayattrs)
walter.doerwald
Fix linefeeds.
r538 had_rows = len(self.items)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 if row == had_rows - 1 and self.iterator is not None:
walter.doerwald
Make a few igrid attributes private....
r542 self._fetch(row + 20)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 sizing_needed = True
walter.doerwald
Fix linefeeds.
r538 have_rows = len(self.items)
walter.doerwald
Make a few igrid attributes private....
r542 have_cols = len(self._displayattrs)
walter.doerwald
Fix linefeeds.
r538 if have_rows > had_rows:
msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED, have_rows - had_rows)
self.GetView().ProcessTableMessage(msg)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 sizing_needed = True
walter.doerwald
Fix linefeeds.
r538 if row >= have_rows:
return ""
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 if have_cols != had_cols:
msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_NOTIFY_COLS_APPENDED, have_cols - had_cols)
walter.doerwald
Fix linefeeds.
r538 self.GetView().ProcessTableMessage(msg)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 sizing_needed = True
if sizing_needed:
self.GetView().AutoSizeColumns(False)
walter.doerwald
Fix linefeeds.
r538 return ""
def SetValue(self, row, col, value):
pass
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 def refresh_content(self, event):
msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_NOTIFY_ROWS_DELETED, 0, self.GetNumberRows())
self.GetView().ProcessTableMessage(msg)
self.iterator = ipipe.xiter(self.input)
self.items = []
self.attrs = [] # _append will calculate new displayattrs
self._fetch(1) # fetch one...
if self.items:
msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED, 1)
self.GetView().ProcessTableMessage(msg)
self.GetValue(0, 0) # and trigger "fetch next 20"
item = self.items[0]
self.GetView().AutoSizeColumns(False)
panel = self.GetView().GetParent()
nb = panel.GetParent()
current = nb.GetSelection()
if nb.GetPage(current) == panel:
self.GetView().set_footer(item)
walter.doerwald
Fix linefeeds.
r538
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)
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546 self.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.cell_leftclicked)
walter.doerwald
Fix linefeeds.
r538 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()
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 frame.SetStatusText(str(text))
walter.doerwald
Fix linefeeds.
r538
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
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 for index in xrange(event.GetTopRow(), event.GetBottomRow()+1):
walter.doerwald
Fix linefeeds.
r538 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):
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546 try:
return key(item)
except (KeyboardInterrupt, SystemExit):
raise
except Exception:
return None
walter.doerwald
Fix linefeeds.
r538 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()
walter.doerwald
Fix typo.
r547 attr = self.table._displayattrs[col]
walter.doerwald
Fix linefeeds.
r538 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()
walter.doerwald
Make a few igrid attributes private....
r542 attr = self.table._displayattrs[col]
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Make a few igrid attributes private....
r542 value = self.table._displayattrs[col].value(self.table.items[row])
walter.doerwald
Fix linefeeds.
r538 (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
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 def searchexpression(self, searchexp, startrow=None, search_forward=True ):
"""
Find by expression
"""
frame = self.GetParent().GetParent().GetParent()
if searchexp:
if search_forward:
if not startrow:
row = self.GetGridCursorRow()+1
else:
row = startrow + 1
while True:
try:
foo = self.table.GetValue(row, 0)
item = self.table.items[row]
try:
globals = ipipe.getglobals(None)
if eval(searchexp, globals, ipipe.AttrNamespace(item)):
self.SetGridCursor(row, 0) # found something
self.MakeCellVisible(row, 0)
break
except (KeyboardInterrupt, SystemExit):
raise
except Exception, exc:
frame.SetStatusText(str(exc))
wx.Bell()
break # break on error
except IndexError:
return
row += 1
else:
if not startrow:
row = self.GetGridCursorRow() - 1
else:
row = startrow - 1
while True:
try:
foo = self.table.GetValue(row, 0)
item = self.table.items[row]
try:
globals = ipipe.getglobals(None)
if eval(searchexp, globals, ipipe.AttrNamespace(item)):
self.SetGridCursor(row, 0) # found something
self.MakeCellVisible(row, 0)
break
except (KeyboardInterrupt, SystemExit):
raise
except Exception, exc:
frame.SetStatusText(str(exc))
wx.Bell()
break # break on error
except IndexError:
return
row -= 1
def search(self, searchtext, startrow=None, startcol=None, search_forward=True):
walter.doerwald
Fix linefeeds.
r538 """
search for ``searchtext``, starting in ``(startrow, startcol)``;
if ``search_forward`` is true the direction is "forward"
"""
searchtext = searchtext.lower()
if search_forward:
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 if startrow is not None and startcol is not None:
row = startrow
else:
startcol = self.GetGridCursorCol() + 1
row = self.GetGridCursorRow()
if startcol >= self.GetNumberCols():
startcol = 0
row += 1
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 if startrow is not None and startcol is not None:
row = startrow
else:
startcol = self.GetGridCursorCol() - 1
row = self.GetGridCursorRow()
if startcol < 0:
startcol = self.GetNumberCols() - 1
row -= 1
walter.doerwald
Fix linefeeds.
r538 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()
walter.doerwald
Map the return key to enter() and shift-return to enterattr().
r578 if sh:
walter.doerwald
Fix linefeeds.
r538 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()
walter.doerwald
Map the return key to enter() and shift-return to enterattr().
r578 elif keycode in (ord("E"), ord("\r")):
walter.doerwald
Fix linefeeds.
r538 row = self.GetGridCursorRow()
walter.doerwald
Map the return key to enter() and shift-return to enterattr().
r578 if sh:
col = self.GetGridCursorCol()
self.enterattr(row, col)
else:
self.enter(row)
walter.doerwald
Fix linefeeds.
r538 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()
walter.doerwald
Make a few igrid attributes private....
r542 attr = self.table._displayattrs[col]
walter.doerwald
Fix picking.
r573 result = []
walter.doerwald
Fix linefeeds.
r538 for i in xrange(self.GetNumberRows()):
walter.doerwald
Fix picking.
r573 result.append(self.table._displayattrs[col].value(self.table.items[i]))
self.quit(result)
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 if sh:
frame.enter_searchexpression(event)
else:
frame.enter_searchtext(event)
walter.doerwald
Fix linefeeds.
r538 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()
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546 elif keycode == wx.WXK_DOWN:
row = self.GetGridCursorRow()
try:
item = self.table.items[row+1]
except IndexError:
item = self.table.items[row]
self.set_footer(item)
event.Skip()
elif keycode == wx.WXK_UP:
row = self.GetGridCursorRow()
if row >= 1:
item = self.table.items[row-1]
else:
item = self.table.items[row]
self.set_footer(item)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 event.Skip()
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546 elif keycode == wx.WXK_RIGHT:
row = self.GetGridCursorRow()
item = self.table.items[row]
self.set_footer(item)
event.Skip()
elif keycode == wx.WXK_LEFT:
row = self.GetGridCursorRow()
item = self.table.items[row]
self.set_footer(item)
event.Skip()
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 elif keycode == ord("R") or keycode == wx.WXK_F5:
self.table.refresh_content(event)
elif keycode == ord("I"):
row = self.GetGridCursorRow()
if not sh:
self.pickinput(row)
else:
col = self.GetGridCursorCol()
self.pickinputattr(row, col)
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Make a few igrid attributes private....
r542 attr = self.table._displayattrs[col]
walter.doerwald
Fix linefeeds.
r538 value = attr.value(self.table.items[row])
except Exception, exc:
self.error_output(str(exc))
else:
self._doenter(value)
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546 def set_footer(self, item):
frame = self.GetParent().GetParent().GetParent()
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 frame.SetStatusText(" ".join([str(text) for (style, text) in ipipe.xformat(item, "footer", 20)[2]]), 0)
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Make a few igrid attributes private....
r542 attr = self.table._displayattrs[col]
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Make a few igrid attributes private....
r542 attr = self.table._displayattrs[col]
walter.doerwald
Fix linefeeds.
r538 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)
walter.doerwald
Fix picking.
r573 def quit(self, result=None):
walter.doerwald
Fix linefeeds.
r538 """
quit
"""
frame = self.GetParent().GetParent().GetParent()
walter.doerwald
Patch by Nik Tautenhahn: Close help window if...
r539 if frame.helpdialog:
frame.helpdialog.Destroy()
walter.doerwald
Using igrid with wxPyhon 2.6 and -wthread should work now....
r574 app = frame.parent
if app is not None:
app.result = result
walter.doerwald
Fix linefeeds.
r538 frame.Close()
frame.Destroy()
def cell_doubleclicked(self, event):
self.enterattr(event.GetRow(), event.GetCol())
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546 event.Skip()
walter.doerwald
Fix linefeeds.
r538
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546 def cell_leftclicked(self, event):
row = event.GetRow()
item = self.table.items[row]
self.set_footer(item)
event.Skip()
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708
walter.doerwald
Fix linefeeds.
r538 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)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 def pickinput(self, row):
try:
value = self.table.items[row]
except Exception, exc:
self.error_output(str(exc))
else:
api = ipapi.get()
api.set_next_input(str(value))
self.quit(value)
def pickinputattr(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:
api = ipapi.get()
api.set_next_input(str(value))
self.quit(value)
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Make a few igrid attributes private....
r542 attr = self.table._displayattrs[col]
walter.doerwald
Fix linefeeds.
r538 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:
walter.doerwald
Make a few igrid attributes private....
r542 attr = self.table._displayattrs[col]
walter.doerwald
Fix linefeeds.
r538 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)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.grid.FitInside()
walter.doerwald
Fix linefeeds.
r538 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):
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 def __init__(self, parent, title, size):
walter.doerwald
Fix linefeeds.
r538 wx.Frame.__init__(self, parent, -1, title, size=size)
html = wx.html.HtmlWindow(self)
if "gtk2" in wx.PlatformInfo:
html.SetStandardFonts()
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 html.SetPage(help)
walter.doerwald
Fix linefeeds.
r538
class IGridFrame(wx.Frame):
maxtitlelen = 30
def __init__(self, parent, input):
walter.doerwald
Nicer code.
r548 title = " ".join([str(text) for (style, text) in ipipe.xformat(input, "header", 20)[2]])
walter.doerwald
Show the input object of the igrid browser as the window tile....
r546 wx.Frame.__init__(self, None, title=title, size=(640, 480))
walter.doerwald
Fix linefeeds.
r538 self.menubar = wx.MenuBar()
self.menucounter = 100
self.m_help = wx.Menu()
self.m_search = wx.Menu()
self.m_sort = wx.Menu()
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.m_refresh = wx.Menu()
walter.doerwald
Fix linefeeds.
r538 self.notebook = wx.Notebook(self, -1, style=0)
self.statusbar = self.CreateStatusBar(1, wx.ST_SIZEGRIP)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.statusbar.SetFieldsCount(2)
self.SetStatusWidths([-1, 200])
walter.doerwald
Fix linefeeds.
r538 self.parent = parent
self._add_notebook(input)
self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.makemenu(self.m_sort, "&Sort (asc)\tV", "Sort ascending", self.sortasc)
self.makemenu(self.m_sort, "Sort (&desc)\tShift-V", "Sort descending", self.sortdesc)
self.makemenu(self.m_help, "&Help\tF1", "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\tCTRL-F", "Find text", self.enter_searchtext)
self.makemenu(self.m_search, "Find by &expression\tCTRL-Shift-F", "Find by expression", self.enter_searchexpression)
self.makemenu(self.m_search, "Find &next\tF3", "Find next", self.find_next)
self.makemenu(self.m_search, "Find &previous\tShift-F3", "Find previous", self.find_previous)
self.makemenu(self.m_refresh, "&Refresh once \tF5", "Refresh once", self.refresh_once)
self.makemenu(self.m_refresh, "Refresh every &1s", "Refresh every second", self.refresh_every_second)
self.makemenu(self.m_refresh, "Refresh every &X seconds", "Refresh every X seconds", self.refresh_interval)
self.makemenu(self.m_refresh, "&Stop all refresh timers", "Stop refresh timers", self.stop_refresh)
walter.doerwald
Fix linefeeds.
r538 self.menubar.Append(self.m_search, "&Find")
self.menubar.Append(self.m_sort, "&Sort")
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.menubar.Append(self.m_refresh, "&Refresh")
walter.doerwald
Fix linefeeds.
r538 self.menubar.Append(self.m_help, "&Help")
self.SetMenuBar(self.menubar)
self.searchtext = ""
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.searchexpression = ""
walter.doerwald
Patch by Nik Tautenhahn: Close help window if...
r539 self.helpdialog = None
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.refresh_interval = 1000
self.SetStatusText("Refreshing inactive", 1)
def refresh_once(self, event):
table = self.notebook.GetPage(self.notebook.GetSelection()).grid.table
table.refresh_content(event)
def refresh_interval(self, event):
table = self.notebook.GetPage(self.notebook.GetSelection()).grid.table
dlg = wx.TextEntryDialog(self, "Enter refresh interval (milliseconds):", "Refresh timer:", defaultValue=str(self.refresh_interval))
if dlg.ShowModal() == wx.ID_OK:
try:
milliseconds = int(dlg.GetValue())
except ValueError, exc:
self.SetStatusText(str(exc))
else:
table.timer.Start(milliseconds=milliseconds, oneShot=False)
self.SetStatusText("Refresh timer set to %s ms" % milliseconds)
self.SetStatusText("Refresh interval: %s ms" % milliseconds, 1)
self.refresh_interval = milliseconds
dlg.Destroy()
def stop_refresh(self, event):
for i in xrange(self.notebook.GetPageCount()):
nb = self.notebook.GetPage(i)
nb.grid.table.timer.Stop()
self.SetStatusText("Refreshing inactive", 1)
def refresh_every_second(self, event):
table = self.notebook.GetPage(self.notebook.GetSelection()).grid.table
table.timer.Start(milliseconds=1000, oneShot=False)
self.SetStatusText("Refresh interval: 1000 ms", 1)
walter.doerwald
Fix linefeeds.
r538
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
"""
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 grid = self.notebook.GetPage(self.notebook.GetSelection()).grid
walter.doerwald
Fix linefeeds.
r538 if self.searchtext:
row = grid.GetGridCursorRow()
col = grid.GetGridCursorCol()
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.SetStatusText('Search mode: text; looking for %s' % self.searchtext)
walter.doerwald
Fix linefeeds.
r538 if col-1 >= 0:
grid.search(self.searchtext, row, col-1, False)
else:
grid.search(self.searchtext, row-1, grid.table.GetNumberCols()-1, False)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 elif self.searchexpression:
self.SetStatusText("Search mode: expression; looking for %s" % repr(self.searchexpression)[2:-1])
grid.searchexpression(searchexp=self.searchexpression, search_forward=False)
walter.doerwald
Fix linefeeds.
r538 else:
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.SetStatusText("No search yet: please enter search-text or -expression")
walter.doerwald
Fix linefeeds.
r538
def find_next(self, event):
"""
find the next occurrence
"""
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 grid = self.notebook.GetPage(self.notebook.GetSelection()).grid
if self.searchtext != "":
walter.doerwald
Fix linefeeds.
r538 row = grid.GetGridCursorRow()
col = grid.GetGridCursorCol()
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.SetStatusText('Search mode: text; looking for %s' % self.searchtext)
walter.doerwald
Fix linefeeds.
r538 if col+1 < grid.table.GetNumberCols():
grid.search(self.searchtext, row, col+1)
else:
grid.search(self.searchtext, row+1, 0)
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 elif self.searchexpression != "":
self.SetStatusText('Search mode: expression; looking for %s' % repr(self.searchexpression)[2:-1])
grid.searchexpression(searchexp=self.searchexpression)
walter.doerwald
Fix linefeeds.
r538 else:
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.SetStatusText("No search yet: please enter search-text or -expression")
walter.doerwald
Fix linefeeds.
r538
def display_help(self, event):
"""
Display a help dialog
"""
walter.doerwald
Patch by Nik Tautenhahn: Close help window if...
r539 if self.helpdialog:
self.helpdialog.Destroy()
walter.doerwald
Incorporate html help into the module, so we don't have to search for the file.
r641 self.helpdialog = IGridHTMLHelp(None, title="Help", size=wx.Size(600,400))
walter.doerwald
Patch by Nik Tautenhahn: Close help window if...
r539 self.helpdialog.Show()
walter.doerwald
Fix linefeeds.
r538
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"):
walter.doerwald
Patch by Nik Tautenhahn: Close help window if...
r539 filename = "file:" + filename
walter.doerwald
Fix linefeeds.
r538 webbrowser.open(filename, new=1, autoraise=True)
walter.doerwald
Patch by Nik Tautenhahn: Close help window if...
r539
walter.doerwald
Fix linefeeds.
r538 def enter_searchexpression(self, event):
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 dlg = wx.TextEntryDialog(self, "Find:", "Find matching expression:", defaultValue=self.searchexpression)
if dlg.ShowModal() == wx.ID_OK:
self.searchexpression = dlg.GetValue()
self.searchtext = ""
self.SetStatusText('Search mode: expression; looking for %s' % repr(self.searchexpression)[2:-1])
self.notebook.GetPage(self.notebook.GetSelection()).grid.searchexpression(self.searchexpression)
dlg.Destroy()
walter.doerwald
Fix linefeeds.
r538
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
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 dlg = wx.TextEntryDialog(self, "Find:", "Find in list", defaultValue=self.searchtext)
walter.doerwald
Fix linefeeds.
r538 if dlg.ShowModal() == wx.ID_OK:
self.searchtext = dlg.GetValue()
walter.doerwald
Add new commands refresh and refresh_timer (mapped to "R"/"F5"...
r708 self.searchexpression = ""
self.SetStatusText('Search mode: text; looking for %s' % self.searchtext)
self.notebook.GetPage(self.notebook.GetSelection()).grid.search(self.searchtext)
walter.doerwald
Fix linefeeds.
r538 dlg.Destroy()
walter.doerwald
Try to get igrid to run with -wthread. This at least...
r559 class App(wx.App):
def __init__(self, input):
self.input = input
walter.doerwald
Fix picking.
r573 self.result = None # Result to be returned to IPython. Set by quit().
walter.doerwald
Try to get igrid to run with -wthread. This at least...
r559 wx.App.__init__(self)
def OnInit(self):
frame = IGridFrame(self, self.input)
frame.Show()
self.SetTopWindow(frame)
frame.Raise()
return True
walter.doerwald
Fix linefeeds.
r538 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).
"""
walter.doerwald
The input object can now be passed to the constructor of ibrowse/igrid....
r951
walter.doerwald
Using igrid with wxPyhon 2.6 and -wthread should work now....
r574 if wx.VERSION < (2, 7):
def display(self):
try:
walter.doerwald
Add an input argument to *all* Display constructors....
r953 # Try to create a "standalone" frame. If this works we're probably
walter.doerwald
Using igrid with wxPyhon 2.6 and -wthread should work now....
r574 # running with -wthread.
# Note that this sets the parent of the frame to None, but we can't
# pass a result object back to the shell anyway.
frame = IGridFrame(None, self.input)
frame.Show()
frame.Raise()
except wx.PyNoAppError:
# There's no wx application yet => create one.
app = App(self.input)
app.MainLoop()
return app.result
else:
# With wx 2.7 it gets simpler.
def display(self):
app = App(self.input)
app.MainLoop()
return app.result