1  import re
       2  import sys
       3  
       4  # Write the config.c file
       5  
       6  never = ['marshal', '_imp', '_ast', '__main__', 'builtins',
       7           'sys', 'gc', '_warnings']
       8  
       9  def makeconfig(infp, outfp, modules, with_ifdef=0):
      10      m1 = re.compile('-- ADDMODULE MARKER 1 --')
      11      m2 = re.compile('-- ADDMODULE MARKER 2 --')
      12      for line in infp:
      13          outfp.write(line)
      14          if m1 and m1.search(line):
      15              m1 = None
      16              for mod in modules:
      17                  if mod in never:
      18                      continue
      19                  if with_ifdef:
      20                      outfp.write("#ifndef PyInit_%s\n"%mod)
      21                  outfp.write('extern PyObject* PyInit_%s(void);\n' % mod)
      22                  if with_ifdef:
      23                      outfp.write("#endif\n")
      24          elif m2 and m2.search(line):
      25              m2 = None
      26              for mod in modules:
      27                  if mod in never:
      28                      continue
      29                  outfp.write('\t{"%s", PyInit_%s},\n' %
      30                              (mod, mod))
      31      if m1:
      32          sys.stderr.write('MARKER 1 never found\n')
      33      elif m2:
      34          sys.stderr.write('MARKER 2 never found\n')
      35  
      36  
      37  # Test program.
      38  
      39  def test():
      40      if not sys.argv[3:]:
      41          print('usage: python makeconfig.py config.c.in outputfile', end=' ')
      42          print('modulename ...')
      43          sys.exit(2)
      44      if sys.argv[1] == '-':
      45          infp = sys.stdin
      46      else:
      47          infp = open(sys.argv[1])
      48      if sys.argv[2] == '-':
      49          outfp = sys.stdout
      50      else:
      51          outfp = open(sys.argv[2], 'w')
      52      makeconfig(infp, outfp, sys.argv[3:])
      53      if outfp != sys.stdout:
      54          outfp.close()
      55      if infp != sys.stdin:
      56          infp.close()
      57  
      58  if __name__ == '__main__':
      59      test()