From: Markus Armbruster <arm...@redhat.com> Signed-off-by: Markus Armbruster <arm...@redhat.com> Signed-off-by: John Snow <js...@redhat.com> --- scripts/qapi/gen.py | 7 +++++++ 1 file changed, 7 insertions(+)
diff --git a/scripts/qapi/gen.py b/scripts/qapi/gen.py index b2bb9d12ffc..a0a5df333e0 100644 --- a/scripts/qapi/gen.py +++ b/scripts/qapi/gen.py @@ -290,6 +290,13 @@ def _add_module(self, name: str, blurb: str) -> None: self._module[name] = (genc, genh) self._current_module = name + @contextmanager + def _temp_module(self, name: str) -> Iterator[None]: + old_module = self._current_module + self._current_module = name + yield + self._current_module = old_module + def write(self, output_dir: str, opt_builtins: bool = False) -> None: for name in self._module: if QAPISchemaModule.is_builtin_module(name) and not opt_builtins: -- 2.29.2