fhdl/module: avoid flushing self._submodules and create do_exit.

enjoy-digital · enjoy-digital · commit ee1091f4910b · 2015-03-09T17:17:21.000+01:00
diff --git a/migen/fhdl/module.py b/migen/fhdl/module.py
@@ -142,8 +142,10 @@ def __setattr__(self, name, value):
 			object.__setattr__(self, name, value)
 
 	def _collect_submodules(self):
-		r = [(name, submodule.get_fragment()) for name, submodule in self._submodules]
-		self._submodules = []
+		r = []
+		for name, submodule in self._submodules:
+			if not submodule._get_fragment_called:
+				r += [(name, submodule.get_fragment())]
 		return r
 
 	def finalize(self, *args, **kwargs):
@@ -176,3 +178,7 @@ def finalize(self, *args, **kwargs):
 
 	def do_finalize(self):
 		pass
+
+	def do_exit(self, *args, **kwargs):
+		for name, submodule in self._submodules:
+			submodule.do_exit(*args, **kwargs)
