From 80fefca2e9a1faed733d14f27b70f831f990d9e7 Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Wed, 13 Sep 2023 19:00:10 +0200
Subject: [PATCH] Simplify copying generated Client signatures

---
 tools/copy_client_signatures.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/tools/copy_client_signatures.py b/tools/copy_client_signatures.py
index 44f46d7f..0c95cdbd 100644
--- a/tools/copy_client_signatures.py
+++ b/tools/copy_client_signatures.py
@@ -79,6 +79,8 @@ def main() -> None:
 
     m_visitor.visit(ast.parse(contents))
 
+    class_body = []
+
     for function in sorted(fm_visitor.methods, key=lambda f: f.name):
         function.body = []
         if doc := m_visitor.method_docs.get(function.name):
@@ -105,8 +107,15 @@ def main() -> None:
                 call = ast.Return(value=call)
 
         function.body.append(call)
+        class_body.append(function)
 
-        print(ast.unparse(function))
+    print(
+        ast.unparse(
+            ast.ClassDef(
+                name="Client", bases=[], keywords=[], body=class_body, decorator_list=[]
+            )
+        )
+    )
 
 
 if __name__ == "__main__":