Jack Jansen | ba1c13d | 2002-08-02 14:57:43 +0000 | [diff] [blame] | 1 | # Emulate sys.argv and run __main__.py or __main__.pyc in an environment that |
| 2 | # is as close to "normal" as possible. |
| 3 | # |
| 4 | # This script is put into __rawmain__.pyc for applets that need argv |
| 5 | # emulation, by BuildApplet and friends. |
| 6 | # |
Benjamin Peterson | 2368193 | 2008-05-12 21:42:13 +0000 | [diff] [blame] | 7 | |
| 8 | from warnings import warnpy3k |
Benjamin Peterson | a6864e0 | 2008-07-14 17:42:17 +0000 | [diff] [blame] | 9 | warnpy3k("In 3.x, the appletrawmain module is removed.", stacklevel=2) |
Benjamin Peterson | 2368193 | 2008-05-12 21:42:13 +0000 | [diff] [blame] | 10 | |
Jack Jansen | ba1c13d | 2002-08-02 14:57:43 +0000 | [diff] [blame] | 11 | import argvemulator |
| 12 | import os |
| 13 | import sys |
| 14 | import marshal |
| 15 | |
| 16 | # |
Jack Jansen | 3e6c80b | 2002-08-02 15:31:25 +0000 | [diff] [blame] | 17 | # Make sure we have an argv[0], and make _dir point to the Resources |
| 18 | # directory. |
| 19 | # |
| 20 | if not sys.argv or sys.argv[0][:1] == '-': |
Jack Jansen | 0ae3220 | 2003-04-09 13:25:43 +0000 | [diff] [blame] | 21 | # Insert our (guessed) name. |
| 22 | _dir = os.path.split(sys.executable)[0] # removes "python" |
| 23 | _dir = os.path.split(_dir)[0] # Removes "MacOS" |
| 24 | _dir = os.path.join(_dir, 'Resources') |
| 25 | sys.argv.insert(0, '__rawmain__') |
Jack Jansen | 3e6c80b | 2002-08-02 15:31:25 +0000 | [diff] [blame] | 26 | else: |
Jack Jansen | 0ae3220 | 2003-04-09 13:25:43 +0000 | [diff] [blame] | 27 | _dir = os.path.split(sys.argv[0])[0] |
Jack Jansen | 3e6c80b | 2002-08-02 15:31:25 +0000 | [diff] [blame] | 28 | # |
Jack Jansen | e87ed57 | 2002-09-12 21:58:47 +0000 | [diff] [blame] | 29 | # Add the Resources directory to the path. This is where files installed |
Tim Peters | 182b5ac | 2004-07-18 06:16:08 +0000 | [diff] [blame] | 30 | # by BuildApplet.py with the --extra option show up, and if those files are |
Jack Jansen | e87ed57 | 2002-09-12 21:58:47 +0000 | [diff] [blame] | 31 | # modules this sys.path modification is necessary to be able to import them. |
| 32 | # |
| 33 | sys.path.insert(0, _dir) |
| 34 | # |
Jack Jansen | ba1c13d | 2002-08-02 14:57:43 +0000 | [diff] [blame] | 35 | # Create sys.argv |
| 36 | # |
| 37 | argvemulator.ArgvCollector().mainloop() |
| 38 | # |
Jack Jansen | 3e6c80b | 2002-08-02 15:31:25 +0000 | [diff] [blame] | 39 | # Find the real main program to run |
Jack Jansen | ba1c13d | 2002-08-02 14:57:43 +0000 | [diff] [blame] | 40 | # |
Jack Jansen | ba1c13d | 2002-08-02 14:57:43 +0000 | [diff] [blame] | 41 | __file__ = os.path.join(_dir, '__main__.py') |
| 42 | if os.path.exists(__file__): |
Jack Jansen | 0ae3220 | 2003-04-09 13:25:43 +0000 | [diff] [blame] | 43 | # |
| 44 | # Setup something resembling a normal environment and go. |
| 45 | # |
| 46 | sys.argv[0] = __file__ |
| 47 | del argvemulator, os, sys, _dir |
| 48 | execfile(__file__) |
Jack Jansen | ba1c13d | 2002-08-02 14:57:43 +0000 | [diff] [blame] | 49 | else: |
Jack Jansen | 0ae3220 | 2003-04-09 13:25:43 +0000 | [diff] [blame] | 50 | __file__ = os.path.join(_dir, '__main__.pyc') |
| 51 | if os.path.exists(__file__): |
| 52 | # |
| 53 | # If we have only a .pyc file we read the code object from that |
| 54 | # |
| 55 | sys.argv[0] = __file__ |
| 56 | _fp = open(__file__, 'rb') |
| 57 | _fp.read(8) |
| 58 | __code__ = marshal.load(_fp) |
| 59 | # |
| 60 | # Again, we create an almost-normal environment (only __code__ is |
| 61 | # funny) and go. |
| 62 | # |
| 63 | del argvemulator, os, sys, marshal, _dir, _fp |
| 64 | exec __code__ |
| 65 | else: |
| 66 | sys.stderr.write("%s: neither __main__.py nor __main__.pyc found\n"%sys.argv[0]) |
| 67 | sys.exit(1) |