# Programmer: Daniel Pozmanter # E-mail: drpython@bluebottle.com # Note: You must reply to the verification e-mail to get through. # # Copyright 2003-2007 Daniel Pozmanter # # Distributed under the terms of the GPL (GNU Public License) # # DrPython is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA #Plugins Menu import os, shutil, sys import wx import drScrolledMessageDialog import drFileDialog from drMenu import drMenu class drPluginConfigureMenu(drMenu): def __init__(self, parent): drMenu.__init__(self, parent) self.parent = parent self.ID_INSTALL = 4300 self.ID_INSTALL_PY = 4301 self.ID_UNINSTALL = 4302 self.ID_INDEX = 4303 self.ID_EDIT = 4304 self.wildcard = "DrPython Plugin (*.py)|*.py" self.Append(self.ID_INSTALL, "&Install...") self.Append(self.ID_INSTALL_PY, "Install From Py...") self.Append(self.ID_UNINSTALL, "&UnInstall...") self.Append(self.ID_INDEX, "&Edit Indexes...") self.Append(self.ID_EDIT, "Edit &Plugin Source...") self.parent.Bind(wx.EVT_MENU, self.OnInstall, id=self.ID_INSTALL) self.parent.Bind(wx.EVT_MENU, self.OnInstallFromPy, id=self.ID_INSTALL_PY) self.parent.Bind(wx.EVT_MENU, self.OnUnInstall, id=self.ID_UNINSTALL) self.parent.Bind(wx.EVT_MENU, self.OnEditIndex, id=self.ID_INDEX) self.parent.Bind(wx.EVT_MENU, self.OnEditPlugin, id=self.ID_EDIT) def EditPlugin(self, plugin): pluginfile = os.path.join(self.parent.pluginsdirectory, plugin) + ".py" #Patch From Franz, Check for already open. #(Slightly edited by Dan, if ... in replaces try ... except. self.parent.OpenOrSwitchToFile(pluginfile) def OnEditIndex(self, event): from drPluginDialog import drEditIndexDialog d = drEditIndexDialog(self.parent) d.ShowModal() d.Destroy() def OnEditPlugin(self, event): plist = os.listdir(self.parent.pluginsdirectory) PluginList = [] for p in plist: i = p.find(".py") l = len(p) if i > -1 and (i + 3 == l): PluginList.append(p[:i]) PluginList.sort() try: d = wx.SingleChoiceDialog(self.parent, "Select the Plugin to Edit:", "Edit Plugin", PluginList, wx.CHOICEDLG_STYLE) d.SetSize(wx.Size(250, 250)) answer = d.ShowModal() d.Destroy() if answer == wx.ID_OK: self.EditPlugin(d.GetStringSelection()) except: drScrolledMessageDialog.ShowMessage(self.parent, "Error Editing Plugin", "Edit Plugin Error") def OnInstall(self, event): from drPluginDialog import drPluginInstallWizard img = wx.Image(self.parent.bitmapdirectory + "/install.wizard.png", wx.BITMAP_TYPE_PNG) installWizard = drPluginInstallWizard(self.parent, "Install DrPython Plugin(s)", wx.BitmapFromImage(img)) installWizard.Run() def OnInstallFromPy (self, event): dlg = drFileDialog.FileDialog(self.parent, "Select Plugin to Install", self.wildcard) if self.parent.pluginsdirectory: try: dlg.SetDirectory(self.parent.pluginsdirectory) except: drScrolledMessageDialog.ShowMessage(self.parent, ("Error Setting Default Directory To: " + self.parent.pluginsdirectory), "DrPython Error") if dlg.ShowModal() == wx.ID_OK: if not os.path.exists(self.parent.pluginsdirectory): os.mkdir(self.parent.pluginsdirectory) pluginfile = dlg.GetPath().replace("\\", "/") pluginrfile = os.path.join(self.parent.pluginsdirectory, os.path.split(pluginfile)[1]) pluginpath, plugin = os.path.split(pluginfile) i = plugin.find(".py") if i == -1: drScrolledMessageDialog.ShowMessage(self.parent, ("Plugins must be .py files."), "DrPython Error") plugininstallfile = pluginfile + ".install" continueinstallation = True if os.path.exists(plugininstallfile): f = open(plugininstallfile, 'r') scripttext = f.read() f.close() try: code = compile((scripttext + '\n'), plugininstallfile, 'exec') except: drScrolledMessageDialog.ShowMessage(self.parent, ("Error compiling install script."), "Error", wx.DefaultPosition, wx.Size(550,300)) return try: cwd = os.getcwd() os.chdir(pluginpath) exec(code) continueinstallation = Install(self.parent) os.chdir(cwd) except: drScrolledMessageDialog.ShowMessage(self.parent, ("Error running install script."), "Error", wx.DefaultPosition, wx.Size(550,300)) return if not continueinstallation: return plugin = plugin[:i] pluginsfile = self.parent.preferencesdirectory + "/default.idx" if not os.path.exists(pluginsfile): f = file(pluginsfile, 'wb') f.write('\n') f.close() try: copyf = True if os.path.exists(pluginrfile): answer = wx.MessageBox('Overwrite"' + pluginrfile + '"?', "DrPython", wx.YES_NO | wx.CANCEL | wx.ICON_QUESTION) if answer == wx.NO: copyf = False if copyf: shutil.copyfile(pluginfile, pluginrfile) #there could be an error check: if idx file does not exist in source, simply create one. shutil.copyfile(os.path.splitext(pluginfile)[0] + '.idx', os.path.splitext(pluginrfile)[0] + '.idx') try: f = open(pluginsfile, 'rU') pluginstoload = [x.strip() for x in f] f.close() except: pluginstoload = [] try: pluginstoload.index(plugin) except: f = file(pluginsfile, 'w') for p in pluginstoload: f.write(p + "\n") f.write(plugin) f.close() except: drScrolledMessageDialog.ShowMessage(self.parent, ("Error with: " + pluginfile), "Install Error") return def OnUnInstall(self, event): from drPluginDialog import drPluginUnInstallWizard img = wx.Image(self.parent.bitmapdirectory + "/uninstall.wizard.png", wx.BITMAP_TYPE_PNG) uninstallWizard = drPluginUnInstallWizard(self.parent, "UnInstall DrPython Plugin(s)", wx.BitmapFromImage(img)) uninstallWizard.Run() class drPluginIndexMenu(wx.Menu): def __init__(self, parent): wx.Menu.__init__(self) self.ID_LOAD_PLUGIN_BASE = 4505 self.parent = parent self.indexes = [] self.loadedindexes = [self.parent.preferencesdirectory + "/default.idx"] self.setupMenu() def OnLoadPluginsFromIndex(self, event): idnum = event.GetId() i = idnum - self.ID_LOAD_PLUGIN_BASE index = self.indexes[i] try: f = file(index, 'r') pluginstoload = f.read().rstrip().split('\n') f.close() for plugin in pluginstoload: try: self.parent.InitializePlugin(plugin) except: errstring = str(sys.exc_info()[0]).lstrip("exceptions.") + ": " + str(sys.exc_info()[1]) drScrolledMessageDialog.ShowMessage(self.parent, ("Error loading plugin: " + plugin + "\n\n" + errstring), "Load Error") self.loadedindexes.append(self.indexes[i]) self.reloadMenu() except: drScrolledMessageDialog.ShowMessage(self.parent, ("Error loading plugins from: " + index), "Load Error") def setupMenu(self): files = os.listdir(self.parent.pluginsdirectory) self.indexes = [] #Added By Franz pluginsfile = self.parent.preferencesdirectory + "/default.idx" if not os.path.exists(pluginsfile): return pluginstoload = [] try: f = file(pluginsfile, 'rU') pluginstoload = [x.strip() for x in f] f.close() except: #Return added by Dan return #/Franz x = 0 for f in files: if f.find(".idx") > -1 and f != "default.idx": index = os.path.join (self.parent.pluginsdirectory, f) if not (index in self.loadedindexes): if not os.path.splitext(os.path.basename(index))[0] in pluginstoload: self.indexes.append(index) self.Append(self.ID_LOAD_PLUGIN_BASE + x, str(os.path.basename(f))) self.parent.Bind(wx.EVT_MENU, self.OnLoadPluginsFromIndex, id=self.ID_LOAD_PLUGIN_BASE+x) x = x + 1 def reloadMenu(self): mnuitems = self.GetMenuItems() num = len(mnuitems) x = 0 while x < num: self.Remove(mnuitems[x].GetId()) x = x + 1 self.setupMenu() class drPluginFunctionMenu(wx.Menu): def __init__(self, parent, function_id_base, functionstring, function): wx.Menu.__init__(self) self.ID_BASE = function_id_base self.functionstring = functionstring self.function = function self.parent = parent self.pluginsArray = [] self.pluginsMenuArray = [] def GetInsertPosition(self, plugin): x = 0 l = len(self.pluginsMenuArray) while x < l: if plugin > self.pluginsMenuArray[x]: x += 1 else: return x return x def AddItem(self, plugin): try: i = self.parent.LoadedPlugins.index(plugin) exec(compile('self.parent.PluginModules[i]' + self.functionstring, plugin, 'exec')) x = len(self.pluginsArray) i = self.GetInsertPosition(plugin) self.pluginsMenuArray.insert(i, plugin) self.pluginsArray.append(plugin) self.Insert(i, self.ID_BASE + x, plugin) self.parent.Bind(wx.EVT_MENU, self.function, id=self.ID_BASE + x) except: pass class drPluginAboutMenu(drPluginFunctionMenu): def __init__(self, parent): self.ID_LOAD_PLUGIN_ABOUT_BASE = 4305 drPluginFunctionMenu.__init__(self, parent, self.ID_LOAD_PLUGIN_ABOUT_BASE, '.OnAbout', self.OnLoadPluginsAbout) def OnLoadPluginsAbout(self, event): idnum = event.GetId() i = idnum - self.ID_LOAD_PLUGIN_ABOUT_BASE plugin = self.pluginsArray[i] try: i = self.parent.LoadedPlugins.index(plugin) self.parent.PluginModules[i].OnAbout(self.parent) except: drScrolledMessageDialog.ShowMessage(self.parent, ("Error With Help."), "Plugin Help Error") class drPluginHelpMenu(drPluginFunctionMenu): def __init__(self, parent): self.ID_LOAD_PLUGIN_HELP_BASE = 4705 drPluginFunctionMenu.__init__(self, parent, self.ID_LOAD_PLUGIN_HELP_BASE, '.OnHelp', self.OnLoadPluginsHelp) def OnLoadPluginsHelp(self, event): idnum = event.GetId() i = idnum - self.ID_LOAD_PLUGIN_HELP_BASE plugin = self.pluginsArray[i] try: i = self.parent.LoadedPlugins.index(plugin) self.parent.PluginModules[i].OnHelp(self.parent) except: drScrolledMessageDialog.ShowMessage(self.parent, ("Error With Help."), "Plugin Help Error") class drPluginPreferencesMenu(drPluginFunctionMenu): def __init__(self, parent): self.ID_LOAD_PLUGIN_PREFS_BASE = 4905 drPluginFunctionMenu.__init__(self, parent, self.ID_LOAD_PLUGIN_PREFS_BASE, '.OnPreferences', self.OnLoadPluginsPreferences) def OnLoadPluginsPreferences(self, event): idnum = event.GetId() i = idnum - self.ID_LOAD_PLUGIN_PREFS_BASE plugin = self.pluginsArray[i] try: i = self.parent.LoadedPlugins.index(plugin) self.parent.PluginModules[i].OnPreferences(self.parent) except: drScrolledMessageDialog.ShowMessage(self.parent, ("Error With Help."), "Plugin Help Error")