# Programmer: Daniel Pozmanter
# E-mail: drpython@bluebottle.com
# Note: You must reply to the verification e-mail to get through.
#
# Copyright 2003-2005 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
#drScript Menu
import os.path
import wx
import drFileDialog
import drScrolledMessageDialog
from drPrefsFile import ExtractPreferenceFromText
import drShortcutsFile
import drShortcuts
from drMenu import drMenu
def updatescripts(scriptfile, paths, titles):
l = len(paths)
x = 0
f = open(scriptfile, 'w')
while (x < l):
f.write("<path>" + paths[x] + "</path><title>" + titles[x] + "</title>\n")
x = x + 1
f.close()
class drNewShellDialog(wx.Dialog):
def __init__(self, parent, title):
wx.Dialog.__init__(self, parent, -1, title, wx.Point(50, 50), wx.Size(375, 250), wx.DEFAULT_DIALOG_STYLE | wx.MAXIMIZE_BOX | wx.THICK_FRAME | wx.RESIZE_BORDER)
self.ID_SAVE = 1005
self.theSizer = wx.FlexGridSizer(6, 3, 5, 10)
self.btnSave = wx.Button(self, self.ID_SAVE, "&Save")
self.parent = parent
self.txtTitle = wx.TextCtrl(self, -1, "", wx.Point(15, 325), wx.Size(250, -1))
self.txtCommand = wx.TextCtrl(self, -1, "", wx.Point(15, 325), wx.Size(250, -1))
self.txtArguments = wx.TextCtrl(self, -1, "", wx.Point(15, 325), wx.Size(250, -1))
self.txtDirectory = wx.TextCtrl(self, -1, "", wx.Point(15, 325), wx.Size(250, -1))
self.chkRunInPrompt = wx.CheckBox(self, -1, "")
self.chkRunInPrompt.SetValue(True)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, "Title:"), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(self.txtTitle, 0, wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, "Command:"), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(self.txtCommand, 0, wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, "Arguments:"), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(self.txtArguments, 0, wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, "Directory:"), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(self.txtDirectory, 0, wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, "Run In Prompt:"), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(self.chkRunInPrompt, 0, wx.SHAPED)
self.btnClose = wx.Button(self, 101, "&Close")
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(wx.StaticText(self, -1, " "), 0, wx.ALIGN_CENTER | wx.SHAPED)
self.theSizer.Add(self.btnClose, 0, wx.SHAPED | wx.ALIGN_CENTER)
self.theSizer.Add(self.btnSave, 0, wx.SHAPED | wx.ALIGN_RIGHT)
self.txtTitle.SetFocus()
self.btnSave.SetDefault()
self.SetAutoLayout(True)
self.SetSizer(self.theSizer)
self.exitstatus = 0
self.Bind(wx.EVT_BUTTON, self.OnbtnSave, id=self.ID_SAVE)
self.Bind(wx.EVT_BUTTON, self.OnbtnClose, id=101)
def GetExitStatus(self):
return self.exitstatus
def GetShellTuple(self):
return self.txtTitle.GetValue(), self.txtCommand.GetValue(), self.txtArguments.GetValue(), self.txtDirectory.GetValue(), int(self.chkRunInPrompt.GetValue())
def OnbtnClose(self, event):
self.exitstatus = 0
self.Close(1)
def OnbtnSave(self, event):
self.exitstatus = 1
self.Close(1)
def SetShellTuple(self, title, command, args, dir, inprompt):
self.txtTitle.SetValue(title)
self.txtCommand.SetValue(command)
self.txtArguments.SetValue(args)
self.txtDirectory.SetValue(dir)
self.chkRunInPrompt.SetValue(inprompt)
class drScriptMenu(drMenu):
def __init__(self, parent):
drMenu.__init__(self, parent)
self.dynamicscriptext = ""
self.ID_ADD_SCRIPT = 3100
self.ID_NEW_SCRIPT = 3101
self.ID_EXISTING_SCRIPT = 3102
self.ID_SHELL_COMMAND = 3103
self.ID_EDIT_SCRIPT = 3003
self.ID_REMOVE_SCRIPT = 3005
self.ID_DYNAMIC_SCRIPT = 3006
self.ID_EXAMPLE_SCRIPTS = 3010
self.ID_SCRIPT_BASE = parent.ID_SCRIPT_BASE
self.ID_SCRIPT_MENU = 3500
self.parent = parent
self.userpreferencesdirectory = parent.userpreferencesdirectory
self.programdirectory = parent.programdirectory
self.scriptcount = 0
self.scripts = []
self.titles = []
self.setupMenu()
self.ExampleScriptCount = 0
self.loadscripts()
def getdrscriptmenulabel(self, label):
shortcuttext = ''
if len(self.parent.DrScriptShortcuts) == 0:
return label
if label in self.titles:
i = self.titles.index(label)
shortcuttext = drShortcuts.GetShortcutLabel(self.parent.DrScriptShortcuts[i])
if len(shortcuttext) > 1:
return label + '\t' + shortcuttext
return label
def loadscripts(self, SyncShortcuts = 1):
self.scriptcount = 0
if self.parent.prefs.drscriptloadexamples:
self.examplemenu = wx.Menu()
self.loadscriptsfromfile(self.programdirectory + "/examples/DrScript/drscript.dat", self.examplemenu, 1)
self.AppendMenu(self.ID_EXAMPLE_SCRIPTS, "Examples", self.examplemenu)
self.AppendSeparator()
self.ExampleScriptCount = self.scriptcount
self.loadscriptsfromfile(self.userpreferencesdirectory + "/drscript.dat", self)
#Either there is no shortcuts file, or it is out of sync.
if SyncShortcuts:
l = len(self.titles)
if not (l == len(self.parent.DrScriptShortcuts)):
self.parent.DrScriptShortcuts = []
self.parent.DrScriptShortcutNames = []
x = 0
while (x < l):
self.parent.DrScriptShortcuts.append(' #')
self.parent.DrScriptShortcutNames.append(self.titles[x])
x = x + 1
def loadscriptsfromfile(self, scriptfile, target, appendprogpath = 0):
if (os.path.exists(scriptfile)):
folders = [target]
folderindex = 0
menuTitles = []
menuTitleindex = -1
lastCount = 0
try:
#Read from the file
f = open(scriptfile, 'rb')
#Initialize
line = f.readline()
while (len(line) > 0):
c = line.count('\t')
line = line[c:].rstrip()
while (lastCount > c):
folders[(folderindex - 1)].AppendMenu(self.ID_SCRIPT_MENU, menuTitles.pop(), folders.pop())
folderindex = folderindex - 1
menuTitleindex = menuTitleindex - 1
lastCount = lastCount - 1
if (line[0] == '>'):
folders.append(wx.Menu())
menuTitles.append(line[1:])
folderindex = folderindex + 1
menuTitleindex = menuTitleindex + 1
c = c + 1
else:
line_path = ExtractPreferenceFromText(line, "path")
if appendprogpath:
line_path = self.programdirectory + "/" + line_path.replace('\\', '/')
line_title = ExtractPreferenceFromText(line, "title")
if (os.path.exists(line_path)):
self.titles.append(line_title)
folders[folderindex].Append((self.ID_SCRIPT_BASE + self.scriptcount), self.getdrscriptmenulabel(line_title))
self.parent.Bind(wx.EVT_MENU, self.OnScript, id=(self.ID_SCRIPT_BASE + self.scriptcount))
self.scripts.append(line_path)
self.scriptcount = self.scriptcount + 1
lastCount = c
line = f.readline()
f.close()
#Add any menus not yet added:
c = 0
while (lastCount > c):
folders[(folderindex - 1)].AppendMenu(self.ID_SCRIPT_MENU, menuTitles.pop(), folders.pop())
folderindex = folderindex - 1
menuTitleindex = menuTitleindex - 1
lastCount = lastCount - 1
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Your drscript file is a tad messed up.\n"), "Error")
def OnAddExistingScript(self, event):
scriptfile = self.userpreferencesdirectory + "/drscript.dat"
dlg = drFileDialog.FileDialog(self.parent, "Select Script File", self.parent.prefs.wildcard)
if (len(self.parent.prefs.drscriptdefaultdirectory) > 0):
try:
dlg.SetDirectory(self.parent.prefs.drscriptdefaultdirectory)
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Error Setting Default Directory To: " + self.parent.prefs.drscriptdefaultdirectory), "DrPython Error")
if (dlg.ShowModal() == wx.ID_OK):
filen = dlg.GetPath().replace("\\", "/")
d = wx.TextEntryDialog(self.parent, 'Enter Script Title:', 'Add DrScript: Title', '')
title = ""
if (d.ShowModal() == wx.ID_OK):
title = d.GetValue()
d.Destroy()
if (len(title) < 1):
drScrolledMessageDialog.ShowMessage(self.parent, ("Bad script title.\nDrPython will politely ignore your request to add this script."), "Error")
return
self.Append((self.ID_SCRIPT_BASE + self.scriptcount), title, title)
self.parent.Bind(wx.EVT_MENU, self.OnScript, id=(self.ID_SCRIPT_BASE + self.scriptcount))
self.scripts.append(filen)
self.titles.append(title)
x = len(self.parent.DrScriptShortcuts)
self.parent.DrScriptShortcuts.append('')
self.parent.DrScriptShortcutNames.append(title)
self.scriptcount = self.scriptcount + 1
try:
f = open(scriptfile, 'a')
f.write("<path>" + filen + "</path><title>" + title + "</title>\n")
f.close()
#Update the shortcuts Too.
shortcutsfile = self.userpreferencesdirectory + "/drscript.shortcuts.dat"
drShortcutsFile.WriteShortcuts(shortcutsfile, self.parent.DrScriptShortcuts, self.parent.DrScriptShortcutNames, "", False)
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Problem saving script.\n"), "Error")
dlg.Destroy()
def OnAddNewScript(self, event):
scriptfile = self.userpreferencesdirectory + "/drscript.dat"
dlg = drFileDialog.FileDialog(self.parent, "Save New Script File As", self.parent.prefs.wildcard, IsASaveDialog=True)
if (len(self.parent.prefs.drscriptdefaultdirectory) > 0):
try:
dlg.SetDirectory(self.parent.prefs.drscriptdefaultdirectory)
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Error Setting Default Directory To: " + self.parent.prefs.drscriptdefaultdirectory), "DrPython Error")
if (dlg.ShowModal() == wx.ID_OK):
filen = dlg.GetPath().replace("\\", "/")
d = wx.TextEntryDialog(self.parent, "Enter Script Title:", "Add DrScript: Title", "")
title = ""
if (d.ShowModal() == wx.ID_OK):
title = d.GetValue()
d.Destroy()
if (len(title) < 1):
drScrolledMessageDialog.ShowMessage(self.parent, ("Bad script title.\nDrPython will politely ignore your request to add this script."), "Error")
return
cfile = file(filen, 'wb')
cfile.write("#drscript\n\n")
cfile.close()
self.Append((self.ID_SCRIPT_BASE + self.scriptcount), title, title)
self.parent.Bind(wx.EVT_MENU, self.OnScript, id=(self.ID_SCRIPT_BASE + self.scriptcount))
self.scripts.append(filen)
self.titles.append(title)
x = len(self.parent.DrScriptShortcuts)
self.parent.DrScriptShortcuts.append('')
self.parent.DrScriptShortcutNames.append(title)
self.scriptcount = self.scriptcount + 1
try:
f = open(scriptfile, 'a')
f.write("<path>" + filen + "</path><title>" + title + "</title>\n")
f.close()
shortcutsfile = self.userpreferencesdirectory + "/drscript.shortcuts.dat"
drShortcutsFile.WriteShortcuts(shortcutsfile, self.parent.DrScriptShortcuts, self.parent.DrScriptShortcutNames, "", False)
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Problem saving script.\n"), "Error")
old = self.parent.txtDocument.filename
filename = filen
if (len(old) > 0) or (self.parent.txtDocument.GetModify()):
self.parent.OpenFile(filename, True)
else:
self.parent.OpenFile(filename, False)
dlg.Destroy()
def OnAddShellCommand(self, event):
scriptfile = self.userpreferencesdirectory + "/drscript.dat"
dlg = drFileDialog.FileDialog(self.parent, "Save New Shell Command As", self.parent.prefs.wildcard, IsASaveDialog=True)
if (len(self.parent.prefs.drscriptdefaultdirectory) > 0):
try:
dlg.SetDirectory(self.parent.prefs.drscriptdefaultdirectory)
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Error Setting Default Directory To: " + self.parent.prefs.drscriptdefaultdirectory), "DrPython Error")
if (dlg.ShowModal() == wx.ID_OK):
filen = dlg.GetPath().replace("\\", "/")
d = drNewShellDialog(self.parent, "New Shell Command")
if self.parent.prefs.platform_is_windows:
d.SetSize(wx.Size(425, 350))
d.ShowModal()
if d.GetExitStatus():
title, cmd, args, dir, inprompt = d.GetShellTuple()
d.Destroy()
else:
d.Destroy()
return
if (len(title) < 1):
drScrolledMessageDialog.ShowMessage(self.parent, ("Bad shell title.\nDrPython will politely ignore your request to add this shell command."), "Error")
return
if args.find("<Current File>") > -1:
args = args.replace("<Current File>", self.parent.txtDocument.filename)
if args.find("<Current Directory>") > -1:
args = args.replace("<Current Directory>", os.path.split(self.parent.txtDocument.filename)[0])
if dir.find("<Current Directory>") > -1:
dir = dir.replace("<Current Directory>", os.path.split(self.parent.txtDocument.filename)[0])
cfile = file(filen, 'wb')
scripttext = "#drscript shell command\n\n"
if len(dir) > 0:
scripttext = scripttext + 'import os\n\nos.chdir("' + dir + '")\n\n'
if inprompt:
scripttext = scripttext + 'DrFrame.Execute("' + cmd + ' ' + args + '", "Running Shell Command")\n\n'
else:
scripttext = scripttext + 'wx.Shell("' + cmd + ' ' + args + '")\n\n'
cfile.write(scripttext)
cfile.close()
self.Append((self.ID_SCRIPT_BASE + self.scriptcount), title, title)
self.parent.Bind(wx.EVT_MENU, self.OnScript, id=(self.ID_SCRIPT_BASE + self.scriptcount))
self.scripts.append(filen)
self.titles.append(title)
x = len(self.parent.DrScriptShortcuts)
self.parent.DrScriptShortcuts.append('')
self.parent.DrScriptShortcutNames.append(title)
self.scriptcount = self.scriptcount + 1
try:
f = open(scriptfile, 'a')
f.write("<path>" + filen + "</path><title>" + title + "</title>\n")
f.close()
shortcutsfile = self.userpreferencesdirectory + "/drscript.shortcuts.dat"
drShortcutsFile.WriteShortcuts(shortcutsfile, self.parent.DrScriptShortcuts, self.parent.DrScriptShortcutNames, "", False)
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Problem saving script.\n"), "Error")
dlg.Destroy()
def OnDynamicScript(self, event):
from drDynamicDrScriptDialog import drDynamicDrScriptDialog
d = drDynamicDrScriptDialog(self.parent, self.dynamicscriptext)
d.ShowModal()
self.dynamicscriptext = d.GetText()
d.Destroy()
def OnEditScript(self, event):
if (len(self.scripts) > 0):
def tuplify(x, y):
return (x, y)
def detuplify(x):
return x[0]
array = map(tuplify, self.titles, self.scripts)
array.sort()
titles = map(detuplify, array)
d = wx.SingleChoiceDialog(self.parent, "Select the Script to Edit:", "Edit Script", titles, wx.OK|wx.CANCEL)
d.SetSize(wx.Size(250, 250))
answer = d.ShowModal()
d.Destroy()
if (answer == wx.ID_OK):
i = d.GetSelection()
old = self.parent.txtDocument.filename
filename = array[i][1].replace("\\", "/")
alreadyopen = map(lambda x: x.filename.replace("\\", "/"), self.parent.txtDocumentArray)
if filename in alreadyopen:
self.parent.setDocumentTo(alreadyopen.index(filename))
return
if (len(old) > 0) or (self.parent.txtDocument.GetModify()):
self.parent.OpenFile(filename, True, False)
else:
self.parent.OpenFile(filename, False, False)
else:
drScrolledMessageDialog.ShowMessage(self.parent, ("There aren't any scripts to edit.\nTry \"New Script\" or \"Add Script\" Instead."), "Error")
def OnScript(self, event):
scriptindex = event.GetId() - self.ID_SCRIPT_BASE
f = open(self.scripts[scriptindex], 'r')
scripttext = f.read()
f.close()
if (scripttext.find("DrFilename") > -1):
scripttext = scripttext.replace("DrFilename", "self.parent.txtDocument.filename")
if (scripttext.find("DrScript") > -1):
scripttext = scripttext.replace("DrScript", "self.parent.DrScript")
if (scripttext.find("DrDocument") > -1):
scripttext = scripttext.replace("DrDocument", "self.parent.txtDocument")
if (scripttext.find("DrPrompt") > -1):
scripttext = scripttext.replace("DrPrompt", "self.parent.txtPrompt")
if (scripttext.find("DrFrame") > -1):
scripttext = scripttext.replace("DrFrame", "self.parent")
try:
code = compile((scripttext + '\n'), self.scripts[scriptindex], 'exec')
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Error compiling script."), "Error", wx.DefaultPosition, wx.Size(550,300))
return
try:
exec(code)
except:
drScrolledMessageDialog.ShowMessage(self.parent, ("Error running script."), "Error", wx.DefaultPosition, wx.Size(550,300))
return
def setupMenu(self):
addscriptmenu = wx.Menu()
addscriptmenu.Append(self.ID_NEW_SCRIPT, "&New Script...")
addscriptmenu.Append(self.ID_EXISTING_SCRIPT, "&Existing Script...")
addscriptmenu.Append(self.ID_SHELL_COMMAND, "&Shell Command...")
self.AppendMenu(self.ID_ADD_SCRIPT, "&Add Script", addscriptmenu)
self.Append(self.ID_EDIT_SCRIPT, "&Edit Script...")
self.Append(self.ID_DYNAMIC_SCRIPT, "&Dynamic DrScript...")
self.AppendSeparator()
self.parent.Bind(wx.EVT_MENU, self.OnAddExistingScript, id=self.ID_EXISTING_SCRIPT)
self.parent.Bind(wx.EVT_MENU, self.OnAddNewScript, id=self.ID_NEW_SCRIPT)
self.parent.Bind(wx.EVT_MENU, self.OnAddShellCommand, id=self.ID_SHELL_COMMAND)
self.parent.Bind(wx.EVT_MENU, self.OnEditScript, id=self.ID_EDIT_SCRIPT)
self.parent.Bind(wx.EVT_MENU, self.OnDynamicScript, id=self.ID_DYNAMIC_SCRIPT)
def reloadscripts(self, oldlength = -1, SyncShortcuts = 1):
if oldlength < 0:
oldlength = len(self.scripts)
mnuitems = self.GetMenuItems()
num = len(mnuitems)
x = 0
while (x < num):
self.Remove(mnuitems[x].GetId())
#mnuitems[x].Destroy()
x = x + 1
self.scripts = []
self.titles = []
self.setupMenu()
self.loadscripts(SyncShortcuts)
syntax highlighted by Code2HTML, v. 0.9.1