From d921d27c1bac0765370a9b9b891f9f0429f4c7c3 Mon Sep 17 00:00:00 2001 From: Markus Armbruster Date: Mon, 1 Feb 2021 14:37:43 -0500 Subject: [PATCH] qapi/gen: Support switching to another module temporarily Signed-off-by: Markus Armbruster Signed-off-by: John Snow Message-Id: <20210201193747.2169670-13-jsnow@redhat.com> [Commit message tweaked] --- scripts/qapi/gen.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/scripts/qapi/gen.py b/scripts/qapi/gen.py index b2bb9d12ff..a0a5df333e 100644 --- a/scripts/qapi/gen.py +++ b/scripts/qapi/gen.py @@ -290,6 +290,13 @@ class QAPISchemaModularCVisitor(QAPISchemaVisitor): 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: