| import re | |
| import sys | |
| # Write the config.c file | |
| never = ['marshal', '__main__', '__builtin__', 'sys', 'exceptions', '_warnings'] | |
| def makeconfig(infp, outfp, modules, with_ifdef=0): | |
| m1 = re.compile('-- ADDMODULE MARKER 1 --') | |
| m2 = re.compile('-- ADDMODULE MARKER 2 --') | |
| while 1: | |
| line = infp.readline() | |
| if not line: break | |
| outfp.write(line) | |
| if m1 and m1.search(line): | |
| m1 = None | |
| for mod in modules: | |
| if mod in never: | |
| continue | |
| if with_ifdef: | |
| outfp.write("#ifndef init%s\n"%mod) | |
| outfp.write('extern void init%s(void);\n' % mod) | |
| if with_ifdef: | |
| outfp.write("#endif\n") | |
| elif m2 and m2.search(line): | |
| m2 = None | |
| for mod in modules: | |
| if mod in never: | |
| continue | |
| outfp.write('\t{"%s", init%s},\n' % | |
| (mod, mod)) | |
| if m1: | |
| sys.stderr.write('MARKER 1 never found\n') | |
| elif m2: | |
| sys.stderr.write('MARKER 2 never found\n') | |
| # Test program. | |
| def test(): | |
| if not sys.argv[3:]: | |
| print 'usage: python makeconfig.py config.c.in outputfile', | |
| print 'modulename ...' | |
| sys.exit(2) | |
| if sys.argv[1] == '-': | |
| infp = sys.stdin | |
| else: | |
| infp = open(sys.argv[1]) | |
| if sys.argv[2] == '-': | |
| outfp = sys.stdout | |
| else: | |
| outfp = open(sys.argv[2], 'w') | |
| makeconfig(infp, outfp, sys.argv[3:]) | |
| if outfp != sys.stdout: | |
| outfp.close() | |
| if infp != sys.stdin: | |
| infp.close() | |
| if __name__ == '__main__': | |
| test() |