imprv: extract class' docstring

pygobject · May 1, 2023 · 9afbf64 · 9afbf64
1 parent 5cba419
commit 9afbf64
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tools/generate.py b/tools/generate.py
@@ -19,6 +19,7 @@
 import itertools
 import pprint
 import re
+import textwrap
 from types import ModuleType
 
 import gi
@@ -703,6 +704,16 @@ def _gi_build_stub(
         else:
             ret += f"class {name}{string_parents}:\n"
 
+            # extracting docs
+            doc = getattr(obj, "__doc__", "") or ""
+            gdoc = getattr(obj, "__gdoc__", "") or ""
+
+            txt = doc + "\n" + gdoc
+            if txt:
+                txt = '"""' + txt + '"""' + "\n"
+                txt = textwrap.indent(txt, "    ")
+                ret += txt
+
         props_override = _check_override(full_name, "Props", overrides)
         if props_override:
             for line in props_override.splitlines():