class InstallExtensionWindow(wx.Frame): def __init__( self, parent, giface, id=wx.ID_ANY, title=_("Fetch & install extension from GRASS Addons"), **kwargs, ): self.parent = parent self._giface = giface self.options = dict() # list of options wx.Frame.__init__(self, parent=parent, id=id, title=title, **kwargs) self.SetIcon( wx.Icon(os.path.join(globalvar.ICONDIR, "grass.ico"), wx.BITMAP_TYPE_ICO)) self.panel = wx.Panel(parent=self, id=wx.ID_ANY) # self.repoBox = StaticBox( # parent=self.panel, id=wx.ID_ANY, label=" %s " % # _("Repository (leave empty to use the official one)")) self.treeBox = StaticBox( parent=self.panel, id=wx.ID_ANY, label=" %s " % _("List of extensions - double-click to install"), ) # self.repo = TextCtrl(parent=self.panel, id=wx.ID_ANY) # modelBuilder loads data into tree model self.modelBuilder = ExtensionTreeModelBuilder() # tree view displays model data self.tree = CTreeView(parent=self.panel, model=self.modelBuilder.GetModel()) self.search = SearchCtrl(self.panel) self.search.SetDescriptiveText(_("Search")) self.search.ShowCancelButton(True) # load data in different thread self.thread = gThread() self.optionBox = StaticBox(parent=self.panel, id=wx.ID_ANY, label=" %s " % _("Options")) task = gtask.parse_interface("g.extension") ignoreFlags = ["l", "c", "g", "a", "f", "t", "help", "quiet"] if sys.platform == "win32": ignoreFlags.append("d") ignoreFlags.append("i") for f in task.get_options()["flags"]: name = f.get("name", "") desc = f.get("label", "") if not desc: desc = f.get("description", "") if not name and not desc: continue if name in ignoreFlags: continue self.options[name] = wx.CheckBox(parent=self.panel, id=wx.ID_ANY, label=desc) # defaultUrl = '' # default/official one will be used when option empty # self.repo.SetValue( # task.get_param( # value='url').get( # 'default', # defaultUrl)) self.statusbar = self.CreateStatusBar(number=1) # self.btnFetch = Button(parent=self.panel, id=wx.ID_ANY, # label=_("&Fetch")) # self.btnFetch.SetToolTip(_("Fetch list of available modules " # "from GRASS Addons repository")) self.btnClose = Button(parent=self.panel, id=wx.ID_CLOSE) self.btnInstall = Button(parent=self.panel, id=wx.ID_ANY, label=_("&Install")) self.btnInstall.SetToolTip(_("Install selected add-ons GRASS module")) self.btnInstall.Enable(False) self.btnHelp = Button(parent=self.panel, id=wx.ID_HELP) self.btnHelp.SetToolTip(_("Show g.extension manual page")) self.btnClose.Bind(wx.EVT_BUTTON, lambda evt: self.Close()) # self.btnFetch.Bind(wx.EVT_BUTTON, self.OnFetch) self.btnInstall.Bind(wx.EVT_BUTTON, self.OnInstall) self.btnHelp.Bind(wx.EVT_BUTTON, self.OnHelp) self.search.Bind(wx.EVT_TEXT, lambda evt: self.Filter(evt.GetString())) self.search.Bind(wx.EVT_SEARCHCTRL_CANCEL_BTN, lambda evt: self.Filter("")) self.tree.selectionChanged.connect(self.OnItemSelected) self.tree.itemActivated.connect(self.OnItemActivated) self.tree.contextMenu.connect(self.OnContextMenu) wx.CallAfter(self._fetch) self._layout() def _layout(self): """Do layout""" sizer = wx.BoxSizer(wx.VERTICAL) # repoSizer = wx.StaticBoxSizer(self.repoBox, wx.VERTICAL) # repo1Sizer = wx.BoxSizer(wx.HORIZONTAL) # repo1Sizer.Add(self.repo, proportion=1, # flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL, border=1) # repo1Sizer.Add(self.btnFetch, proportion=0, # flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL, border=1) # repoSizer.Add(repo1Sizer, # flag=wx.EXPAND) sizer.Add(self.search, proportion=0, flag=wx.EXPAND | wx.ALL, border=3) treeSizer = wx.StaticBoxSizer(self.treeBox, wx.HORIZONTAL) treeSizer.Add(self.tree, proportion=1, flag=wx.ALL | wx.EXPAND, border=1) # options optionSizer = wx.StaticBoxSizer(self.optionBox, wx.VERTICAL) for key in self.options.keys(): optionSizer.Add(self.options[key], proportion=0) btnSizer = wx.BoxSizer(wx.HORIZONTAL) btnSizer.Add(self.btnHelp, proportion=0) btnSizer.AddStretchSpacer() btnSizer.Add(self.btnClose, proportion=0, flag=wx.RIGHT, border=5) btnSizer.Add(self.btnInstall, proportion=0) # sizer.Add(repoSizer, proportion=0, # flag=wx.ALL | wx.EXPAND, border=3) sizer.Add( treeSizer, proportion=1, flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=3, ) sizer.Add( optionSizer, proportion=0, flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=3, ) sizer.Add(btnSizer, proportion=0, flag=wx.ALL | wx.EXPAND, border=5) self.panel.SetSizer(sizer) sizer.Fit(self.panel) self.Layout() def _getCmd(self): item = self.tree.GetSelected() if not item or "command" not in item[0].data: GError(_("Extension not defined"), parent=self) return name = item[0].data["command"] flags = list() for key in self.options.keys(): if self.options[key].IsChecked(): if len(key) == 1: flags.append("-%s" % key) else: flags.append("--%s" % key) # 'url=' + self.repo.GetValue().strip()] return ["g.extension"] + flags + ["extension={}".format(name)] def OnFetch(self, event): """Fetch list of available extensions""" self._fetch() def _fetch(self): """Fetch list of available extensions""" wx.BeginBusyCursor() self.SetStatusText( _("Fetching list of modules from GRASS-Addons (be patient)..."), 0) try: self.thread.Run( callable=self.modelBuilder.Load, url="", # self.repo.GetValue().strip(), ondone=lambda event: self._fetchDone(), ) except GException as error: self._fetchDone() GError(str(error), parent=self, showTraceback=False) def _fetchDone(self): self.tree.RefreshItems() nitems = len(self.modelBuilder.GetModel().SearchNodes(key="command", value="*")) self.SetStatusText(_("%d extensions loaded") % nitems, 0) wx.EndBusyCursor() def Filter(self, text): model = self.modelBuilder.GetModel() if text: model = model.Filtered(key=["command", "keywords", "description"], value=text) self.tree.SetModel(model) self.tree.ExpandAll() else: self.tree.SetModel(model) def OnContextMenu(self, node): if not hasattr(self, "popupID"): self.popupID = dict() for key in ("install", "help"): self.popupID[key] = NewId() data = node.data if data and "command" in data: self.popupMenu = Menu() self.popupMenu.Append(self.popupID["install"], _("Install")) self.Bind(wx.EVT_MENU, self.OnInstall, id=self.popupID["install"]) self.popupMenu.AppendSeparator() self.popupMenu.Append(self.popupID["help"], _("Show manual page")) self.Bind(wx.EVT_MENU, self.OnItemHelp, id=self.popupID["help"]) self.PopupMenu(self.popupMenu) self.popupMenu.Destroy() def OnItemActivated(self, node): data = node.data if data and "command" in data: self.OnInstall(event=None) def OnInstall(self, event): """Install selected extension""" log = self.parent.GetLogWindow() cmd = self._getCmd() if cmd: log.RunCmd(cmd, onDone=self.OnDone) def OnDone(self, event): if event.returncode == 0: if not os.getenv("GRASS_ADDON_BASE"): SetAddOnPath(key="BASE") globalvar.UpdateGRASSAddOnCommands() toolboxesOutdated() def OnItemHelp(self, event): item = self.tree.GetSelected() if not item or "command" not in item[0].data: return self._giface.Help(entry=item[0].data["command"], online=True) def OnHelp(self, event): self._giface.Help(entry="g.extension") def OnItemSelected(self, node): """Item selected""" data = node.data if data is None: self.SetStatusText("", 0) self.btnInstall.Enable(False) else: self.SetStatusText(data.get("description", ""), 0) self.btnInstall.Enable(True)
class InstallExtensionWindow(wx.Frame): def __init__(self, parent, giface, id=wx.ID_ANY, title=_("Fetch & install extension from GRASS Addons"), **kwargs): self.parent = parent self._giface = giface self.options = dict() # list of options wx.Frame.__init__(self, parent=parent, id=id, title=title, **kwargs) self.SetIcon( wx.Icon(os.path.join(globalvar.ICONDIR, 'grass.ico'), wx.BITMAP_TYPE_ICO)) self.panel = wx.Panel(parent=self, id=wx.ID_ANY) self.repoBox = wx.StaticBox( parent=self.panel, id=wx.ID_ANY, label=" %s " % _("Repository (leave empty to use the official one)")) self.treeBox = wx.StaticBox( parent=self.panel, id=wx.ID_ANY, label=" %s " % _("List of extensions - double-click to install")) self.repo = wx.TextCtrl(parent=self.panel, id=wx.ID_ANY) # modelBuilder loads data into tree model self.modelBuilder = ExtensionTreeModelBuilder() # tree view displays model data self.tree = CTreeView(parent=self.panel, model=self.modelBuilder.GetModel()) self.search = SearchModuleWidget(parent=self.panel, model=self.modelBuilder.GetModel(), showChoice=False) self.search.showSearchResult.connect( lambda result: self.tree.Select(result)) # show text in statusbar when notification appears self.search.showNotification.connect( lambda message: self.SetStatusText(message)) # load data in different thread self.thread = gThread() self.optionBox = wx.StaticBox(parent=self.panel, id=wx.ID_ANY, label=" %s " % _("Options")) task = gtask.parse_interface('g.extension') ignoreFlags = ['l', 'c', 'g', 'a', 'f', 't', 'help', 'quiet'] if sys.platform == 'win32': ignoreFlags.append('d') ignoreFlags.append('i') for f in task.get_options()['flags']: name = f.get('name', '') desc = f.get('label', '') if not desc: desc = f.get('description', '') if not name and not desc: continue if name in ignoreFlags: continue self.options[name] = wx.CheckBox(parent=self.panel, id=wx.ID_ANY, label=desc) defaultUrl = '' # default/official one will be used when option empty self.repo.SetValue( task.get_param(value='url').get('default', defaultUrl)) self.statusbar = self.CreateStatusBar(number=1) self.btnFetch = wx.Button(parent=self.panel, id=wx.ID_ANY, label=_("&Fetch")) self.btnFetch.SetToolTipString( _("Fetch list of available modules " "from GRASS Addons SVN repository")) self.btnClose = wx.Button(parent=self.panel, id=wx.ID_CLOSE) self.btnInstall = wx.Button(parent=self.panel, id=wx.ID_ANY, label=_("&Install")) self.btnInstall.SetToolTipString( _("Install selected add-ons GRASS module")) self.btnInstall.Enable(False) self.btnHelp = wx.Button(parent=self.panel, id=wx.ID_HELP) self.btnHelp.SetToolTipString(_("Show g.extension manual page")) self.btnClose.Bind(wx.EVT_BUTTON, lambda evt: self.Close()) self.btnFetch.Bind(wx.EVT_BUTTON, self.OnFetch) self.btnInstall.Bind(wx.EVT_BUTTON, self.OnInstall) self.btnHelp.Bind(wx.EVT_BUTTON, self.OnHelp) self.tree.selectionChanged.connect(self.OnItemSelected) self.tree.itemActivated.connect(self.OnItemActivated) self.tree.contextMenu.connect(self.OnContextMenu) wx.CallAfter(self._fetch) self._layout() def _layout(self): """Do layout""" sizer = wx.BoxSizer(wx.VERTICAL) repoSizer = wx.StaticBoxSizer(self.repoBox, wx.VERTICAL) repo1Sizer = wx.BoxSizer(wx.HORIZONTAL) repo1Sizer.Add(item=self.repo, proportion=1, flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL, border=1) repo1Sizer.Add(item=self.btnFetch, proportion=0, flag=wx.ALL | wx.ALIGN_CENTER_VERTICAL, border=1) repoSizer.Add(item=repo1Sizer, flag=wx.EXPAND) findSizer = wx.BoxSizer(wx.HORIZONTAL) findSizer.Add(item=self.search, proportion=1) treeSizer = wx.StaticBoxSizer(self.treeBox, wx.HORIZONTAL) treeSizer.Add(item=self.tree, proportion=1, flag=wx.ALL | wx.EXPAND, border=1) # options optionSizer = wx.StaticBoxSizer(self.optionBox, wx.VERTICAL) for key in self.options.keys(): optionSizer.Add(item=self.options[key], proportion=0) btnSizer = wx.BoxSizer(wx.HORIZONTAL) btnSizer.Add(item=self.btnHelp, proportion=0) btnSizer.AddStretchSpacer() btnSizer.Add(item=self.btnClose, proportion=0, flag=wx.RIGHT, border=5) btnSizer.Add(item=self.btnInstall, proportion=0) sizer.Add(item=repoSizer, proportion=0, flag=wx.ALL | wx.EXPAND, border=3) sizer.Add(item=findSizer, proportion=0, flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=3) sizer.Add(item=treeSizer, proportion=1, flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=3) sizer.Add(item=optionSizer, proportion=0, flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, border=3) sizer.Add(item=btnSizer, proportion=0, flag=wx.ALIGN_RIGHT | wx.ALL | wx.EXPAND, border=5) self.panel.SetSizer(sizer) sizer.Fit(self.panel) self.Layout() def _getCmd(self): item = self.tree.GetSelected() if not item or 'command' not in item[0].data: GError(_("Extension not defined"), parent=self) return name = item[0].data['command'] flags = list() for key in self.options.keys(): if self.options[key].IsChecked(): if len(key) == 1: flags.append('-%s' % key) else: flags.append('--%s' % key) return ['g.extension'] + flags + [ 'extension=' + name, 'url=' + self.repo.GetValue().strip() ] def OnFetch(self, event): """Fetch list of available extensions""" self._fetch() def _fetch(self): """Fetch list of available extensions""" wx.BeginBusyCursor() self.SetStatusText( _("Fetching list of modules from GRASS-Addons SVN (be patient)..." ), 0) try: self.thread.Run(callable=self.modelBuilder.Load, url=self.repo.GetValue().strip(), ondone=lambda event: self._fetchDone()) except GException as e: self._fetchDone() GError(unicode(e), parent=self, showTraceback=False) def _fetchDone(self): self.tree.RefreshItems() nitems = len(self.modelBuilder.GetModel().SearchNodes(key='command', value='*')) self.SetStatusText(_("%d extensions loaded") % nitems, 0) wx.EndBusyCursor() def OnContextMenu(self, node): if not hasattr(self, "popupID"): self.popupID = dict() for key in ('install', 'help'): self.popupID[key] = wx.NewId() data = node.data if data and 'command' in data: self.popupMenu = wx.Menu() self.popupMenu.Append(self.popupID['install'], text=_("Install")) self.Bind(wx.EVT_MENU, self.OnInstall, id=self.popupID['install']) self.popupMenu.AppendSeparator() self.popupMenu.Append(self.popupID['help'], text=_("Show manual page")) self.Bind(wx.EVT_MENU, self.OnItemHelp, id=self.popupID['help']) self.PopupMenu(self.popupMenu) self.popupMenu.Destroy() def OnItemActivated(self, node): data = node.data if data and 'command' in data: self.OnInstall(event=None) def OnInstall(self, event): """Install selected extension""" log = self.parent.GetLogWindow() cmd = self._getCmd() if cmd: log.RunCmd(cmd, onDone=self.OnDone) def OnDone(self, event): if event.returncode == 0: if not os.getenv('GRASS_ADDON_BASE'): SetAddOnPath(key='BASE') globalvar.UpdateGRASSAddOnCommands() toolboxesOutdated() def OnItemHelp(self, event): item = self.tree.GetSelected() if not item or 'command' not in item[0].data: return self._giface.Help(entry=item[0].data['command'], online=True) def OnHelp(self, event): self._giface.Help(entry='g.extension') def OnItemSelected(self, node): """Item selected""" data = node.data if data is None: self.SetStatusText('', 0) self.btnInstall.Enable(False) else: self.SetStatusText(data.get('description', ''), 0) self.btnInstall.Enable(True)