home *** CD-ROM | disk | FTP | other *** search
/ PC World 2001 April / PCWorld_2001-04_cd.bin / Software / TemaCD / webclean / !!!python!!! / BeOpen-Python-2.0.exe / FIXHEADER.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  1.0 KB  |  50 lines
  1. #! /usr/bin/env python
  2.  
  3. # Add some standard cpp magic to a header file
  4.  
  5. import sys
  6. import string
  7.  
  8. def main():
  9.     args = sys.argv[1:]
  10.     for file in args:
  11.         process(file)
  12.  
  13. def process(file):
  14.     try:
  15.         f = open(file, 'r')
  16.     except IOError, msg:
  17.         sys.stderr.write('%s: can\'t open: %s\n' % (file, str(msg)))
  18.         return
  19.     data = f.read()
  20.     f.close()
  21.     if data[:2] <> '/*':
  22.         sys.stderr.write('%s does not begin with C comment\n' % file)
  23.         return
  24.     try:
  25.         f = open(file, 'w')
  26.     except IOError, msg:
  27.         sys.stderr.write('%s: can\'t write: %s\n' % (file, str(msg)))
  28.         return
  29.     sys.stderr.write('Processing %s ...\n' % file)
  30.     magic = 'Py_'
  31.     for c in file:
  32.         if c in string.letters + string.digits:
  33.             magic = magic + string.upper(c)
  34.         else: magic = magic + '_'
  35.     sys.stdout = f
  36.     print '#ifndef', magic
  37.     print '#define', magic
  38.     print '#ifdef __cplusplus'
  39.     print 'extern "C" {'
  40.     print '#endif'
  41.     print
  42.     f.write(data)
  43.     print
  44.     print '#ifdef __cplusplus'
  45.     print '}'
  46.     print '#endif'
  47.     print '#endif /*', '!'+magic, '*/'
  48.  
  49. main()
  50.