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()