1 #! /usr/bin/env python

     2 

     3 # Add some standard cpp magic to a header file

     4 

     5 import sys

     6 

     7 def main():

     8     args = sys.argv[1:]

     9     for filename in args:

    10         process(filename)

    11 

    12 def process(filename):

    13     try:

    14         f = open(filename, 'r')

    15     except IOError, msg:

    16         sys.stderr.write('%s: can\'t open: %s\n' % (filename, str(msg)))

    17         return

    18     data = f.read()

    19     f.close()

    20     if data[:2] <> '/*':

    21         sys.stderr.write('%s does not begin with C comment\n' % filename)

    22         return

    23     try:

    24         f = open(filename, 'w')

    25     except IOError, msg:

    26         sys.stderr.write('%s: can\'t write: %s\n' % (filename, str(msg)))

    27         return

    28     sys.stderr.write('Processing %s ...\n' % filename)

    29     magic = 'Py_'

    30     for c in filename:

    31         if ord(c)<=0x80 and c.isalnum():

    32             magic = magic + c.upper()

    33         else: magic = magic + '_'

    34     sys.stdout = f

    35     print '#ifndef', magic

    36     print '#define', magic

    37     print '#ifdef __cplusplus'

    38     print 'extern "C" {'

    39     print '#endif'

    40     print

    41     f.write(data)

    42     print

    43     print '#ifdef __cplusplus'

    44     print '}'

    45     print '#endif'

    46     print '#endif /*', '!'+magic, '*/'

    47 

    48 if __name__ == '__main__':

    49     main()