| 1 | # Parse Makefiles and Python Setup(.in) files.
|
|---|
| 2 |
|
|---|
| 3 | import re
|
|---|
| 4 |
|
|---|
| 5 |
|
|---|
| 6 | # Extract variable definitions from a Makefile.
|
|---|
| 7 | # Return a dictionary mapping names to values.
|
|---|
| 8 | # May raise IOError.
|
|---|
| 9 |
|
|---|
| 10 | makevardef = re.compile('^([a-zA-Z0-9_]+)[ \t]*=(.*)')
|
|---|
| 11 |
|
|---|
| 12 | def getmakevars(filename):
|
|---|
| 13 | variables = {}
|
|---|
| 14 | fp = open(filename)
|
|---|
| 15 | pendingline = ""
|
|---|
| 16 | try:
|
|---|
| 17 | while 1:
|
|---|
| 18 | line = fp.readline()
|
|---|
| 19 | if pendingline:
|
|---|
| 20 | line = pendingline + line
|
|---|
| 21 | pendingline = ""
|
|---|
| 22 | if not line:
|
|---|
| 23 | break
|
|---|
| 24 | if line.endswith('\\\n'):
|
|---|
| 25 | pendingline = line[:-2]
|
|---|
| 26 | matchobj = makevardef.match(line)
|
|---|
| 27 | if not matchobj:
|
|---|
| 28 | continue
|
|---|
| 29 | (name, value) = matchobj.group(1, 2)
|
|---|
| 30 | # Strip trailing comment
|
|---|
| 31 | i = value.find('#')
|
|---|
| 32 | if i >= 0:
|
|---|
| 33 | value = value[:i]
|
|---|
| 34 | value = value.strip()
|
|---|
| 35 | variables[name] = value
|
|---|
| 36 | finally:
|
|---|
| 37 | fp.close()
|
|---|
| 38 | return variables
|
|---|
| 39 |
|
|---|
| 40 |
|
|---|
| 41 | # Parse a Python Setup(.in) file.
|
|---|
| 42 | # Return two dictionaries, the first mapping modules to their
|
|---|
| 43 | # definitions, the second mapping variable names to their values.
|
|---|
| 44 | # May raise IOError.
|
|---|
| 45 |
|
|---|
| 46 | setupvardef = re.compile('^([a-zA-Z0-9_]+)=(.*)')
|
|---|
| 47 |
|
|---|
| 48 | def getsetupinfo(filename):
|
|---|
| 49 | modules = {}
|
|---|
| 50 | variables = {}
|
|---|
| 51 | fp = open(filename)
|
|---|
| 52 | pendingline = ""
|
|---|
| 53 | try:
|
|---|
| 54 | while 1:
|
|---|
| 55 | line = fp.readline()
|
|---|
| 56 | if pendingline:
|
|---|
| 57 | line = pendingline + line
|
|---|
| 58 | pendingline = ""
|
|---|
| 59 | if not line:
|
|---|
| 60 | break
|
|---|
| 61 | # Strip comments
|
|---|
| 62 | i = line.find('#')
|
|---|
| 63 | if i >= 0:
|
|---|
| 64 | line = line[:i]
|
|---|
| 65 | if line.endswith('\\\n'):
|
|---|
| 66 | pendingline = line[:-2]
|
|---|
| 67 | continue
|
|---|
| 68 | matchobj = setupvardef.match(line)
|
|---|
| 69 | if matchobj:
|
|---|
| 70 | (name, value) = matchobj.group(1, 2)
|
|---|
| 71 | variables[name] = value.strip()
|
|---|
| 72 | else:
|
|---|
| 73 | words = line.split()
|
|---|
| 74 | if words:
|
|---|
| 75 | modules[words[0]] = words[1:]
|
|---|
| 76 | finally:
|
|---|
| 77 | fp.close()
|
|---|
| 78 | return modules, variables
|
|---|
| 79 |
|
|---|
| 80 |
|
|---|
| 81 | # Test the above functions.
|
|---|
| 82 |
|
|---|
| 83 | def test():
|
|---|
| 84 | import sys
|
|---|
| 85 | import os
|
|---|
| 86 | if not sys.argv[1:]:
|
|---|
| 87 | print 'usage: python parsesetup.py Makefile*|Setup* ...'
|
|---|
| 88 | sys.exit(2)
|
|---|
| 89 | for arg in sys.argv[1:]:
|
|---|
| 90 | base = os.path.basename(arg)
|
|---|
| 91 | if base[:8] == 'Makefile':
|
|---|
| 92 | print 'Make style parsing:', arg
|
|---|
| 93 | v = getmakevars(arg)
|
|---|
| 94 | prdict(v)
|
|---|
| 95 | elif base[:5] == 'Setup':
|
|---|
| 96 | print 'Setup style parsing:', arg
|
|---|
| 97 | m, v = getsetupinfo(arg)
|
|---|
| 98 | prdict(m)
|
|---|
| 99 | prdict(v)
|
|---|
| 100 | else:
|
|---|
| 101 | print arg, 'is neither a Makefile nor a Setup file'
|
|---|
| 102 | print '(name must begin with "Makefile" or "Setup")'
|
|---|
| 103 |
|
|---|
| 104 | def prdict(d):
|
|---|
| 105 | keys = d.keys()
|
|---|
| 106 | keys.sort()
|
|---|
| 107 | for key in keys:
|
|---|
| 108 | value = d[key]
|
|---|
| 109 | print "%-15s" % key, str(value)
|
|---|
| 110 |
|
|---|
| 111 | if __name__ == '__main__':
|
|---|
| 112 | test()
|
|---|
