Fred Drake | 522af3a | 1999-01-06 16:28:34 +0000 | [diff] [blame] | 1 | """Provide access to Python's configuration information. The specific names |
| 2 | defined in the module depend heavily on the platform and configuration. |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 3 | |
| 4 | Written by: Fred L. Drake, Jr. |
| 5 | Email: <fdrake@acm.org> |
| 6 | Initial date: 17-Dec-1998 |
| 7 | """ |
| 8 | |
| 9 | __version__ = "$Revision$" |
| 10 | |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 11 | import os |
| 12 | import re |
| 13 | import string |
| 14 | import sys |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 15 | |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 16 | |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 17 | def get_config_h_filename(): |
Fred Drake | 522af3a | 1999-01-06 16:28:34 +0000 | [diff] [blame] | 18 | """Return full pathname of installed config.h file.""" |
Fred Drake | 6a1b53c | 1999-01-11 15:34:55 +0000 | [diff] [blame] | 19 | return os.path.join(sys.exec_prefix, "include", "python" + sys.version[:3], |
| 20 | "config.h") |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 21 | |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 22 | def get_makefile_filename(): |
Fred Drake | 522af3a | 1999-01-06 16:28:34 +0000 | [diff] [blame] | 23 | """Return full pathname of installed Makefile from the Python build.""" |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 24 | return os.path.join(sys.exec_prefix, "lib", "python" + sys.version[:3], |
| 25 | "config", "Makefile") |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 26 | |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 27 | def parse_config_h(fp, g=None): |
Fred Drake | 522af3a | 1999-01-06 16:28:34 +0000 | [diff] [blame] | 28 | """Parse a config.h-style file. A dictionary containing name/value |
| 29 | pairs is returned. If an optional dictionary is passed in as the second |
| 30 | argument, it is used instead of a new dictionary. |
| 31 | """ |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 32 | if g is None: |
| 33 | g = {} |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 34 | define_rx = re.compile("#define ([A-Z][A-Z0-9_]+) (.*)\n") |
| 35 | undef_rx = re.compile("/[*] #undef ([A-Z][A-Z0-9_]+) [*]/\n") |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 36 | # |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 37 | while 1: |
| 38 | line = fp.readline() |
| 39 | if not line: |
| 40 | break |
| 41 | m = define_rx.match(line) |
| 42 | if m: |
| 43 | n, v = m.group(1, 2) |
Greg Ward | 3c8e54b | 1998-12-22 12:42:04 +0000 | [diff] [blame] | 44 | try: v = string.atoi(v) |
| 45 | except ValueError: pass |
| 46 | g[n] = v |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 47 | else: |
| 48 | m = undef_rx.match(line) |
| 49 | if m: |
| 50 | g[m.group(1)] = 0 |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 51 | return g |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 52 | |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 53 | def parse_makefile(fp, g=None): |
Fred Drake | 522af3a | 1999-01-06 16:28:34 +0000 | [diff] [blame] | 54 | """Parse a Makefile-style file. A dictionary containing name/value |
| 55 | pairs is returned. If an optional dictionary is passed in as the second |
| 56 | argument, it is used instead of a new dictionary. |
| 57 | """ |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 58 | if g is None: |
| 59 | g = {} |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 60 | variable_rx = re.compile("([a-zA-Z][a-zA-Z0-9_]+)\s*=\s*(.*)\n") |
| 61 | done = {} |
| 62 | notdone = {} |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 63 | # |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 64 | while 1: |
| 65 | line = fp.readline() |
| 66 | if not line: |
| 67 | break |
| 68 | m = variable_rx.match(line) |
| 69 | if m: |
| 70 | n, v = m.group(1, 2) |
Greg Ward | 3c8e54b | 1998-12-22 12:42:04 +0000 | [diff] [blame] | 71 | v = string.strip(v) |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 72 | if "$" in v: |
| 73 | notdone[n] = v |
| 74 | else: |
Greg Ward | 3c8e54b | 1998-12-22 12:42:04 +0000 | [diff] [blame] | 75 | try: v = string.atoi(v) |
| 76 | except ValueError: pass |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 77 | done[n] = v |
| 78 | |
| 79 | # do variable interpolation here |
| 80 | findvar1_rx = re.compile(r"\$\(([A-Za-z][A-Za-z0-9_]*)\)") |
| 81 | findvar2_rx = re.compile(r"\${([A-Za-z][A-Za-z0-9_]*)}") |
| 82 | while notdone: |
| 83 | for name in notdone.keys(): |
| 84 | value = notdone[name] |
| 85 | m = findvar1_rx.search(value) |
| 86 | if not m: |
| 87 | m = findvar2_rx.search(value) |
| 88 | if m: |
| 89 | n = m.group(1) |
| 90 | if done.has_key(n): |
| 91 | after = value[m.end():] |
| 92 | value = value[:m.start()] + done[n] + after |
| 93 | if "$" in after: |
| 94 | notdone[name] = value |
| 95 | else: |
Greg Ward | 3c8e54b | 1998-12-22 12:42:04 +0000 | [diff] [blame] | 96 | try: value = string.atoi(value) |
| 97 | except ValueError: pass |
| 98 | done[name] = string.strip(value) |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 99 | del notdone[name] |
| 100 | elif notdone.has_key(n): |
| 101 | # get it on a subsequent round |
| 102 | pass |
| 103 | else: |
| 104 | done[n] = "" |
| 105 | after = value[m.end():] |
| 106 | value = value[:m.start()] + after |
| 107 | if "$" in after: |
| 108 | notdone[name] = value |
| 109 | else: |
Greg Ward | 3c8e54b | 1998-12-22 12:42:04 +0000 | [diff] [blame] | 110 | try: value = string.atoi(value) |
| 111 | except ValueError: pass |
| 112 | done[name] = string.strip(value) |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 113 | del notdone[name] |
| 114 | else: |
Greg Ward | 3c8e54b | 1998-12-22 12:42:04 +0000 | [diff] [blame] | 115 | # bogus variable reference; just drop it since we can't deal |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 116 | del notdone[name] |
| 117 | |
| 118 | # save the results in the global dictionary |
| 119 | g.update(done) |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 120 | return g |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 121 | |
| 122 | |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 123 | def _init_posix(): |
Fred Drake | 522af3a | 1999-01-06 16:28:34 +0000 | [diff] [blame] | 124 | """Initialize the module as appropriate for POSIX systems.""" |
Greg Ward | 9ddaaa1 | 1999-01-06 14:46:06 +0000 | [diff] [blame] | 125 | g = globals() |
| 126 | # load the installed config.h: |
| 127 | parse_config_h(open(get_config_h_filename()), g) |
| 128 | # load the installed Makefile.pre.in: |
| 129 | parse_makefile(open(get_makefile_filename()), g) |
| 130 | |
| 131 | |
| 132 | |
| 133 | try: |
| 134 | exec "_init_" + os.name |
| 135 | except NameError: |
| 136 | # not needed for this platform |
| 137 | pass |
| 138 | else: |
| 139 | exec "_init_%s()" % os.name |
| 140 | |
Greg Ward | 1190ee3 | 1998-12-18 23:46:33 +0000 | [diff] [blame] | 141 | del _init_posix |