home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2010 May / maximum-cd-2010-05.iso / DiscContents / boxee-0.9.20.10711.exe / system / python / Lib / plat-mac / EasyDialogs.pyo (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2009-07-20  |  24.5 KB  |  860 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.4)
  3.  
  4. '''Easy to use dialogs.
  5.  
  6. Message(msg) -- display a message and an OK button.
  7. AskString(prompt, default) -- ask for a string, display OK and Cancel buttons.
  8. AskPassword(prompt, default) -- like AskString(), but shows text as bullets.
  9. AskYesNoCancel(question, default) -- display a question and Yes, No and Cancel buttons.
  10. GetArgv(optionlist, commandlist) -- fill a sys.argv-like list using a dialog
  11. AskFileForOpen(...) -- Ask the user for an existing file
  12. AskFileForSave(...) -- Ask the user for an output file
  13. AskFolder(...) -- Ask the user to select a folder
  14. bar = Progress(label, maxvalue) -- Display a progress bar
  15. bar.set(value) -- Set value
  16. bar.inc( *amount ) -- increment value by amount (default=1)
  17. bar.label( *newlabel ) -- get or set text label.
  18.  
  19. More documentation in each function.
  20. This module uses DLOG resources 260 and on.
  21. Based upon STDWIN dialogs with the same names and functions.
  22. '''
  23. from Carbon.Dlg import GetNewDialog, SetDialogItemText, GetDialogItemText, ModalDialog
  24. from Carbon import Qd
  25. from Carbon import QuickDraw
  26. from Carbon import Dialogs
  27. from Carbon import Windows
  28. from Carbon import Dlg, Win, Evt, Events
  29. from Carbon import Ctl
  30. from Carbon import Controls
  31. from Carbon import Menu
  32. from Carbon import AE
  33. import Nav
  34. import MacOS
  35. import string
  36. from Carbon.ControlAccessor import *
  37. import Carbon.File as Carbon
  38. import macresource
  39. import os
  40. import sys
  41. __all__ = [
  42.     'Message',
  43.     'AskString',
  44.     'AskPassword',
  45.     'AskYesNoCancel',
  46.     'GetArgv',
  47.     'AskFileForOpen',
  48.     'AskFileForSave',
  49.     'AskFolder',
  50.     'ProgressBar']
  51. _initialized = 0
  52.  
  53. def _initialize():
  54.     if _initialized:
  55.         return None
  56.     
  57.     macresource.need('DLOG', 260, 'dialogs.rsrc', __name__)
  58.  
  59.  
  60. def _interact():
  61.     '''Make sure the application is in the foreground'''
  62.     AE.AEInteractWithUser(50000000)
  63.  
  64.  
  65. def cr2lf(text):
  66.     if '\r' in text:
  67.         text = string.join(string.split(text, '\r'), '\n')
  68.     
  69.     return text
  70.  
  71.  
  72. def lf2cr(text):
  73.     if '\n' in text:
  74.         text = string.join(string.split(text, '\n'), '\r')
  75.     
  76.     if len(text) > 253:
  77.         text = text[:253] + '\xc9'
  78.     
  79.     return text
  80.  
  81.  
  82. def Message(msg, id = 260, ok = None):
  83.     '''Display a MESSAGE string.
  84.  
  85.     Return when the user clicks the OK button or presses Return.
  86.  
  87.     The MESSAGE string can be at most 255 characters long.
  88.     '''
  89.     _initialize()
  90.     _interact()
  91.     d = GetNewDialog(id, -1)
  92.     if not d:
  93.         print "EasyDialogs: Can't get DLOG resource with id =", id, ' (missing resource file?)'
  94.         return None
  95.     
  96.     h = d.GetDialogItemAsControl(2)
  97.     SetDialogItemText(h, lf2cr(msg))
  98.     if ok != None:
  99.         h = d.GetDialogItemAsControl(1)
  100.         h.SetControlTitle(ok)
  101.     
  102.     d.SetDialogDefaultItem(1)
  103.     d.AutoSizeDialog()
  104.     d.GetDialogWindow().ShowWindow()
  105.     while None:
  106.         n = ModalDialog(None)
  107.         if n == 1:
  108.             return None
  109.             continue
  110.  
  111.  
  112. def AskString(prompt, default = '', id = 261, ok = None, cancel = None):
  113.     '''Display a PROMPT string and a text entry field with a DEFAULT string.
  114.  
  115.     Return the contents of the text entry field when the user clicks the
  116.     OK button or presses Return.
  117.     Return None when the user clicks the Cancel button.
  118.  
  119.     If omitted, DEFAULT is empty.
  120.  
  121.     The PROMPT and DEFAULT strings, as well as the return value,
  122.     can be at most 255 characters long.
  123.     '''
  124.     _initialize()
  125.     _interact()
  126.     d = GetNewDialog(id, -1)
  127.     if not d:
  128.         print "EasyDialogs: Can't get DLOG resource with id =", id, ' (missing resource file?)'
  129.         return None
  130.     
  131.     h = d.GetDialogItemAsControl(3)
  132.     SetDialogItemText(h, lf2cr(prompt))
  133.     h = d.GetDialogItemAsControl(4)
  134.     SetDialogItemText(h, lf2cr(default))
  135.     d.SelectDialogItemText(4, 0, 999)
  136.     if ok != None:
  137.         h = d.GetDialogItemAsControl(1)
  138.         h.SetControlTitle(ok)
  139.     
  140.     if cancel != None:
  141.         h = d.GetDialogItemAsControl(2)
  142.         h.SetControlTitle(cancel)
  143.     
  144.     d.SetDialogDefaultItem(1)
  145.     d.SetDialogCancelItem(2)
  146.     d.AutoSizeDialog()
  147.     d.GetDialogWindow().ShowWindow()
  148.     while None:
  149.         n = ModalDialog(None)
  150.         if n == 1:
  151.             h = d.GetDialogItemAsControl(4)
  152.             return cr2lf(GetDialogItemText(h))
  153.         
  154.         if n == 2:
  155.             return None
  156.             continue
  157.  
  158.  
  159. def AskPassword(prompt, default = '', id = 264, ok = None, cancel = None):
  160.     '''Display a PROMPT string and a text entry field with a DEFAULT string.
  161.     The string is displayed as bullets only.
  162.  
  163.     Return the contents of the text entry field when the user clicks the
  164.     OK button or presses Return.
  165.     Return None when the user clicks the Cancel button.
  166.  
  167.     If omitted, DEFAULT is empty.
  168.  
  169.     The PROMPT and DEFAULT strings, as well as the return value,
  170.     can be at most 255 characters long.
  171.     '''
  172.     _initialize()
  173.     _interact()
  174.     d = GetNewDialog(id, -1)
  175.     if not d:
  176.         print "EasyDialogs: Can't get DLOG resource with id =", id, ' (missing resource file?)'
  177.         return None
  178.     
  179.     h = d.GetDialogItemAsControl(3)
  180.     SetDialogItemText(h, lf2cr(prompt))
  181.     pwd = d.GetDialogItemAsControl(4)
  182.     bullets = '\xa5' * len(default)
  183.     SetControlData(pwd, kControlEditTextPart, kControlEditTextPasswordTag, default)
  184.     d.SelectDialogItemText(4, 0, 999)
  185.     Ctl.SetKeyboardFocus(d.GetDialogWindow(), pwd, kControlEditTextPart)
  186.     if ok != None:
  187.         h = d.GetDialogItemAsControl(1)
  188.         h.SetControlTitle(ok)
  189.     
  190.     if cancel != None:
  191.         h = d.GetDialogItemAsControl(2)
  192.         h.SetControlTitle(cancel)
  193.     
  194.     d.SetDialogDefaultItem(Dialogs.ok)
  195.     d.SetDialogCancelItem(Dialogs.cancel)
  196.     d.AutoSizeDialog()
  197.     d.GetDialogWindow().ShowWindow()
  198.     while None:
  199.         n = ModalDialog(None)
  200.         if n == 1:
  201.             h = d.GetDialogItemAsControl(4)
  202.             return cr2lf(GetControlData(pwd, kControlEditTextPart, kControlEditTextPasswordTag))
  203.         
  204.         if n == 2:
  205.             return None
  206.             continue
  207.  
  208.  
  209. def AskYesNoCancel(question, default = 0, yes = None, no = None, cancel = None, id = 262):
  210.     '''Display a QUESTION string which can be answered with Yes or No.
  211.  
  212.     Return 1 when the user clicks the Yes button.
  213.     Return 0 when the user clicks the No button.
  214.     Return -1 when the user clicks the Cancel button.
  215.  
  216.     When the user presses Return, the DEFAULT value is returned.
  217.     If omitted, this is 0 (No).
  218.  
  219.     The QUESTION string can be at most 255 characters.
  220.     '''
  221.     _initialize()
  222.     _interact()
  223.     d = GetNewDialog(id, -1)
  224.     if not d:
  225.         print "EasyDialogs: Can't get DLOG resource with id =", id, ' (missing resource file?)'
  226.         return None
  227.     
  228.     h = d.GetDialogItemAsControl(5)
  229.     SetDialogItemText(h, lf2cr(question))
  230.     if yes != None:
  231.         if yes == '':
  232.             d.HideDialogItem(2)
  233.         else:
  234.             h = d.GetDialogItemAsControl(2)
  235.             h.SetControlTitle(yes)
  236.     
  237.     if no != None:
  238.         if no == '':
  239.             d.HideDialogItem(3)
  240.         else:
  241.             h = d.GetDialogItemAsControl(3)
  242.             h.SetControlTitle(no)
  243.     
  244.     if cancel != None:
  245.         if cancel == '':
  246.             d.HideDialogItem(4)
  247.         else:
  248.             h = d.GetDialogItemAsControl(4)
  249.             h.SetControlTitle(cancel)
  250.     
  251.     d.SetDialogCancelItem(4)
  252.     if default == 1:
  253.         d.SetDialogDefaultItem(2)
  254.     elif default == 0:
  255.         d.SetDialogDefaultItem(3)
  256.     elif default == -1:
  257.         d.SetDialogDefaultItem(4)
  258.     
  259.     d.AutoSizeDialog()
  260.     d.GetDialogWindow().ShowWindow()
  261.     while None:
  262.         n = ModalDialog(None)
  263.         if n == 1:
  264.             return default
  265.         
  266.         if n == 2:
  267.             return 1
  268.         
  269.         if n == 3:
  270.             return 0
  271.         
  272.         if n == 4:
  273.             return -1
  274.             continue
  275.  
  276. screenbounds = Qd.GetQDGlobalsScreenBits().bounds
  277. screenbounds = (screenbounds[0] + 4, screenbounds[1] + 4, screenbounds[2] - 4, screenbounds[3] - 4)
  278. kControlProgressBarIndeterminateTag = 'inde'
  279.  
  280. class ProgressBar:
  281.     
  282.     def __init__(self, title = 'Working...', maxval = 0, label = '', id = 263):
  283.         self.w = None
  284.         self.d = None
  285.         _initialize()
  286.         self.d = GetNewDialog(id, -1)
  287.         self.w = self.d.GetDialogWindow()
  288.         self.label(label)
  289.         self.title(title)
  290.         self.set(0, maxval)
  291.         self.d.AutoSizeDialog()
  292.         self.w.ShowWindow()
  293.         self.d.DrawDialog()
  294.  
  295.     
  296.     def __del__(self):
  297.         if self.w:
  298.             self.w.BringToFront()
  299.             self.w.HideWindow()
  300.         
  301.         del self.w
  302.         del self.d
  303.  
  304.     
  305.     def title(self, newstr = ''):
  306.         '''title(text) - Set title of progress window'''
  307.         self.w.BringToFront()
  308.         self.w.SetWTitle(newstr)
  309.  
  310.     
  311.     def label(self, *newstr):
  312.         '''label(text) - Set text in progress box'''
  313.         self.w.BringToFront()
  314.         if newstr:
  315.             self._label = lf2cr(newstr[0])
  316.         
  317.         text_h = self.d.GetDialogItemAsControl(2)
  318.         SetDialogItemText(text_h, self._label)
  319.  
  320.     
  321.     def _update(self, value):
  322.         maxval = self.maxval
  323.         if maxval == 0:
  324.             Ctl.IdleControls(self.w)
  325.         elif maxval > 32767:
  326.             value = int(value / maxval / 32767.0)
  327.             maxval = 32767
  328.         
  329.         maxval = int(maxval)
  330.         value = int(value)
  331.         progbar = self.d.GetDialogItemAsControl(3)
  332.         progbar.SetControlMaximum(maxval)
  333.         progbar.SetControlValue(value)
  334.         (ready, ev) = Evt.WaitNextEvent(Events.mDownMask, 1)
  335.         if ready:
  336.             (what, msg, when, where, mod) = ev
  337.             part = Win.FindWindow(where)[0]
  338.             if Dlg.IsDialogEvent(ev):
  339.                 ds = Dlg.DialogSelect(ev)
  340.                 if ds[0] and ds[1] == self.d and ds[-1] == 1:
  341.                     self.w.HideWindow()
  342.                     self.w = None
  343.                     self.d = None
  344.                     raise KeyboardInterrupt, ev
  345.                 
  346.             elif part == 4:
  347.                 self.w.DragWindow(where, screenbounds)
  348.             else:
  349.                 MacOS.HandleEvent(ev)
  350.         
  351.  
  352.     
  353.     def set(self, value, max = None):
  354.         '''set(value) - Set progress bar position'''
  355.         if max != None:
  356.             self.maxval = max
  357.             bar = self.d.GetDialogItemAsControl(3)
  358.             if max <= 0:
  359.                 bar.SetControlData(0, kControlProgressBarIndeterminateTag, '\x01')
  360.             else:
  361.                 bar.SetControlData(0, kControlProgressBarIndeterminateTag, '\x00')
  362.         
  363.         if value < 0:
  364.             value = 0
  365.         elif value > self.maxval:
  366.             value = self.maxval
  367.         
  368.         self.curval = value
  369.         self._update(value)
  370.  
  371.     
  372.     def inc(self, n = 1):
  373.         '''inc(amt) - Increment progress bar position'''
  374.         self.set(self.curval + n)
  375.  
  376.  
  377. ARGV_ID = 265
  378. ARGV_ITEM_OK = 1
  379. ARGV_ITEM_CANCEL = 2
  380. ARGV_OPTION_GROUP = 3
  381. ARGV_OPTION_EXPLAIN = 4
  382. ARGV_OPTION_VALUE = 5
  383. ARGV_OPTION_ADD = 6
  384. ARGV_COMMAND_GROUP = 7
  385. ARGV_COMMAND_EXPLAIN = 8
  386. ARGV_COMMAND_ADD = 9
  387. ARGV_ADD_OLDFILE = 10
  388. ARGV_ADD_NEWFILE = 11
  389. ARGV_ADD_FOLDER = 12
  390. ARGV_CMDLINE_GROUP = 13
  391. ARGV_CMDLINE_DATA = 14
  392.  
  393. def _setmenu(control, items):
  394.     mhandle = control.GetControlData_Handle(Controls.kControlMenuPart, Controls.kControlPopupButtonMenuHandleTag)
  395.     menu = Menu.as_Menu(mhandle)
  396.     for item in items:
  397.         if type(item) == type(()):
  398.             label = item[0]
  399.         else:
  400.             label = item
  401.         if label[-1] == '=' or label[-1] == ':':
  402.             label = label[:-1]
  403.         
  404.         menu.AppendMenu(label)
  405.     
  406.     control.SetControlMinimum(1)
  407.     control.SetControlMaximum(len(items) + 1)
  408.  
  409.  
  410. def _selectoption(d, optionlist, idx):
  411.     if idx < 0 or idx >= len(optionlist):
  412.         MacOS.SysBeep()
  413.         return None
  414.     
  415.     option = optionlist[idx]
  416.     if type(option) == type(()):
  417.         if len(option) == 4:
  418.             help = option[2]
  419.         elif len(option) > 1:
  420.             help = option[-1]
  421.         else:
  422.             help = ''
  423.     else:
  424.         help = ''
  425.     h = d.GetDialogItemAsControl(ARGV_OPTION_EXPLAIN)
  426.     if help and len(help) > 250:
  427.         help = help[:250] + '...'
  428.     
  429.     Dlg.SetDialogItemText(h, help)
  430.     hasvalue = 0
  431.     if type(option) == type(()):
  432.         label = option[0]
  433.     else:
  434.         label = option
  435.     if label[-1] == '=' or label[-1] == ':':
  436.         hasvalue = 1
  437.     
  438.     h = d.GetDialogItemAsControl(ARGV_OPTION_VALUE)
  439.     Dlg.SetDialogItemText(h, '')
  440.     if hasvalue:
  441.         d.ShowDialogItem(ARGV_OPTION_VALUE)
  442.         d.SelectDialogItemText(ARGV_OPTION_VALUE, 0, 0)
  443.     else:
  444.         d.HideDialogItem(ARGV_OPTION_VALUE)
  445.  
  446.  
  447. def GetArgv(optionlist = None, commandlist = None, addoldfile = 1, addnewfile = 1, addfolder = 1, id = ARGV_ID):
  448.     _initialize()
  449.     _interact()
  450.     d = GetNewDialog(id, -1)
  451.     if not d:
  452.         print "EasyDialogs: Can't get DLOG resource with id =", id, ' (missing resource file?)'
  453.         return None
  454.     
  455.     if optionlist:
  456.         _setmenu(d.GetDialogItemAsControl(ARGV_OPTION_GROUP), optionlist)
  457.         _selectoption(d, optionlist, 0)
  458.     else:
  459.         d.GetDialogItemAsControl(ARGV_OPTION_GROUP).DeactivateControl()
  460.     if commandlist:
  461.         _setmenu(d.GetDialogItemAsControl(ARGV_COMMAND_GROUP), commandlist)
  462.         if type(commandlist[0]) == type(()) and len(commandlist[0]) > 1:
  463.             help = commandlist[0][-1]
  464.             h = d.GetDialogItemAsControl(ARGV_COMMAND_EXPLAIN)
  465.             Dlg.SetDialogItemText(h, help)
  466.         
  467.     else:
  468.         d.GetDialogItemAsControl(ARGV_COMMAND_GROUP).DeactivateControl()
  469.     if not addoldfile:
  470.         d.GetDialogItemAsControl(ARGV_ADD_OLDFILE).DeactivateControl()
  471.     
  472.     if not addnewfile:
  473.         d.GetDialogItemAsControl(ARGV_ADD_NEWFILE).DeactivateControl()
  474.     
  475.     if not addfolder:
  476.         d.GetDialogItemAsControl(ARGV_ADD_FOLDER).DeactivateControl()
  477.     
  478.     d.SetDialogDefaultItem(ARGV_ITEM_OK)
  479.     d.SetDialogCancelItem(ARGV_ITEM_CANCEL)
  480.     d.GetDialogWindow().ShowWindow()
  481.     d.DrawDialog()
  482.     if hasattr(MacOS, 'SchedParams'):
  483.         appsw = MacOS.SchedParams(1, 0)
  484.     
  485.     
  486.     try:
  487.         while None:
  488.             stringstoadd = []
  489.             n = ModalDialog(None)
  490.             if n == ARGV_ITEM_OK:
  491.                 break
  492.             elif n == ARGV_ITEM_CANCEL:
  493.                 raise SystemExit
  494.             elif n == ARGV_OPTION_GROUP:
  495.                 idx = d.GetDialogItemAsControl(ARGV_OPTION_GROUP).GetControlValue() - 1
  496.                 _selectoption(d, optionlist, idx)
  497.             elif n == ARGV_OPTION_VALUE:
  498.                 pass
  499.             elif n == ARGV_OPTION_ADD:
  500.                 idx = d.GetDialogItemAsControl(ARGV_OPTION_GROUP).GetControlValue() - 1
  501.                 if idx <= idx:
  502.                     pass
  503.                 elif idx < len(optionlist):
  504.                     option = optionlist[idx]
  505.                     if type(option) == type(()):
  506.                         option = option[0]
  507.                     
  508.                     if option[-1] == '=' or option[-1] == ':':
  509.                         option = option[:-1]
  510.                         h = d.GetDialogItemAsControl(ARGV_OPTION_VALUE)
  511.                         value = Dlg.GetDialogItemText(h)
  512.                     else:
  513.                         value = ''
  514.                     if len(option) == 1:
  515.                         stringtoadd = '-' + option
  516.                     else:
  517.                         stringtoadd = '--' + option
  518.                     stringstoadd = [
  519.                         stringtoadd]
  520.                     if value:
  521.                         stringstoadd.append(value)
  522.                     
  523.                 else:
  524.                     MacOS.SysBeep()
  525.             elif n == ARGV_COMMAND_GROUP:
  526.                 idx = d.GetDialogItemAsControl(ARGV_COMMAND_GROUP).GetControlValue() - 1
  527.                 if idx <= idx:
  528.                     pass
  529.                 elif idx < len(commandlist) and type(commandlist[idx]) == type(()) and len(commandlist[idx]) > 1:
  530.                     help = commandlist[idx][-1]
  531.                     h = d.GetDialogItemAsControl(ARGV_COMMAND_EXPLAIN)
  532.                     Dlg.SetDialogItemText(h, help)
  533.                 
  534.             elif n == ARGV_COMMAND_ADD:
  535.                 idx = d.GetDialogItemAsControl(ARGV_COMMAND_GROUP).GetControlValue() - 1
  536.                 if idx <= idx:
  537.                     pass
  538.                 elif idx < len(commandlist):
  539.                     command = commandlist[idx]
  540.                     if type(command) == type(()):
  541.                         command = command[0]
  542.                     
  543.                     stringstoadd = [
  544.                         command]
  545.                 else:
  546.                     MacOS.SysBeep()
  547.             elif n == ARGV_ADD_OLDFILE:
  548.                 pathname = AskFileForOpen()
  549.                 if pathname:
  550.                     stringstoadd = [
  551.                         pathname]
  552.                 
  553.             elif n == ARGV_ADD_NEWFILE:
  554.                 pathname = AskFileForSave()
  555.                 if pathname:
  556.                     stringstoadd = [
  557.                         pathname]
  558.                 
  559.             elif n == ARGV_ADD_FOLDER:
  560.                 pathname = AskFolder()
  561.                 if pathname:
  562.                     stringstoadd = [
  563.                         pathname]
  564.                 
  565.             elif n == ARGV_CMDLINE_DATA:
  566.                 pass
  567.             else:
  568.                 raise RuntimeError, 'Unknown dialog item %d' % n
  569.             for stringtoadd in stringstoadd:
  570.                 if '"' in stringtoadd and "'" in stringtoadd or ' ' in stringtoadd:
  571.                     stringtoadd = repr(stringtoadd)
  572.                 
  573.                 h = d.GetDialogItemAsControl(ARGV_CMDLINE_DATA)
  574.                 oldstr = GetDialogItemText(h)
  575.                 if oldstr and oldstr[-1] != ' ':
  576.                     oldstr = oldstr + ' '
  577.                 
  578.                 oldstr = oldstr + stringtoadd
  579.                 if oldstr[-1] != ' ':
  580.                     oldstr = oldstr + ' '
  581.                 
  582.                 SetDialogItemText(h, oldstr)
  583.                 d.SelectDialogItemText(ARGV_CMDLINE_DATA, 32767, 32767)
  584.             
  585.         h = d.GetDialogItemAsControl(ARGV_CMDLINE_DATA)
  586.         oldstr = GetDialogItemText(h)
  587.         tmplist = string.split(oldstr)
  588.         newlist = []
  589.         while tmplist:
  590.             item = tmplist[0]
  591.             del tmplist[0]
  592.             if item[0] == '"':
  593.                 while item[-1] != '"':
  594.                     if not tmplist:
  595.                         raise RuntimeError, 'Unterminated quoted argument'
  596.                     
  597.                     item = item + ' ' + tmplist[0]
  598.                     del tmplist[0]
  599.                 item = item[1:-1]
  600.             
  601.             if item[0] == "'":
  602.                 while item[-1] != "'":
  603.                     if not tmplist:
  604.                         raise RuntimeError, 'Unterminated quoted argument'
  605.                     
  606.                     item = item + ' ' + tmplist[0]
  607.                     del tmplist[0]
  608.                 item = item[1:-1]
  609.             
  610.             newlist.append(item)
  611.         return newlist
  612.     finally:
  613.         if hasattr(MacOS, 'SchedParams'):
  614.             MacOS.SchedParams(*appsw)
  615.         
  616.         del d
  617.  
  618.  
  619.  
  620. def _process_Nav_args(dftflags, **args):
  621.     import aepack
  622.     import Carbon.AE as Carbon
  623.     import Carbon.File as Carbon
  624.     for k in args.keys():
  625.         if args[k] is None:
  626.             del args[k]
  627.             continue
  628.     
  629.     if not args.has_key('dialogOptionFlags'):
  630.         args['dialogOptionFlags'] = dftflags
  631.     
  632.     if args.has_key('defaultLocation') and not isinstance(args['defaultLocation'], Carbon.AE.AEDesc):
  633.         defaultLocation = args['defaultLocation']
  634.         if isinstance(defaultLocation, (Carbon.File.FSSpec, Carbon.File.FSRef)):
  635.             args['defaultLocation'] = aepack.pack(defaultLocation)
  636.         else:
  637.             defaultLocation = Carbon.File.FSRef(defaultLocation)
  638.             args['defaultLocation'] = aepack.pack(defaultLocation)
  639.     
  640.     if args.has_key('typeList') and not isinstance(args['typeList'], Carbon.Res.ResourceType):
  641.         typeList = args['typeList'][:]
  642.         if 'TEXT' in typeList and '\x00\x00\x00\x00' not in typeList:
  643.             typeList = typeList + ('\x00\x00\x00\x00',)
  644.         
  645.         data = 'Pyth' + struct.pack('hh', 0, len(typeList))
  646.         for type in typeList:
  647.             data = data + type
  648.         
  649.         args['typeList'] = Carbon.Res.Handle(data)
  650.     
  651.     tpwanted = str
  652.     if args.has_key('wanted'):
  653.         tpwanted = args['wanted']
  654.         del args['wanted']
  655.     
  656.     return (args, tpwanted)
  657.  
  658.  
  659. def _dummy_Nav_eventproc(msg, data):
  660.     pass
  661.  
  662. _default_Nav_eventproc = _dummy_Nav_eventproc
  663.  
  664. def SetDefaultEventProc(proc):
  665.     global _default_Nav_eventproc
  666.     rv = _default_Nav_eventproc
  667.     if proc is None:
  668.         proc = _dummy_Nav_eventproc
  669.     
  670.     _default_Nav_eventproc = proc
  671.     return rv
  672.  
  673.  
  674. def AskFileForOpen(message = None, typeList = None, version = None, defaultLocation = None, dialogOptionFlags = None, location = None, clientName = None, windowTitle = None, actionButtonLabel = None, cancelButtonLabel = None, preferenceKey = None, popupExtension = None, eventProc = _dummy_Nav_eventproc, previewProc = None, filterProc = None, wanted = None, multiple = None):
  675.     """Display a dialog asking the user for a file to open.
  676.  
  677.     wanted is the return type wanted: FSSpec, FSRef, unicode or string (default)
  678.     the other arguments can be looked up in Apple's Navigation Services documentation"""
  679.     default_flags = 86
  680.     (args, tpwanted) = _process_Nav_args(default_flags, version = version, defaultLocation = defaultLocation, dialogOptionFlags = dialogOptionFlags, location = location, clientName = clientName, windowTitle = windowTitle, actionButtonLabel = actionButtonLabel, cancelButtonLabel = cancelButtonLabel, message = message, preferenceKey = preferenceKey, popupExtension = popupExtension, eventProc = eventProc, previewProc = previewProc, filterProc = filterProc, typeList = typeList, wanted = wanted, multiple = multiple)
  681.     _interact()
  682.     
  683.     try:
  684.         rr = Nav.NavChooseFile(args)
  685.         good = 1
  686.     except Nav.error:
  687.         arg = None
  688.         if arg[0] != -128:
  689.             raise Nav.error, arg
  690.         
  691.         return None
  692.  
  693.     if not (rr.validRecord) or not (rr.selection):
  694.         return None
  695.     
  696.     if issubclass(tpwanted, Carbon.File.FSRef):
  697.         return tpwanted(rr.selection_fsr[0])
  698.     
  699.     if issubclass(tpwanted, Carbon.File.FSSpec):
  700.         return tpwanted(rr.selection[0])
  701.     
  702.     if issubclass(tpwanted, str):
  703.         return tpwanted(rr.selection_fsr[0].as_pathname())
  704.     
  705.     if issubclass(tpwanted, unicode):
  706.         return tpwanted(rr.selection_fsr[0].as_pathname(), 'utf8')
  707.     
  708.     raise TypeError, "Unknown value for argument 'wanted': %s" % repr(tpwanted)
  709.  
  710.  
  711. def AskFileForSave(message = None, savedFileName = None, version = None, defaultLocation = None, dialogOptionFlags = None, location = None, clientName = None, windowTitle = None, actionButtonLabel = None, cancelButtonLabel = None, preferenceKey = None, popupExtension = None, eventProc = _dummy_Nav_eventproc, fileType = None, fileCreator = None, wanted = None, multiple = None):
  712.     """Display a dialog asking the user for a filename to save to.
  713.  
  714.     wanted is the return type wanted: FSSpec, FSRef, unicode or string (default)
  715.     the other arguments can be looked up in Apple's Navigation Services documentation"""
  716.     default_flags = 7
  717.     (args, tpwanted) = _process_Nav_args(default_flags, version = version, defaultLocation = defaultLocation, dialogOptionFlags = dialogOptionFlags, location = location, clientName = clientName, windowTitle = windowTitle, actionButtonLabel = actionButtonLabel, cancelButtonLabel = cancelButtonLabel, savedFileName = savedFileName, message = message, preferenceKey = preferenceKey, popupExtension = popupExtension, eventProc = eventProc, fileType = fileType, fileCreator = fileCreator, wanted = wanted, multiple = multiple)
  718.     _interact()
  719.     
  720.     try:
  721.         rr = Nav.NavPutFile(args)
  722.         good = 1
  723.     except Nav.error:
  724.         arg = None
  725.         if arg[0] != -128:
  726.             raise Nav.error, arg
  727.         
  728.         return None
  729.  
  730.     if not (rr.validRecord) or not (rr.selection):
  731.         return None
  732.     
  733.     if issubclass(tpwanted, Carbon.File.FSRef):
  734.         raise TypeError, 'Cannot pass wanted=FSRef to AskFileForSave'
  735.     
  736.     if issubclass(tpwanted, Carbon.File.FSSpec):
  737.         return tpwanted(rr.selection[0])
  738.     
  739.     if issubclass(tpwanted, (str, unicode)):
  740.         if sys.platform == 'mac':
  741.             fullpath = rr.selection[0].as_pathname()
  742.         else:
  743.             (vrefnum, dirid, name) = rr.selection[0].as_tuple()
  744.             pardir_fss = Carbon.File.FSSpec((vrefnum, dirid, ''))
  745.             pardir_fsr = Carbon.File.FSRef(pardir_fss)
  746.             pardir_path = pardir_fsr.FSRefMakePath()
  747.             name_utf8 = unicode(name, 'macroman').encode('utf8')
  748.             fullpath = os.path.join(pardir_path, name_utf8)
  749.         if issubclass(tpwanted, unicode):
  750.             return unicode(fullpath, 'utf8')
  751.         
  752.         return tpwanted(fullpath)
  753.     
  754.     raise TypeError, "Unknown value for argument 'wanted': %s" % repr(tpwanted)
  755.  
  756.  
  757. def AskFolder(message = None, version = None, defaultLocation = None, dialogOptionFlags = None, location = None, clientName = None, windowTitle = None, actionButtonLabel = None, cancelButtonLabel = None, preferenceKey = None, popupExtension = None, eventProc = _dummy_Nav_eventproc, filterProc = None, wanted = None, multiple = None):
  758.     """Display a dialog asking the user for select a folder.
  759.  
  760.     wanted is the return type wanted: FSSpec, FSRef, unicode or string (default)
  761.     the other arguments can be looked up in Apple's Navigation Services documentation"""
  762.     default_flags = 23
  763.     (args, tpwanted) = _process_Nav_args(default_flags, version = version, defaultLocation = defaultLocation, dialogOptionFlags = dialogOptionFlags, location = location, clientName = clientName, windowTitle = windowTitle, actionButtonLabel = actionButtonLabel, cancelButtonLabel = cancelButtonLabel, message = message, preferenceKey = preferenceKey, popupExtension = popupExtension, eventProc = eventProc, filterProc = filterProc, wanted = wanted, multiple = multiple)
  764.     _interact()
  765.     
  766.     try:
  767.         rr = Nav.NavChooseFolder(args)
  768.         good = 1
  769.     except Nav.error:
  770.         arg = None
  771.         if arg[0] != -128:
  772.             raise Nav.error, arg
  773.         
  774.         return None
  775.  
  776.     if not (rr.validRecord) or not (rr.selection):
  777.         return None
  778.     
  779.     if issubclass(tpwanted, Carbon.File.FSRef):
  780.         return tpwanted(rr.selection_fsr[0])
  781.     
  782.     if issubclass(tpwanted, Carbon.File.FSSpec):
  783.         return tpwanted(rr.selection[0])
  784.     
  785.     if issubclass(tpwanted, str):
  786.         return tpwanted(rr.selection_fsr[0].as_pathname())
  787.     
  788.     if issubclass(tpwanted, unicode):
  789.         return tpwanted(rr.selection_fsr[0].as_pathname(), 'utf8')
  790.     
  791.     raise TypeError, "Unknown value for argument 'wanted': %s" % repr(tpwanted)
  792.  
  793.  
  794. def test():
  795.     import time
  796.     Message('Testing EasyDialogs.')
  797.     optionlist = (('v', 'Verbose'), ('verbose', 'Verbose as long option'), ('flags=', 'Valued option'), ('f:', 'Short valued option'))
  798.     commandlist = (('start', 'Start something'), ('stop', 'Stop something'))
  799.     argv = GetArgv(optionlist = optionlist, commandlist = commandlist, addoldfile = 0)
  800.     Message('Command line: %s' % ' '.join(argv))
  801.     for i in range(len(argv)):
  802.         print 'arg[%d] = %r' % (i, argv[i])
  803.     
  804.     ok = AskYesNoCancel('Do you want to proceed?')
  805.     ok = AskYesNoCancel('Do you want to identify?', yes = 'Identify', no = 'No')
  806.     if ok > 0:
  807.         s = AskString('Enter your first name', 'Joe')
  808.         s2 = AskPassword('Okay %s, tell us your nickname' % s, s, cancel = 'None')
  809.         if not s2:
  810.             Message('%s has no secret nickname' % s)
  811.         else:
  812.             Message('Hello everybody!!\nThe secret nickname of %s is %s!!!' % (s, s2))
  813.     else:
  814.         s = 'Anonymous'
  815.     rv = AskFileForOpen(message = 'Gimme a file, %s' % s, wanted = Carbon.File.FSSpec)
  816.     Message('rv: %s' % rv)
  817.     rv = AskFileForSave(wanted = Carbon.File.FSRef, savedFileName = '%s.txt' % s)
  818.     Message('rv.as_pathname: %s' % rv.as_pathname())
  819.     rv = AskFolder()
  820.     Message('Folder name: %s' % rv)
  821.     text = ('Working Hard...', 'Hardly Working...', 'So far, so good!', "Keep on truckin'")
  822.     bar = ProgressBar('Progress, progress...', 0, label = 'Ramping up...')
  823.     
  824.     try:
  825.         if hasattr(MacOS, 'SchedParams'):
  826.             appsw = MacOS.SchedParams(1, 0)
  827.         
  828.         for i in xrange(20):
  829.             bar.inc()
  830.             time.sleep(0.050000000000000003)
  831.         
  832.         bar.set(0, 100)
  833.         for i in xrange(100):
  834.             bar.set(i)
  835.             time.sleep(0.050000000000000003)
  836.             if i % 10 == 0:
  837.                 bar.label(text[i / 10 % 4])
  838.                 continue
  839.         
  840.         bar.label('Done.')
  841.         time.sleep(1.0)
  842.     finally:
  843.         del bar
  844.         if hasattr(MacOS, 'SchedParams'):
  845.             MacOS.SchedParams(*appsw)
  846.         
  847.  
  848.  
  849. if __name__ == '__main__':
  850.     
  851.     try:
  852.         test()
  853.     except KeyboardInterrupt:
  854.         Message('Operation Canceled.')
  855.     except:
  856.         None<EXCEPTION MATCH>KeyboardInterrupt
  857.     
  858.  
  859. None<EXCEPTION MATCH>KeyboardInterrupt
  860.