diff options
Diffstat (limited to 'Mac/scripts')
| -rw-r--r-- | Mac/scripts/BuildApplet.py | 260 | ||||
| -rw-r--r-- | Mac/scripts/EditPythonPrefs.py | 168 | ||||
| -rw-r--r-- | Mac/scripts/PackLibDir.py | 93 | ||||
| -rw-r--r-- | Mac/scripts/RunLibScript.py | 108 |
4 files changed, 629 insertions, 0 deletions
diff --git a/Mac/scripts/BuildApplet.py b/Mac/scripts/BuildApplet.py new file mode 100644 index 0000000..cf90901 --- /dev/null +++ b/Mac/scripts/BuildApplet.py @@ -0,0 +1,260 @@ +"""Create an applet from a Python script. + +This puts up a dialog asking for a Python source file ('TEXT'). +The output is a file with the same name but its ".py" suffix dropped. +It is created by copying an applet template and then adding a 'PYC ' +resource named __main__ containing the compiled, marshalled script. +""" + +import sys +sys.stdout = sys.stderr + +import string +import os +import marshal +import imp +import macfs +import MacOS +from Res import * + +# .pyc file (and 'PYC ' resource magic number) +MAGIC = imp.get_magic() + +# Template file (searched on sys.path) +TEMPLATE = "PythonApplet" + +# Specification of our resource +RESTYPE = 'PYC ' +RESNAME = '__main__' + +# A resource with this name sets the "owner" (creator) of the destination +OWNERNAME = "owner resource" + +# OpenResFile mode parameters +READ = 1 +WRITE = 2 + +def main(): + + # Find the template + # (there's no point in proceeding if we can't find it) + + for p in sys.path: + template = os.path.join(p, TEMPLATE) + try: + tmpl = open(template, "rb") + tmpl.close() + break + except IOError: + continue + else: + die("Template %s not found" % `template`) + return + + # Convert to full pathname + template = macfs.FSSpec(template).as_pathname() + + # Ask for source text if not specified in sys.argv[1:] + + if not sys.argv[1:]: + srcfss, ok = macfs.StandardGetFile('TEXT') + if not ok: + return + filename = srcfss.as_pathname() + tp, tf = os.path.split(filename) + if tf[-3:] == '.py': + tf = tf[:-3] + else: + tf = tf + '.applet' + dstfss, ok = macfs.StandardPutFile('Save application as:', tf) + if not ok: return + process(template, filename, dstfss.as_pathname()) + else: + + # Loop over all files to be processed + for filename in sys.argv[1:]: + process(template, filename, '') + +undefs = ('Atmp', '????', ' ', '\0\0\0\0', 'BINA') + +def process(template, filename, output): + + print "Processing", `filename`, "..." + + # Read the source and compile it + # (there's no point overwriting the destination if it has a syntax error) + + fp = open(filename) + text = fp.read() + fp.close() + try: + code = compile(text, filename, "exec") + except (SyntaxError, EOFError): + die("Syntax error in script %s" % `filename`) + return + + # Set the destination file name + + if string.lower(filename[-3:]) == ".py": + destname = filename[:-3] + rsrcname = destname + '.rsrc' + else: + destname = filename + ".applet" + rsrcname = filename + '.rsrc' + + if output: + destname = output + # Copy the data from the template (creating the file as well) + + tmpl = open(template, "rb") + dest = open(destname, "wb") + data = tmpl.read() + if data: + dest.write(data) + dest.close() + tmpl.close() + + # Copy the creator of the template to the destination + # unless it already got one. Set type to APPL + + tctor, ttype = MacOS.GetCreatorAndType(template) + ctor, type = MacOS.GetCreatorAndType(destname) + if type in undefs: type = 'APPL' + if ctor in undefs: ctor = tctor + + # Open the output resource fork + + try: + output = FSpOpenResFile(destname, WRITE) + except MacOS.Error: + print "Creating resource fork..." + CreateResFile(destname) + output = FSpOpenResFile(destname, WRITE) + + # Copy the resources from the template + + input = FSpOpenResFile(template, READ) + newctor = copyres(input, output) + CloseResFile(input) + if newctor: ctor = newctor + + # Copy the resources from the target specific resource template, if any + + try: + input = FSpOpenResFile(rsrcname, READ) + except MacOS.Error: + pass + else: + newctor = copyres(input, output) + CloseResFile(input) + if newctor: ctor = newctor + + # Now set the creator and type of the destination + + MacOS.SetCreatorAndType(destname, ctor, type) + + # Make sure we're manipulating the output resource file now + + UseResFile(output) + + # Delete any existing 'PYC 'resource named __main__ + + try: + res = Get1NamedResource(RESTYPE, RESNAME) + res.RemoveResource() + except Error: + pass + + # Create the raw data for the resource from the code object + + data = marshal.dumps(code) + del code + data = (MAGIC + '\0\0\0\0') + data + + # Create the resource and write it + + id = 0 + while id < 128: + id = Unique1ID(RESTYPE) + res = Resource(data) + res.AddResource(RESTYPE, id, RESNAME) + res.WriteResource() + res.ReleaseResource() + + # Close the output file + + CloseResFile(output) + + # Give positive feedback + + message("Applet %s created." % `destname`) + + +# Copy resources between two resource file descriptors. +# Exception: don't copy a __main__ resource. +# If a resource's name is "owner resource", its type is returned +# (so the caller can use it to set the destination's creator) + +def copyres(input, output): + ctor = None + UseResFile(input) + ntypes = Count1Types() + for itype in range(1, 1+ntypes): + type = Get1IndType(itype) + nresources = Count1Resources(type) + for ires in range(1, 1+nresources): + res = Get1IndResource(type, ires) + id, type, name = res.GetResInfo() + lcname = string.lower(name) + if (type, lcname) == (RESTYPE, RESNAME): + continue # Don't copy __main__ from template + if lcname == OWNERNAME: ctor = type + size = res.size + attrs = res.GetResAttrs() + print id, type, name, size, hex(attrs) + res.LoadResource() + res.DetachResource() + UseResFile(output) + try: + res2 = Get1Resource(type, id) + except MacOS.Error: + res2 = None + if res2: + print "Overwriting..." + res2.RemoveResource() + res.AddResource(type, id, name) + res.WriteResource() + attrs = attrs | res.GetResAttrs() + print "New attrs =", hex(attrs) + res.SetResAttrs(attrs) + UseResFile(input) + return ctor + + +# Show a message and exit + +def die(str): + message(str) + sys.exit(1) + + +# Show a message + +def message(str, id = 256): + from Dlg import * + d = GetNewDialog(id, -1) + if not d: + print "Error:", `str` + print "DLOG id =", id, "not found." + return + tp, h, rect = d.GetDialogItem(2) + SetDialogItemText(h, str) + while 1: + n = ModalDialog(None) + if n == 1: break + del d + + +if __name__ == '__main__': + main() + diff --git a/Mac/scripts/EditPythonPrefs.py b/Mac/scripts/EditPythonPrefs.py new file mode 100644 index 0000000..3e14ff3 --- /dev/null +++ b/Mac/scripts/EditPythonPrefs.py @@ -0,0 +1,168 @@ +"""Edit the Python Preferences file.""" +import addpack +addpack.addpack('Tools') +addpack.addpack('bgen') +addpack.addpack('evt') + +from Dlg import * +from Events import * +from Res import * +import string +import struct +import macfs +import MacOS +import os +import sys +import Res # For Res.Error + +# resource IDs in our own resources (dialogs, etc) +MESSAGE_ID = 256 + +DIALOG_ID = 131 +TEXT_ITEM = 1 +OK_ITEM = 2 +CANCEL_ITEM = 3 +REVERT_ITEM = 4 +DIR_ITEM = 5 + +# Resource IDs in the preferences file +PATH_STRINGS_ID = 128 +DIRECTORY_ID = 128 + +WRITE = 2 +smAllScripts = -3 +kOnSystemDisk = 0x8000 + +def restolist(data): + """Convert STR# resource data to a list of strings""" + if not data: + return [] + num, = struct.unpack('h', data[:2]) + data = data[2:] + rv = [] + for i in range(num): + strlen = ord(data[0]) + if strlen < 0: strlen = strlen + 256 + str = data[1:strlen+1] + data = data[strlen+1:] + rv.append(str) + return rv + +def listtores(list): + """Convert a list of strings to STR# resource data""" + rv = struct.pack('h', len(list)) + for str in list: + rv = rv + chr(len(str)) + str + return rv + +def message(str = "Hello, world!", id = MESSAGE_ID): + """Show a simple alert with a text message""" + d = GetNewDialog(id, -1) + print 'd=', d + tp, h, rect = d.GetDialogItem(2) + SetDialogItemText(h, str) + while 1: + n = ModalDialog(None) + if n == 1: break + +def interact(list, pythondir): + """Let the user interact with the dialog""" + opythondir = pythondir + try: + # Try to go to the "correct" dir for GetDirectory + os.chdir(pythondir.as_pathname()) + except os.error: + pass + d = GetNewDialog(DIALOG_ID, -1) + tp, h, rect = d.GetDialogItem(1) + SetDialogItemText(h, string.joinfields(list, '\r')) + while 1: + n = ModalDialog(None) + if n == OK_ITEM: + break + if n == CANCEL_ITEM: + return None + if n == REVERT_ITEM: + return [], pythondir + if n == DIR_ITEM: + fss, ok = macfs.GetDirectory() + if ok: + pythondir = fss + tmp = string.splitfields(GetDialogItemText(h), '\r') + rv = [] + for i in tmp: + if i: + rv.append(i) + return rv, pythondir + +def main(): + try: + h = OpenResFile('EditPythonPrefs.rsrc') + except Res.Error: + pass # Assume we already have acces to our own resource + + # Find the preferences folder and our prefs file, create if needed. + vrefnum, dirid = macfs.FindFolder(kOnSystemDisk, 'pref', 0) + preff_fss = macfs.FSSpec((vrefnum, dirid, 'Python Preferences')) + try: + preff_handle = FSpOpenResFile(preff_fss, WRITE) + except Res.Error: + # Create it + message('No preferences file, creating one...') + FSpCreateResFile(preff_fss, 'PYTH', 'pref', smAllScripts) + preff_handle = FSpOpenResFile(preff_fss, WRITE) + + # Load the path and directory resources + try: + sr = GetResource('STR#', PATH_STRINGS_ID) + except (MacOS.Error, Res.Error): + message('Cannot find any sys.path resource! (Old python?)') + sys.exit(0) + d = sr.data + l = restolist(d) + + try: + dr = GetResource('alis', DIRECTORY_ID) + fss, fss_changed = macfs.RawAlias(dr.data).Resolve() + except (MacOS.Error, Res.Error): + dr = None + fss = macfs.FSSpec(os.getcwd()) + fss_changed = 1 + + # Let the user play away + result = interact(l, fss) + + # See what we have to update, and how + if result == None: + sys.exit(0) + + pathlist, nfss = result + if nfss != fss: + fss_changed = 1 + + if fss_changed or pathlist != l: + if fss_changed: + alias = nfss.NewAlias() + if dr: + dr.data = alias.data + dr.ChangedResource() + else: + dr = Resource(alias.data) + dr.AddResource('alis', DIRECTORY_ID, '') + + if pathlist != l: + if pathlist == []: + if sr.HomeResFile() == preff_handle: + sr.RemoveResource() + elif sr.HomeResFile() == preff_handle: + sr.data = listtores(pathlist) + sr.ChangedResource() + else: + sr = Resource(listtores(pathlist)) + sr.AddResource('STR#', PATH_STRINGS_ID, '') + + CloseResFile(preff_handle) + +if __name__ == '__main__': + print # Stupid, to init toolboxes... + main() diff --git a/Mac/scripts/PackLibDir.py b/Mac/scripts/PackLibDir.py new file mode 100644 index 0000000..cdda87d --- /dev/null +++ b/Mac/scripts/PackLibDir.py @@ -0,0 +1,93 @@ +# +# Turn a pyc file into a resource file containing it in 'PYC ' resource form +import addpack +addpack.addpack('Tools') +addpack.addpack('bgen') +addpack.addpack('res') +from Res import * +import Res +from Resources import * +import os +import macfs +import sys + +READ = 1 +WRITE = 2 +smAllScripts = -3 + +error = 'mkpycresourcefile.error' + +def Pstring(str): + if len(str) > 255: + raise ValueError, 'String too large' + return chr(len(str))+str + +def createoutput(dst): + """Create output file. Return handle and first id to use.""" + + + FSpCreateResFile(dst, 'PYTH', 'rsrc', smAllScripts) + output = FSpOpenResFile(dst, WRITE) + UseResFile(output) + num = 128 + return output, num + +def writemodule(name, id, data): + """Write pyc code to a PYC resource with given name and id.""" + # XXXX Check that it doesn't exist + res = Resource(data) + res.AddResource('PYC ', id, name) + res.WriteResource() + res.ReleaseResource() + +def mkpycresourcefile(src, dst): + """Copy pyc file/dir src to resource file dst.""" + + if not os.path.isdir(src) and src[-4:] <> '.pyc': + raise error, 'I can only handle .pyc files or directories' + handle, oid = createoutput(dst) + if os.path.isdir(src): + id = handlesubdir(handle, oid, src) + else: + id = handleonepycfile(handle, oid, src) + print 'Wrote',id-oid,'PYC resources to', dst + CloseResFile(handle) + +def handleonepycfile(handle, id, file): + """Copy one pyc file to the open resource file""" + d, name = os.path.split(file) + name = name[:-4] + print ' module', name + writemodule(name, id, open(file, 'rb').read()) + return id+1 + +def handlesubdir(handle, id, srcdir): + """Recursively scan a directory for pyc files and copy to resources""" + print 'Directory', srcdir + src = os.listdir(srcdir) + for file in src: + file = os.path.join(srcdir, file) + if os.path.isdir(file): + id = handlesubdir(handle, id, file) + elif file[-4:] == '.pyc': + id = handleonepycfile(handle, id, file) + return id + + +if __name__ == '__main__': + args = sys.argv[1:] + if not args: + ifss, ok = macfs.StandardGetFile('PYC ') + if ok: + args = [ifss.as_pathname()] + else: + ifss, ok = macfs.GetDirectory() + if not ok: + sys.exit(0) + args = [ifss.as_pathname()] + for ifn in args: + ofss, ok = macfs.StandardPutFile('Output for '+os.path.split(ifn)[1]) + if not ok: + sys.exit(0) + mkpycresourcefile(ifn, ofss.as_pathname()) + sys.exit(1) # So we can see something... diff --git a/Mac/scripts/RunLibScript.py b/Mac/scripts/RunLibScript.py new file mode 100644 index 0000000..36787d9 --- /dev/null +++ b/Mac/scripts/RunLibScript.py @@ -0,0 +1,108 @@ +"""Import a module while pretending its name is __main__. This +can be used to run scripts from the PackedLib resource file while pretending +they have been double-clicked.""" + +import imp +import sys +import os +import string +import Dlg +import macfs + +DIALOG_ID = 140 +OK = 1 +CANCEL = 2 +SCRIPTNAME=3 +ARGV=4 +STDIN_CONS=5 +STDIN_FILE=6 +STDOUT_CONS=7 +STDOUT_FILE=8 +WORKING_DIR=9 +PAUSE=10 + +def import_as_main(name): + fp, path, (suffix, mode, type) = imp.find_module(name) + if type == imp.PY_SOURCE: + imp.load_source('__main__', path) + elif type == imp.PY_COMPILED: + imp.load_compiled('__main__', path) + elif type == imp.PY_RESOURCE: + imp.load_resource('__main__', path) + +def interact(): + d = Dlg.GetNewDialog(DIALOG_ID, -1) + wdir = stdin = stdout = None + pause = 0 + + tp, in_c_h, rect = d.GetDialogItem(STDIN_CONS) + tp, in_f_h, rect = d.GetDialogItem(STDIN_FILE) + tp, out_c_h, rect = d.GetDialogItem(STDOUT_CONS) + tp, out_f_h, rect = d.GetDialogItem(STDOUT_FILE) + tp, pause_h, rect = d.GetDialogItem(PAUSE) + in_c_h = in_c_h.as_Control() + in_f_h = in_f_h.as_Control() + out_c_h = out_c_h.as_Control() + out_f_h = out_f_h.as_Control() + pause_h = pause_h.as_Control() + + while 1: + in_c_h.SetControlValue(not stdin) + in_f_h.SetControlValue(not not stdin) + out_c_h.SetControlValue(not stdout) + out_f_h.SetControlValue(not not stdout) + pause_h.SetControlValue(pause) + + n = Dlg.ModalDialog(None) + if n == OK: + break + elif n == CANCEL: + sys.exit(0) + elif n == STDIN_CONS: + stdin = None + elif n == STDIN_FILE: + fss, ok = macfs.StandardGetFile('TEXT') + if ok: + stdin = fss + elif n == STDOUT_FILE: + fss, ok = macfs.StandardPutFile('stdout:') + if ok: + stdout = fss + elif n == WORKING_DIR: + fss, ok = macfs.GetDirectory() + if ok: + wdir = fss + elif n == PAUSE: + pause = (not pause) + + tp, h, rect = d.GetDialogItem(SCRIPTNAME) + name = Dlg.GetDialogItemText(h) + tp, h, rect = d.GetDialogItem(ARGV) + argv = Dlg.GetDialogItemText(h) + return name, argv, stdin, stdout, wdir, pause + +def main(): + curdir = os.getcwd() + import Res + Res.OpenResFile('RunLibScript.rsrc') + name, argv, stdin, stdout, wdir, pause = interact() + if not name: + sys.exit(0) + sys.argv = [name] + string.split(argv) + if stdin: + sys.stdin = open(stdin.as_pathname()) + if stdout: + sys.stdout = open(stdout.as_pathname(), 'w') + if wdir: + os.chdir(wdir.as_pathname()) + else: + os.chdir(curdir) + + import_as_main(name) + + if pause: + sys.exit(1) + +if __name__ == '__main__': + main() + |