mirror of
https://github.com/python/cpython.git
synced 2025-10-30 05:01:30 +00:00
f0d12da482
are going to replace StandardGetFile() and friends. Main differences are that these allow you to ask for specific datatypes to be returned (FSSpec, FSRef, string, unicode or subtypes thereof) and that they provide access to underlying features of Navigation Services through keyword arguments.
750 lines
22 KiB
Python
750 lines
22 KiB
Python
"""Easy to use dialogs.
|
|
|
|
Message(msg) -- display a message and an OK button.
|
|
AskString(prompt, default) -- ask for a string, display OK and Cancel buttons.
|
|
AskPassword(prompt, default) -- like AskString(), but shows text as bullets.
|
|
AskYesNoCancel(question, default) -- display a question and Yes, No and Cancel buttons.
|
|
bar = Progress(label, maxvalue) -- Display a progress bar
|
|
bar.set(value) -- Set value
|
|
bar.inc( *amount ) -- increment value by amount (default=1)
|
|
bar.label( *newlabel ) -- get or set text label.
|
|
|
|
More documentation in each function.
|
|
This module uses DLOG resources 260 and on.
|
|
Based upon STDWIN dialogs with the same names and functions.
|
|
"""
|
|
|
|
from Carbon.Dlg import GetNewDialog, SetDialogItemText, GetDialogItemText, ModalDialog
|
|
from Carbon import Qd
|
|
from Carbon import QuickDraw
|
|
from Carbon import Dialogs
|
|
from Carbon import Windows
|
|
from Carbon import Dlg,Win,Evt,Events # sdm7g
|
|
from Carbon import Ctl
|
|
from Carbon import Controls
|
|
from Carbon import Menu
|
|
import Nav
|
|
import MacOS
|
|
import string
|
|
from Carbon.ControlAccessor import * # Also import Controls constants
|
|
import Carbon.File
|
|
import macfs
|
|
import macresource
|
|
|
|
_initialized = 0
|
|
|
|
def _initialize():
|
|
global _initialized
|
|
if _initialized: return
|
|
macresource.need("DLOG", 260, "dialogs.rsrc", __name__)
|
|
|
|
|
|
def cr2lf(text):
|
|
if '\r' in text:
|
|
text = string.join(string.split(text, '\r'), '\n')
|
|
return text
|
|
|
|
def lf2cr(text):
|
|
if '\n' in text:
|
|
text = string.join(string.split(text, '\n'), '\r')
|
|
if len(text) > 253:
|
|
text = text[:253] + '\311'
|
|
return text
|
|
|
|
def Message(msg, id=260, ok=None):
|
|
"""Display a MESSAGE string.
|
|
|
|
Return when the user clicks the OK button or presses Return.
|
|
|
|
The MESSAGE string can be at most 255 characters long.
|
|
"""
|
|
_initialize()
|
|
d = GetNewDialog(id, -1)
|
|
if not d:
|
|
print "EasyDialogs: Can't get DLOG resource with id =", id, " (missing resource file?)"
|
|
return
|
|
h = d.GetDialogItemAsControl(2)
|
|
SetDialogItemText(h, lf2cr(msg))
|
|
if ok != None:
|
|
h = d.GetDialogItemAsControl(1)
|
|
h.SetControlTitle(ok)
|
|
d.SetDialogDefaultItem(1)
|
|
d.AutoSizeDialog()
|
|
d.GetDialogWindow().ShowWindow()
|
|
while 1:
|
|
n = ModalDialog(None)
|
|
if n == 1:
|
|
return
|
|
|
|
|
|
def AskString(prompt, default = "", id=261, ok=None, cancel=None):
|
|
"""Display a PROMPT string and a text entry field with a DEFAULT string.
|
|
|
|
Return the contents of the text entry field when the user clicks the
|
|
OK button or presses Return.
|
|
Return None when the user clicks the Cancel button.
|
|
|
|
If omitted, DEFAULT is empty.
|
|
|
|
The PROMPT and DEFAULT strings, as well as the return value,
|
|
can be at most 255 characters long.
|
|
"""
|
|
|
|
_initialize()
|
|
d = GetNewDialog(id, -1)
|
|
if not d:
|
|
print "EasyDialogs: Can't get DLOG resource with id =", id, " (missing resource file?)"
|
|
return
|
|
h = d.GetDialogItemAsControl(3)
|
|
SetDialogItemText(h, lf2cr(prompt))
|
|
h = d.GetDialogItemAsControl(4)
|
|
SetDialogItemText(h, lf2cr(default))
|
|
d.SelectDialogItemText(4, 0, 999)
|
|
# d.SetDialogItem(4, 0, 255)
|
|
if ok != None:
|
|
h = d.GetDialogItemAsControl(1)
|
|
h.SetControlTitle(ok)
|
|
if cancel != None:
|
|
h = d.GetDialogItemAsControl(2)
|
|
h.SetControlTitle(cancel)
|
|
d.SetDialogDefaultItem(1)
|
|
d.SetDialogCancelItem(2)
|
|
d.AutoSizeDialog()
|
|
d.GetDialogWindow().ShowWindow()
|
|
while 1:
|
|
n = ModalDialog(None)
|
|
if n == 1:
|
|
h = d.GetDialogItemAsControl(4)
|
|
return cr2lf(GetDialogItemText(h))
|
|
if n == 2: return None
|
|
|
|
def AskPassword(prompt, default='', id=264, ok=None, cancel=None):
|
|
"""Display a PROMPT string and a text entry field with a DEFAULT string.
|
|
The string is displayed as bullets only.
|
|
|
|
Return the contents of the text entry field when the user clicks the
|
|
OK button or presses Return.
|
|
Return None when the user clicks the Cancel button.
|
|
|
|
If omitted, DEFAULT is empty.
|
|
|
|
The PROMPT and DEFAULT strings, as well as the return value,
|
|
can be at most 255 characters long.
|
|
"""
|
|
_initialize()
|
|
d = GetNewDialog(id, -1)
|
|
if not d:
|
|
print "EasyDialogs: Can't get DLOG resource with id =", id, " (missing resource file?)"
|
|
return
|
|
h = d.GetDialogItemAsControl(3)
|
|
SetDialogItemText(h, lf2cr(prompt))
|
|
pwd = d.GetDialogItemAsControl(4)
|
|
bullets = '\245'*len(default)
|
|
## SetControlData(pwd, kControlEditTextPart, kControlEditTextTextTag, bullets)
|
|
SetControlData(pwd, kControlEditTextPart, kControlEditTextPasswordTag, default)
|
|
d.SelectDialogItemText(4, 0, 999)
|
|
Ctl.SetKeyboardFocus(d.GetDialogWindow(), pwd, kControlEditTextPart)
|
|
if ok != None:
|
|
h = d.GetDialogItemAsControl(1)
|
|
h.SetControlTitle(ok)
|
|
if cancel != None:
|
|
h = d.GetDialogItemAsControl(2)
|
|
h.SetControlTitle(cancel)
|
|
d.SetDialogDefaultItem(Dialogs.ok)
|
|
d.SetDialogCancelItem(Dialogs.cancel)
|
|
d.AutoSizeDialog()
|
|
d.GetDialogWindow().ShowWindow()
|
|
while 1:
|
|
n = ModalDialog(None)
|
|
if n == 1:
|
|
h = d.GetDialogItemAsControl(4)
|
|
return cr2lf(GetControlData(pwd, kControlEditTextPart, kControlEditTextPasswordTag))
|
|
if n == 2: return None
|
|
|
|
def AskYesNoCancel(question, default = 0, yes=None, no=None, cancel=None, id=262):
|
|
"""Display a QUESTION string which can be answered with Yes or No.
|
|
|
|
Return 1 when the user clicks the Yes button.
|
|
Return 0 when the user clicks the No button.
|
|
Return -1 when the user clicks the Cancel button.
|
|
|
|
When the user presses Return, the DEFAULT value is returned.
|
|
If omitted, this is 0 (No).
|
|
|
|
The QUESTION string can be at most 255 characters.
|
|
"""
|
|
|
|
_initialize()
|
|
d = GetNewDialog(id, -1)
|
|
if not d:
|
|
print "EasyDialogs: Can't get DLOG resource with id =", id, " (missing resource file?)"
|
|
return
|
|
# Button assignments:
|
|
# 1 = default (invisible)
|
|
# 2 = Yes
|
|
# 3 = No
|
|
# 4 = Cancel
|
|
# The question string is item 5
|
|
h = d.GetDialogItemAsControl(5)
|
|
SetDialogItemText(h, lf2cr(question))
|
|
if yes != None:
|
|
if yes == '':
|
|
d.HideDialogItem(2)
|
|
else:
|
|
h = d.GetDialogItemAsControl(2)
|
|
h.SetControlTitle(yes)
|
|
if no != None:
|
|
if no == '':
|
|
d.HideDialogItem(3)
|
|
else:
|
|
h = d.GetDialogItemAsControl(3)
|
|
h.SetControlTitle(no)
|
|
if cancel != None:
|
|
if cancel == '':
|
|
d.HideDialogItem(4)
|
|
else:
|
|
h = d.GetDialogItemAsControl(4)
|
|
h.SetControlTitle(cancel)
|
|
d.SetDialogCancelItem(4)
|
|
if default == 1:
|
|
d.SetDialogDefaultItem(2)
|
|
elif default == 0:
|
|
d.SetDialogDefaultItem(3)
|
|
elif default == -1:
|
|
d.SetDialogDefaultItem(4)
|
|
d.AutoSizeDialog()
|
|
d.GetDialogWindow().ShowWindow()
|
|
while 1:
|
|
n = ModalDialog(None)
|
|
if n == 1: return default
|
|
if n == 2: return 1
|
|
if n == 3: return 0
|
|
if n == 4: return -1
|
|
|
|
|
|
|
|
|
|
screenbounds = Qd.GetQDGlobalsScreenBits().bounds
|
|
screenbounds = screenbounds[0]+4, screenbounds[1]+4, \
|
|
screenbounds[2]-4, screenbounds[3]-4
|
|
|
|
kControlProgressBarIndeterminateTag = 'inde' # from Controls.py
|
|
|
|
|
|
class ProgressBar:
|
|
def __init__(self, title="Working...", maxval=0, label="", id=263):
|
|
self.w = None
|
|
self.d = None
|
|
_initialize()
|
|
self.d = GetNewDialog(id, -1)
|
|
self.w = self.d.GetDialogWindow()
|
|
self.label(label)
|
|
self.title(title)
|
|
self.set(0, maxval)
|
|
self.d.AutoSizeDialog()
|
|
self.w.ShowWindow()
|
|
self.d.DrawDialog()
|
|
|
|
def __del__( self ):
|
|
if self.w:
|
|
self.w.BringToFront()
|
|
self.w.HideWindow()
|
|
del self.w
|
|
del self.d
|
|
|
|
def title(self, newstr=""):
|
|
"""title(text) - Set title of progress window"""
|
|
self.w.BringToFront()
|
|
self.w.SetWTitle(newstr)
|
|
|
|
def label( self, *newstr ):
|
|
"""label(text) - Set text in progress box"""
|
|
self.w.BringToFront()
|
|
if newstr:
|
|
self._label = lf2cr(newstr[0])
|
|
text_h = self.d.GetDialogItemAsControl(2)
|
|
SetDialogItemText(text_h, self._label)
|
|
|
|
def _update(self, value):
|
|
maxval = self.maxval
|
|
if maxval == 0: # an indeterminate bar
|
|
Ctl.IdleControls(self.w) # spin the barber pole
|
|
else: # a determinate bar
|
|
if maxval > 32767:
|
|
value = int(value/(maxval/32767.0))
|
|
maxval = 32767
|
|
progbar = self.d.GetDialogItemAsControl(3)
|
|
progbar.SetControlMaximum(maxval)
|
|
progbar.SetControlValue(value) # set the bar length
|
|
|
|
# Test for cancel button
|
|
ready, ev = Evt.WaitNextEvent( Events.mDownMask, 1 )
|
|
if ready :
|
|
what,msg,when,where,mod = ev
|
|
part = Win.FindWindow(where)[0]
|
|
if Dlg.IsDialogEvent(ev):
|
|
ds = Dlg.DialogSelect(ev)
|
|
if ds[0] and ds[1] == self.d and ds[-1] == 1:
|
|
self.w.HideWindow()
|
|
self.w = None
|
|
self.d = None
|
|
raise KeyboardInterrupt, ev
|
|
else:
|
|
if part == 4: # inDrag
|
|
self.w.DragWindow(where, screenbounds)
|
|
else:
|
|
MacOS.HandleEvent(ev)
|
|
|
|
|
|
def set(self, value, max=None):
|
|
"""set(value) - Set progress bar position"""
|
|
if max != None:
|
|
self.maxval = max
|
|
bar = self.d.GetDialogItemAsControl(3)
|
|
if max <= 0: # indeterminate bar
|
|
bar.SetControlData(0,kControlProgressBarIndeterminateTag,'\x01')
|
|
else: # determinate bar
|
|
bar.SetControlData(0,kControlProgressBarIndeterminateTag,'\x00')
|
|
if value < 0:
|
|
value = 0
|
|
elif value > self.maxval:
|
|
value = self.maxval
|
|
self.curval = value
|
|
self._update(value)
|
|
|
|
def inc(self, n=1):
|
|
"""inc(amt) - Increment progress bar position"""
|
|
self.set(self.curval + n)
|
|
|
|
ARGV_ID=265
|
|
ARGV_ITEM_OK=1
|
|
ARGV_ITEM_CANCEL=2
|
|
ARGV_OPTION_GROUP=3
|
|
ARGV_OPTION_EXPLAIN=4
|
|
ARGV_OPTION_VALUE=5
|
|
ARGV_OPTION_ADD=6
|
|
ARGV_COMMAND_GROUP=7
|
|
ARGV_COMMAND_EXPLAIN=8
|
|
ARGV_COMMAND_ADD=9
|
|
ARGV_ADD_OLDFILE=10
|
|
ARGV_ADD_NEWFILE=11
|
|
ARGV_ADD_FOLDER=12
|
|
ARGV_CMDLINE_GROUP=13
|
|
ARGV_CMDLINE_DATA=14
|
|
|
|
##def _myModalDialog(d):
|
|
## while 1:
|
|
## ready, ev = Evt.WaitNextEvent(0xffff, -1)
|
|
## print 'DBG: WNE', ready, ev
|
|
## if ready :
|
|
## what,msg,when,where,mod = ev
|
|
## part, window = Win.FindWindow(where)
|
|
## if Dlg.IsDialogEvent(ev):
|
|
## didit, dlgdone, itemdone = Dlg.DialogSelect(ev)
|
|
## print 'DBG: DialogSelect', didit, dlgdone, itemdone, d
|
|
## if didit and dlgdone == d:
|
|
## return itemdone
|
|
## elif window == d.GetDialogWindow():
|
|
## d.GetDialogWindow().SelectWindow()
|
|
## if part == 4: # inDrag
|
|
## d.DragWindow(where, screenbounds)
|
|
## else:
|
|
## MacOS.HandleEvent(ev)
|
|
## else:
|
|
## MacOS.HandleEvent(ev)
|
|
##
|
|
def _setmenu(control, items):
|
|
mhandle = control.GetControlData_Handle(Controls.kControlMenuPart,
|
|
Controls.kControlPopupButtonMenuHandleTag)
|
|
menu = Menu.as_Menu(mhandle)
|
|
for item in items:
|
|
if type(item) == type(()):
|
|
label = item[0]
|
|
else:
|
|
label = item
|
|
if label[-1] == '=' or label[-1] == ':':
|
|
label = label[:-1]
|
|
menu.AppendMenu(label)
|
|
## mhandle, mid = menu.getpopupinfo()
|
|
## control.SetControlData_Handle(Controls.kControlMenuPart,
|
|
## Controls.kControlPopupButtonMenuHandleTag, mhandle)
|
|
control.SetControlMinimum(1)
|
|
control.SetControlMaximum(len(items)+1)
|
|
|
|
def _selectoption(d, optionlist, idx):
|
|
if idx < 0 or idx >= len(optionlist):
|
|
MacOS.SysBeep()
|
|
return
|
|
option = optionlist[idx]
|
|
if type(option) == type(()):
|
|
if len(option) == 4:
|
|
help = option[2]
|
|
elif len(option) > 1:
|
|
help = option[-1]
|
|
else:
|
|
help = ''
|
|
else:
|
|
help = ''
|
|
h = d.GetDialogItemAsControl(ARGV_OPTION_EXPLAIN)
|
|
if help and len(help) > 250:
|
|
help = help[:250] + '...'
|
|
Dlg.SetDialogItemText(h, help)
|
|
hasvalue = 0
|
|
if type(option) == type(()):
|
|
label = option[0]
|
|
else:
|
|
label = option
|
|
if label[-1] == '=' or label[-1] == ':':
|
|
hasvalue = 1
|
|
h = d.GetDialogItemAsControl(ARGV_OPTION_VALUE)
|
|
Dlg.SetDialogItemText(h, '')
|
|
if hasvalue:
|
|
d.ShowDialogItem(ARGV_OPTION_VALUE)
|
|
d.SelectDialogItemText(ARGV_OPTION_VALUE, 0, 0)
|
|
else:
|
|
d.HideDialogItem(ARGV_OPTION_VALUE)
|
|
|
|
|
|
def GetArgv(optionlist=None, commandlist=None, addoldfile=1, addnewfile=1, addfolder=1, id=ARGV_ID):
|
|
_initialize()
|
|
d = GetNewDialog(id, -1)
|
|
if not d:
|
|
print "EasyDialogs: Can't get DLOG resource with id =", id, " (missing resource file?)"
|
|
return
|
|
# h = d.GetDialogItemAsControl(3)
|
|
# SetDialogItemText(h, lf2cr(prompt))
|
|
# h = d.GetDialogItemAsControl(4)
|
|
# SetDialogItemText(h, lf2cr(default))
|
|
# d.SelectDialogItemText(4, 0, 999)
|
|
# d.SetDialogItem(4, 0, 255)
|
|
if optionlist:
|
|
_setmenu(d.GetDialogItemAsControl(ARGV_OPTION_GROUP), optionlist)
|
|
_selectoption(d, optionlist, 0)
|
|
else:
|
|
d.GetDialogItemAsControl(ARGV_OPTION_GROUP).DeactivateControl()
|
|
if commandlist:
|
|
_setmenu(d.GetDialogItemAsControl(ARGV_COMMAND_GROUP), commandlist)
|
|
if type(commandlist[0]) == type(()) and len(commandlist[0]) > 1:
|
|
help = commandlist[0][-1]
|
|
h = d.GetDialogItemAsControl(ARGV_COMMAND_EXPLAIN)
|
|
Dlg.SetDialogItemText(h, help)
|
|
else:
|
|
d.GetDialogItemAsControl(ARGV_COMMAND_GROUP).DeactivateControl()
|
|
if not addoldfile:
|
|
d.GetDialogItemAsControl(ARGV_ADD_OLDFILE).DeactivateControl()
|
|
if not addnewfile:
|
|
d.GetDialogItemAsControl(ARGV_ADD_NEWFILE).DeactivateControl()
|
|
if not addfolder:
|
|
d.GetDialogItemAsControl(ARGV_ADD_FOLDER).DeactivateControl()
|
|
d.SetDialogDefaultItem(ARGV_ITEM_OK)
|
|
d.SetDialogCancelItem(ARGV_ITEM_CANCEL)
|
|
d.GetDialogWindow().ShowWindow()
|
|
d.DrawDialog()
|
|
if hasattr(MacOS, 'SchedParams'):
|
|
appsw = MacOS.SchedParams(1, 0)
|
|
try:
|
|
while 1:
|
|
stringstoadd = []
|
|
n = ModalDialog(None)
|
|
if n == ARGV_ITEM_OK:
|
|
break
|
|
elif n == ARGV_ITEM_CANCEL:
|
|
raise SystemExit
|
|
elif n == ARGV_OPTION_GROUP:
|
|
idx = d.GetDialogItemAsControl(ARGV_OPTION_GROUP).GetControlValue()-1
|
|
_selectoption(d, optionlist, idx)
|
|
elif n == ARGV_OPTION_VALUE:
|
|
pass
|
|
elif n == ARGV_OPTION_ADD:
|
|
idx = d.GetDialogItemAsControl(ARGV_OPTION_GROUP).GetControlValue()-1
|
|
if 0 <= idx < len(optionlist):
|
|
option = optionlist[idx]
|
|
if type(option) == type(()):
|
|
option = option[0]
|
|
if option[-1] == '=' or option[-1] == ':':
|
|
option = option[:-1]
|
|
h = d.GetDialogItemAsControl(ARGV_OPTION_VALUE)
|
|
value = Dlg.GetDialogItemText(h)
|
|
else:
|
|
value = ''
|
|
if len(option) == 1:
|
|
stringtoadd = '-' + option
|
|
else:
|
|
stringtoadd = '--' + option
|
|
stringstoadd = [stringtoadd]
|
|
if value:
|
|
stringstoadd.append(value)
|
|
else:
|
|
MacOS.SysBeep()
|
|
elif n == ARGV_COMMAND_GROUP:
|
|
idx = d.GetDialogItemAsControl(ARGV_COMMAND_GROUP).GetControlValue()-1
|
|
if 0 <= idx < len(commandlist) and type(commandlist[idx]) == type(()) and \
|
|
len(commandlist[idx]) > 1:
|
|
help = commandlist[idx][-1]
|
|
h = d.GetDialogItemAsControl(ARGV_COMMAND_EXPLAIN)
|
|
Dlg.SetDialogItemText(h, help)
|
|
elif n == ARGV_COMMAND_ADD:
|
|
idx = d.GetDialogItemAsControl(ARGV_COMMAND_GROUP).GetControlValue()-1
|
|
if 0 <= idx < len(commandlist):
|
|
command = commandlist[idx]
|
|
if type(command) == type(()):
|
|
command = command[0]
|
|
stringstoadd = [command]
|
|
else:
|
|
MacOS.SysBeep()
|
|
elif n == ARGV_ADD_OLDFILE:
|
|
fss, ok = macfs.StandardGetFile()
|
|
if ok:
|
|
stringstoadd = [fss.as_pathname()]
|
|
elif n == ARGV_ADD_NEWFILE:
|
|
fss, ok = macfs.StandardPutFile('')
|
|
if ok:
|
|
stringstoadd = [fss.as_pathname()]
|
|
elif n == ARGV_ADD_FOLDER:
|
|
fss, ok = macfs.GetDirectory()
|
|
if ok:
|
|
stringstoadd = [fss.as_pathname()]
|
|
elif n == ARGV_CMDLINE_DATA:
|
|
pass # Nothing to do
|
|
else:
|
|
raise RuntimeError, "Unknown dialog item %d"%n
|
|
|
|
for stringtoadd in stringstoadd:
|
|
if '"' in stringtoadd or "'" in stringtoadd or " " in stringtoadd:
|
|
stringtoadd = `stringtoadd`
|
|
h = d.GetDialogItemAsControl(ARGV_CMDLINE_DATA)
|
|
oldstr = GetDialogItemText(h)
|
|
if oldstr and oldstr[-1] != ' ':
|
|
oldstr = oldstr + ' '
|
|
oldstr = oldstr + stringtoadd
|
|
if oldstr[-1] != ' ':
|
|
oldstr = oldstr + ' '
|
|
SetDialogItemText(h, oldstr)
|
|
d.SelectDialogItemText(ARGV_CMDLINE_DATA, 0x7fff, 0x7fff)
|
|
h = d.GetDialogItemAsControl(ARGV_CMDLINE_DATA)
|
|
oldstr = GetDialogItemText(h)
|
|
tmplist = string.split(oldstr)
|
|
newlist = []
|
|
while tmplist:
|
|
item = tmplist[0]
|
|
del tmplist[0]
|
|
if item[0] == '"':
|
|
while item[-1] != '"':
|
|
if not tmplist:
|
|
raise RuntimeError, "Unterminated quoted argument"
|
|
item = item + ' ' + tmplist[0]
|
|
del tmplist[0]
|
|
item = item[1:-1]
|
|
if item[0] == "'":
|
|
while item[-1] != "'":
|
|
if not tmplist:
|
|
raise RuntimeError, "Unterminated quoted argument"
|
|
item = item + ' ' + tmplist[0]
|
|
del tmplist[0]
|
|
item = item[1:-1]
|
|
newlist.append(item)
|
|
return newlist
|
|
finally:
|
|
if hasattr(MacOS, 'SchedParams'):
|
|
apply(MacOS.SchedParams, appsw)
|
|
del d
|
|
|
|
def _mktypelist(typelist):
|
|
# Workaround for OSX typeless files:
|
|
if 'TEXT' in typelist and not '\0\0\0\0' in typelist:
|
|
typelist = typelist + ('\0\0\0\0',)
|
|
if not typelist:
|
|
return None
|
|
data = 'Pyth' + struct.pack("hh", 0, len(typelist))
|
|
for type in typelist:
|
|
data = data+type
|
|
return Carbon.Res.Handle(data)
|
|
|
|
_ALLOWED_KEYS = {
|
|
'version':1,
|
|
'defaultLocation':1,
|
|
'dialogOptionFlags':1,
|
|
'location':1,
|
|
'clientName':1,
|
|
'windowTitle':1,
|
|
'actionButtonLabel':1,
|
|
'cancelButtonLabel':1,
|
|
'savedFileName':1,
|
|
'message':1,
|
|
'preferenceKey':1,
|
|
'popupExtension':1,
|
|
'eventProc':1,
|
|
'previewProc':1,
|
|
'filterProc':1,
|
|
'typeList':1,
|
|
'fileType':1,
|
|
'fileCreator':1,
|
|
# Our extension:
|
|
'wanted':1,
|
|
'multiple':1,
|
|
}
|
|
|
|
def _process_Nav_args(argsargs, allowed, dftflags):
|
|
import aepack
|
|
import Carbon.AE
|
|
import Carbon.File
|
|
args = argsargs.copy()
|
|
for k in args.keys():
|
|
if not allowed.has_key(k):
|
|
raise TypeError, "Unknown keyword argument: %s" % repr(k)
|
|
# Set some defaults, and modify some arguments
|
|
if not args.has_key('dialogOptionFlags'):
|
|
args['dialogOptionFlags'] = dftflags
|
|
if args.has_key('defaultLocation') and \
|
|
not isinstance(args['defaultLocation'], Carbon.AE.AEDesc):
|
|
defaultLocation = args['defaultLocation']
|
|
if isinstance(defaultLocation, (Carbon.File.FSSpec, Carbon.File.FSRef)):
|
|
args['defaultLocation'] = aepack.pack(defaultLocation)
|
|
else:
|
|
defaultLocation = Carbon.File.FSRef(defaultLocation)
|
|
args['defaultLocation'] = aepack.pack(defaultLocation)
|
|
if args.has_key('typeList') and not isinstance(args['typeList'], Carbon.Res.ResourceType):
|
|
typeList = args['typeList'].copy()
|
|
# Workaround for OSX typeless files:
|
|
if 'TEXT' in typeList and not '\0\0\0\0' in typeList:
|
|
typeList = typeList + ('\0\0\0\0',)
|
|
data = 'Pyth' + struct.pack("hh", 0, len(typeList))
|
|
for type in typeList:
|
|
data = data+type
|
|
args['typeList'] = Carbon.Res.Handle(data)
|
|
tpwanted = str
|
|
if args.has_key('wanted'):
|
|
tpwanted = args['wanted']
|
|
del args['wanted']
|
|
return args, tpwanted
|
|
|
|
def AskFileForOpen(**args):
|
|
default_flags = 0x56 # Or 0xe4?
|
|
args, tpwanted = _process_Nav_args(args, _ALLOWED_KEYS, default_flags)
|
|
try:
|
|
rr = Nav.NavChooseFile(args)
|
|
good = 1
|
|
except Nav.error, arg:
|
|
if arg[0] != -128: # userCancelledErr
|
|
raise Nav.error, arg
|
|
return None
|
|
if not rr.validRecord or not rr.selection:
|
|
return None
|
|
if issubclass(tpwanted, Carbon.File.FSRef):
|
|
return tpwanted(rr.selection_fsr[0])
|
|
if issubclass(tpwanted, Carbon.File.FSSpec):
|
|
return tpwanted(rr.selection[0])
|
|
if issubclass(tpwanted, str):
|
|
return tpwanted(rr.selection_fsr[0].FSRefMakePath())
|
|
if issubclass(tpwanted, unicode):
|
|
return tpwanted(rr.selection_fsr[0].FSRefMakePath(), 'utf8')
|
|
raise TypeError, "Unknown value for argument 'wanted': %s" % repr(tpwanted)
|
|
|
|
def AskFileForSave(**args):
|
|
default_flags = 0x07
|
|
args, tpwanted = _process_Nav_args(args, _ALLOWED_KEYS, default_flags)
|
|
try:
|
|
rr = Nav.NavPutFile(args)
|
|
good = 1
|
|
except Nav.error, arg:
|
|
if arg[0] != -128: # userCancelledErr
|
|
raise Nav.error, arg
|
|
return None
|
|
if not rr.validRecord or not rr.selection:
|
|
return None
|
|
if issubclass(tpwanted, Carbon.File.FSRef):
|
|
raise TypeError, "Cannot pass wanted=FSRef to AskFileForSave"
|
|
if issubclass(tpwanted, Carbon.File.FSSpec):
|
|
return tpwanted(rr.selection[0])
|
|
if issubclass(tpwanted, (str, unicode)):
|
|
# This is gross, and probably incorrect too
|
|
vrefnum, dirid, name = rr.selection[0].as_tuple()
|
|
pardir_fss = Carbon.File.FSSpec((vrefnum, dirid, ''))
|
|
pardir_fsr = Carbon.File.FSRef(fss)
|
|
pardir_path = pardir_fsr.FSRefMakePath() # This is utf-8
|
|
name_utf8 = unicode(name, 'macroman').encode('utf8')
|
|
fullpath = os.path.join(pardir_path, name_utf8)
|
|
if issubclass(tpwanted, unicode):
|
|
return unicode(fullpath, 'utf8')
|
|
return tpwanted(fullpath)
|
|
raise TypeError, "Unknown value for argument 'wanted': %s" % repr(tpwanted)
|
|
|
|
def AskFolder(**args):
|
|
default_flags = 0x17
|
|
args, tpwanted = _process_Nav_args(args, _ALLOWED_KEYS, default_flags)
|
|
try:
|
|
rr = Nav.NavChooseFolder(args)
|
|
good = 1
|
|
except Nav.error, arg:
|
|
if arg[0] != -128: # userCancelledErr
|
|
raise Nav.error, arg
|
|
return None
|
|
if not rr.validRecord or not rr.selection:
|
|
return None
|
|
if issubclass(tpwanted, Carbon.File.FSRef):
|
|
return tpwanted(rr.selection_fsr[0])
|
|
if issubclass(tpwanted, Carbon.File.FSSpec):
|
|
return tpwanted(rr.selection[0])
|
|
if issubclass(tpwanted, str):
|
|
return tpwanted(rr.selection_fsr[0].FSRefMakePath())
|
|
if issubclass(tpwanted, unicode):
|
|
return tpwanted(rr.selection_fsr[0].FSRefMakePath(), 'utf8')
|
|
raise TypeError, "Unknown value for argument 'wanted': %s" % repr(tpwanted)
|
|
|
|
|
|
def test():
|
|
import time, sys
|
|
|
|
Message("Testing EasyDialogs.")
|
|
optionlist = (('v', 'Verbose'), ('verbose', 'Verbose as long option'),
|
|
('flags=', 'Valued option'), ('f:', 'Short valued option'))
|
|
commandlist = (('start', 'Start something'), ('stop', 'Stop something'))
|
|
argv = GetArgv(optionlist=optionlist, commandlist=commandlist, addoldfile=0)
|
|
Message("Command line: %s"%' '.join(argv))
|
|
for i in range(len(argv)):
|
|
print 'arg[%d] = %s'%(i, `argv[i]`)
|
|
ok = AskYesNoCancel("Do you want to proceed?")
|
|
ok = AskYesNoCancel("Do you want to identify?", yes="Identify", no="No")
|
|
if ok > 0:
|
|
s = AskString("Enter your first name", "Joe")
|
|
s2 = AskPassword("Okay %s, tell us your nickname"%s, s, cancel="None")
|
|
if not s2:
|
|
Message("%s has no secret nickname"%s)
|
|
else:
|
|
Message("Hello everybody!!\nThe secret nickname of %s is %s!!!"%(s, s2))
|
|
else:
|
|
s = 'Anonymous'
|
|
rv = AskFileForOpen(message="Gimme a file, %s"%s, wanted=macfs.FSSpec)
|
|
Message("rv: %s"%rv)
|
|
rv = AskFileForSave(wanted=macfs.FSSpec, savedFileName="%s.txt"%s)
|
|
Message("rv.as_pathname: %s"%rv.as_pathname())
|
|
rv = AskFolder()
|
|
Message("Folder name: %s"%rv)
|
|
text = ( "Working Hard...", "Hardly Working..." ,
|
|
"So far, so good!", "Keep on truckin'" )
|
|
bar = ProgressBar("Progress, progress...", 0, label="Ramping up...")
|
|
try:
|
|
if hasattr(MacOS, 'SchedParams'):
|
|
appsw = MacOS.SchedParams(1, 0)
|
|
for i in xrange(20):
|
|
bar.inc()
|
|
time.sleep(0.05)
|
|
bar.set(0,100)
|
|
for i in xrange(100):
|
|
bar.set(i)
|
|
time.sleep(0.05)
|
|
if i % 10 == 0:
|
|
bar.label(text[(i/10) % 4])
|
|
bar.label("Done.")
|
|
time.sleep(1.0) # give'em a chance to see "Done."
|
|
finally:
|
|
del bar
|
|
if hasattr(MacOS, 'SchedParams'):
|
|
apply(MacOS.SchedParams, appsw)
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
test()
|
|
except KeyboardInterrupt:
|
|
Message("Operation Canceled.")
|
|
|