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