bgengeneratorgroup.py

来自「Python语言编译器」· Python 代码 · 共 36 行

from bgenOutput import *

class GeneratorGroup:

	def __init__(self, prefix):
		self.prefix = prefix
		self.generators = []

	def add(self, g):
		g.setprefix(self.prefix)
		self.generators.append(g)

	def generate(self):
		for g in self.generators:
			g.generate()
		Output()
		Output("static PyMethodDef %s_methods[] = {", self.prefix)
		IndentLevel()
		for g in self.generators:
			g.reference()
		Output("{NULL, NULL, 0}")
		DedentLevel()
		Output("};")


def _test():
	from bgenGenerator import Generator
	group = GeneratorGroup("spam")
	eggs = Generator(void, "eggs")
	group.add(eggs)
	print "/* START */"
	group.generate()

if __name__ == "__main__":
	_test()