fixapplepython23.py revision 9cf41d0c9d48e6f7df971ebc596aa4cf269b70bb 
1#!/usr/bin/env python
2"""fixapplepython23 - Fix Apple-installed Python 2.3 (on Mac OS X 10.3)
3
4Python 2.3 (and 2.3.X for X<5) have the problem that building an extension
5for a framework installation may accidentally pick up the framework
6of a newer Python, in stead of the one that was used to build the extension.
7
8This script modifies the Makefile (in .../lib/python2.3/config) to use
9the newer method of linking extensions with "-undefined dynamic_lookup"
10which fixes this problem.
11
12The script will first check all prerequisites, and return a zero exit
13status also when nothing needs to be fixed.
14"""
15import sys
16import os
17import gestalt
18
19MAKEFILE='/System/Library/Frameworks/Python.framework/Versions/2.3/lib/python2.3/config/Makefile'
20CHANGES=((
21    'LDSHARED=\t$(CC) $(LDFLAGS) -bundle -framework $(PYTHONFRAMEWORK)\n',
22    'LDSHARED=\t$(CC) $(LDFLAGS) -bundle -undefined dynamic_lookup\n'
23    ),(
24    'BLDSHARED=\t$(CC) $(LDFLAGS) -bundle -framework $(PYTHONFRAMEWORK)\n',
25    'BLDSHARED=\t$(CC) $(LDFLAGS) -bundle -undefined dynamic_lookup\n'
26    ),(
27    'CC=\t\tgcc\n',
28    'CC=\t\t/System/Library/Frameworks/Python.framework/Versions/2.3/lib/python2.3/config/PantherPythonFix/run-gcc\n'
29    ),(
30    'CXX=\t\tc++\n',
31    'CXX=\t\t/System/Library/Frameworks/Python.framework/Versions/2.3/lib/python2.3/config/PantherPythonFix/run-g++\n'
32))
33
34GCC_SCRIPT='/System/Library/Frameworks/Python.framework/Versions/2.3/lib/python2.3/config/PantherPythonFix/run-gcc'
35GXX_SCRIPT='/System/Library/Frameworks/Python.framework/Versions/2.3/lib/python2.3/config/PantherPythonFix/run-g++'
36SCRIPT="""#!/bin/sh
37export MACOSX_DEPLOYMENT_TARGET=10.3
38exec %s "${@}"
39"""
40
41def findline(lines, start):
42    """return line starting with given string or -1"""
43    for i in range(len(lines)):
44        if lines[i][:len(start)] == start:
45            return i
46    return -1
47
48def fix(makefile, do_apply):
49    """Fix the Makefile, if required."""
50    fixed = False
51    lines = open(makefile).readlines()
52
53    for old, new in CHANGES:
54        i = findline(lines, new)
55        if i >= 0:
56            # Already fixed
57            continue
58        i = findline(lines, old)
59        if i < 0:
60            print 'fixapplepython23: Python installation not fixed (appears broken)'
61            print 'fixapplepython23: missing line:', old
62            return 2
63        lines[i] = new
64        fixed = True
65
66    if fixed:
67        if do_apply:
68            print 'fixapplepython23: Fix to Apple-installed Python 2.3 applied'
69            os.rename(makefile, makefile + '~')
70            open(makefile, 'w').writelines(lines)
71            return 0
72        else:
73            print 'fixapplepython23: Fix to Apple-installed Python 2.3 should be applied'
74            return 1
75    else:
76        print 'fixapplepython23: No fix needed, appears to have been applied before'
77        return 0
78
79def makescript(filename, compiler):
80    """Create a wrapper script for a compiler"""
81    dirname = os.path.split(filename)[0]
82    if not os.access(dirname, os.X_OK):
83        os.mkdir(dirname, 0755)
84    fp = open(filename, 'w')
85    fp.write(SCRIPT % compiler)
86    fp.close()
87    os.chmod(filename, 0755)
88    print 'fixapplepython23: Created', filename
89
90def main():
91    # Check for -n option
92    if len(sys.argv) > 1 and sys.argv[1] == '-n':
93        do_apply = False
94    else:
95        do_apply = True
96    # First check OS version
97    if sys.byteorder == 'little':
98        # All intel macs are fine
99        print "fixapplypython23: no fix is needed on MacOSX on Intel"
100        sys.exit(0)
101
102    if gestalt.gestalt('sysv') < 0x1030:
103        print 'fixapplepython23: no fix needed on MacOSX < 10.3'
104        sys.exit(0)
105
106    if gestalt.gestalt('sysv') >= 0x1040:
107        print 'fixapplepython23: no fix needed on MacOSX >= 10.4'
108        sys.exit(0)
109
110    # Test that a framework Python is indeed installed
111    if not os.path.exists(MAKEFILE):
112        print 'fixapplepython23: Python framework does not appear to be installed (?), nothing fixed'
113        sys.exit(0)
114    # Check that we can actually write the file
115    if do_apply and not os.access(MAKEFILE, os.W_OK):
116        print 'fixapplepython23: No write permission, please run with "sudo"'
117        sys.exit(2)
118    # Create the shell scripts
119    if do_apply:
120        if not os.access(GCC_SCRIPT, os.X_OK):
121            makescript(GCC_SCRIPT, "gcc")
122        if not os.access(GXX_SCRIPT, os.X_OK):
123            makescript(GXX_SCRIPT, "g++")
124    #  Finally fix the makefile
125    rv = fix(MAKEFILE, do_apply)
126    #sys.exit(rv)
127    sys.exit(0)
128
129if __name__ == '__main__':
130    main()
131