def __init__(self, parent, data=None):
wx.Dialog.__init__(self,
parent,
id=wx.ID_ANY,
title=_("Query results"),
size=(420, 400),
style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER)
# send query output to console
self.redirectOutput = Signal('QueryDialog.redirectOutput')
self.data = data
self.panel = wx.Panel(self, id=wx.ID_ANY)
self.mainSizer = wx.BoxSizer(wx.VERTICAL)
helpText = StaticText(
self.panel,
wx.ID_ANY,
label=_("Right click to copy selected values to clipboard."))
helpText.SetForegroundColour(
wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT))
self.mainSizer.Add(helpText, proportion=0, flag=wx.ALL, border=5)
self._colNames = [_("Feature"), _("Value")]
self._model = QueryTreeBuilder(self.data, column=self._colNames[1])
self.tree = TreeListView(model=self._model,
parent=self.panel,
columns=self._colNames,
style=wx.TR_DEFAULT_STYLE | wx.TR_HIDE_ROOT
| wx.TR_FULL_ROW_HIGHLIGHT | wx.TR_MULTIPLE)
self.tree.SetColumnWidth(0, 220)
self.tree.SetColumnWidth(1, 1000)
self.tree.ExpandAll(self._model.root)
self.tree.RefreshItems()
self.tree.contextMenu.connect(self.ShowContextMenu)
self.mainSizer.Add(self.tree,
proportion=1,
flag=wx.EXPAND | wx.ALL,
border=5)
close = Button(self.panel, id=wx.ID_CLOSE)
close.Bind(wx.EVT_BUTTON, lambda event: self.Close())
copy = Button(self.panel,
id=wx.ID_ANY,
label=_("Copy all to clipboard"))
copy.Bind(wx.EVT_BUTTON, self.Copy)
self.Bind(wx.EVT_CLOSE, self.OnClose)
self.redirect = wx.CheckBox(self.panel, label=_("Redirect to console"))
self.redirect.SetValue(False)
self.redirect.Bind(wx.EVT_CHECKBOX,
lambda evt: self._onRedirect(evt.IsChecked()))
hbox = wx.BoxSizer(wx.HORIZONTAL)
hbox.Add(self.redirect,
proportion=0,
flag=wx.EXPAND | wx.RIGHT,
border=5)
hbox.AddStretchSpacer(1)
hbox.Add(copy, proportion=0, flag=wx.EXPAND | wx.RIGHT, border=5)
hbox.Add(close, proportion=0, flag=wx.EXPAND | wx.ALL, border=0)
self.mainSizer.Add(hbox,
proportion=0,
flag=wx.EXPAND | wx.ALL,
border=5)
self.panel.SetSizer(self.mainSizer)
self.mainSizer.Fit(self.panel)
# for Windows
self.SendSizeEvent()
class QueryDialog(wx.Dialog):
def __init__(self, parent, data=None):
wx.Dialog.__init__(self,
parent,
id=wx.ID_ANY,
title=_("Query results"),
size=(420, 400),
style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER)
# send query output to console
self.redirectOutput = Signal('QueryDialog.redirectOutput')
self.data = data
self.panel = wx.Panel(self, id=wx.ID_ANY)
self.mainSizer = wx.BoxSizer(wx.VERTICAL)
helpText = StaticText(
self.panel,
wx.ID_ANY,
label=_("Right click to copy selected values to clipboard."))
helpText.SetForegroundColour(
wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT))
self.mainSizer.Add(helpText, proportion=0, flag=wx.ALL, border=5)
self._colNames = [_("Feature"), _("Value")]
self._model = QueryTreeBuilder(self.data, column=self._colNames[1])
self.tree = TreeListView(model=self._model,
parent=self.panel,
columns=self._colNames,
style=wx.TR_DEFAULT_STYLE | wx.TR_HIDE_ROOT
| wx.TR_FULL_ROW_HIGHLIGHT | wx.TR_MULTIPLE)
self.tree.SetColumnWidth(0, 220)
self.tree.SetColumnWidth(1, 1000)
self.tree.ExpandAll(self._model.root)
self.tree.RefreshItems()
self.tree.contextMenu.connect(self.ShowContextMenu)
self.mainSizer.Add(self.tree,
proportion=1,
flag=wx.EXPAND | wx.ALL,
border=5)
close = Button(self.panel, id=wx.ID_CLOSE)
close.Bind(wx.EVT_BUTTON, lambda event: self.Close())
copy = Button(self.panel,
id=wx.ID_ANY,
label=_("Copy all to clipboard"))
copy.Bind(wx.EVT_BUTTON, self.Copy)
self.Bind(wx.EVT_CLOSE, self.OnClose)
self.redirect = wx.CheckBox(self.panel, label=_("Redirect to console"))
self.redirect.SetValue(False)
self.redirect.Bind(wx.EVT_CHECKBOX,
lambda evt: self._onRedirect(evt.IsChecked()))
hbox = wx.BoxSizer(wx.HORIZONTAL)
hbox.Add(self.redirect,
proportion=0,
flag=wx.EXPAND | wx.RIGHT,
border=5)
hbox.AddStretchSpacer(1)
hbox.Add(copy, proportion=0, flag=wx.EXPAND | wx.RIGHT, border=5)
hbox.Add(close, proportion=0, flag=wx.EXPAND | wx.ALL, border=0)
self.mainSizer.Add(hbox,
proportion=0,
flag=wx.EXPAND | wx.ALL,
border=5)
self.panel.SetSizer(self.mainSizer)
self.mainSizer.Fit(self.panel)
# for Windows
self.SendSizeEvent()
def SetData(self, data):
state = self.tree.GetExpansionState()
self.data = data
self._model = QueryTreeBuilder(self.data, column=self._colNames[1])
self.tree.SetModel(self._model)
self.tree.SetExpansionState(state)
if self.redirect.IsChecked():
self.redirectOutput.emit(output=self._textToRedirect())
def Copy(self, event):
text = printResults(self._model, self._colNames[1])
self._copyText(text)
def ShowContextMenu(self, node):
"""Show context menu.
Menu for copying distinguishes single and multiple selection.
"""
nodes = self.tree.GetSelected()
if not nodes:
return
menu = Menu()
texts = []
if len(nodes) > 1:
values = []
for node in nodes:
values.append(
(node.label,
node.data[self._colNames[1]] if node.data else ''))
col1 = '\n'.join([val[1] for val in values if val[1]])
col2 = '\n'.join([val[0] for val in values if val[0]])
table = '\n'.join([val[0] + ': ' + val[1] for val in values])
texts.append(
(_("Copy from '%s' column") % self._colNames[1], col1))
texts.append(
(_("Copy from '%s' column") % self._colNames[0], col2))
texts.append((_("Copy selected lines"), table))
else:
label1 = nodes[0].label
texts.append((_("Copy '%s'" % self._cutLabel(label1)), label1))
if nodes[0].data and nodes[0].data[self._colNames[1]]:
label2 = nodes[0].data[self._colNames[1]]
texts.insert(0,
(_("Copy '%s'" % self._cutLabel(label2)), label2))
texts.append((_("Copy line"), label1 + ': ' + label2))
ids = []
for text in texts:
id = NewId()
ids.append(id)
self.Bind(wx.EVT_MENU,
lambda evt, t=text[1], id=id: self._copyText(t),
id=id)
menu.Append(id, text[0])
# show the popup menu
self.PopupMenu(menu)
menu.Destroy()
for id in ids:
self.Unbind(wx.EVT_MENU, id=id)
def _onRedirect(self, redirect):
"""Emits instructions to redirect query results.
:param redirect: True to start redirecting, False to stop
"""
if redirect:
self.redirectOutput.emit(output=_("Query results:"), style='cmd')
self.redirectOutput.emit(output=self._textToRedirect())
else:
self.redirectOutput.emit(output=_(" "), style='cmd')
def _textToRedirect(self):
text = printResults(self._model, self._colNames[1])
text += '\n' + "-" * 50 + '\n'
return text
def _cutLabel(self, label):
limit = 15
if len(label) > limit:
return label[:limit] + '...'
return label
def _copyText(self, text):
"""Helper function for copying"""
if wx.TheClipboard.Open():
do = wx.TextDataObject()
do.SetText(text)
wx.TheClipboard.SetData(do)
wx.TheClipboard.Close()
def OnClose(self, event):
if self.redirect.IsChecked():
self._onRedirect(False)
self.Destroy()
event.Skip()