home *** CD-ROM | disk | FTP | other *** search
/ AmigActive 15 / AACD15.ISO / AACD / Programming / Python2 / Python20_source / Tools / freeze / makeconfig.py < prev    next >
Encoding:
Python Source  |  2000-10-25  |  1.3 KB  |  62 lines
  1. import regex
  2.  
  3.  
  4. # Write the config.c file
  5.  
  6. never = ['marshal', '__main__', '__builtin__', 'sys', 'exceptions']
  7.  
  8. def makeconfig(infp, outfp, modules, with_ifdef=0):
  9.     m1 = regex.compile('-- ADDMODULE MARKER 1 --')
  10.     m2 = regex.compile('-- ADDMODULE MARKER 2 --')
  11.     while 1:
  12.         line = infp.readline()
  13.         if not line: break
  14.         outfp.write(line)
  15.         if m1 and m1.search(line) >= 0:
  16.             m1 = None
  17.             for mod in modules:
  18.                 if mod in never:
  19.                     continue
  20.                 if with_ifdef:
  21.                     outfp.write("#ifndef init%s\n"%mod)
  22.                 outfp.write('extern void init%s();\n' % mod)
  23.                 if with_ifdef:
  24.                     outfp.write("#endif\n")
  25.         elif m2 and m2.search(line) >= 0:
  26.             m2 = None
  27.             for mod in modules:
  28.                 if mod in never:
  29.                     continue
  30.                 outfp.write('\t{"%s", init%s},\n' %
  31.                         (mod, mod))
  32.     if m1:
  33.         sys.stderr.write('MARKER 1 never found\n')
  34.     elif m2:
  35.         sys.stderr.write('MARKER 2 never found\n')
  36.  
  37.  
  38. # Test program.
  39.  
  40. def test():
  41.     import sys
  42.     if not sys.argv[3:]:
  43.         print 'usage: python makeconfig.py config.c.in outputfile',
  44.         print 'modulename ...'
  45.         sys.exit(2)
  46.     if sys.argv[1] == '-':
  47.         infp = sys.stdin
  48.     else:
  49.         infp = open(sys.argv[1])
  50.     if sys.argv[2] == '-':
  51.         outfp = sys.stdout
  52.     else:
  53.         outfp = open(sys.argv[2], 'w')
  54.     makeconfig(infp, outfp, sys.argv[3:])
  55.     if outfp != sys.stdout:
  56.         outfp.close()
  57.     if infp != sys.stdin:
  58.         infp.close()
  59.  
  60. if __name__ == '__main__':
  61.     test()
  62.