diff options
-rw-r--r-- | Mac/Lib/appletrawmain.py | 46 | ||||
-rw-r--r-- | Mac/Lib/buildtools.py | 10 |
2 files changed, 56 insertions, 0 deletions
diff --git a/Mac/Lib/appletrawmain.py b/Mac/Lib/appletrawmain.py new file mode 100644 index 0000000..d2fcd6e --- /dev/null +++ b/Mac/Lib/appletrawmain.py @@ -0,0 +1,46 @@ +# Emulate sys.argv and run __main__.py or __main__.pyc in an environment that +# is as close to "normal" as possible. +# +# This script is put into __rawmain__.pyc for applets that need argv +# emulation, by BuildApplet and friends. +# +import argvemulator +import os +import sys +import marshal + +# +# Create sys.argv +# +argvemulator.ArgvCollector().mainloop() +# +# Find the realy main program to run +# +_dir = os.path.split(sys.argv[0])[0] +__file__ = os.path.join(_dir, '__main__.py') +if os.path.exists(__file__): + # + # Setup something resembling a normal environment and go. + # + sys.argv[0] = __file__ + del argvemulator, os, sys, _dir + execfile(__file__) +else: + __file__ = os.path.join(_dir, '__main__.pyc') + if os.path.exists(__file__): + # + # If we have only a .pyc file we read the code object from that + # + sys.argv[0] = __file__ + _fp = open(__file__, 'rb') + _fp.read(8) + __code__ = marshal.load(_fp) + # + # Again, we create an almost-normal environment (only __code__ is + # funny) and go. + # + del argvemulator, os, sys, marshal, _dir, _fp + exec __code__ + else: + sys.stderr.write("%s: neither __main__.py nor __main__.pyc found\n"%sys.argv[0]) + sys.exit(1) diff --git a/Mac/Lib/buildtools.py b/Mac/Lib/buildtools.py index 9825708..1047da5 100644 --- a/Mac/Lib/buildtools.py +++ b/Mac/Lib/buildtools.py @@ -398,6 +398,16 @@ def process_common_macho(template, progress, code, rsrcname, destname, is_update pycname = '__rawmain__.pyc' else: pycname = '__main__.pyc' + # And we also create __rawmain__.pyc + outputfilename = os.path.join(destname, 'Contents', 'Resources', '__rawmain__.pyc') + if progress: + progress.label('Creating __rawmain__.pyc') + progress.inc(0) + rawsourcefile = os.path.join(sys.prefix, 'Mac', 'Lib', 'appletrawmain.py') + rawsource = open(rawsourcefile, 'rU').read() + rawcode = compile(rawsource, rawsourcefile, 'exec') + writepycfile(rawcode, outputfilename) + outputfilename = os.path.join(destname, 'Contents', 'Resources', pycname) if progress: progress.label('Creating '+pycname) |