Fix docstring generation for generated packet types

Cirras · Feb 26, 2024 · 9434037 · 9434037
1 parent 9095709
commit 9434037
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/protocol_code_generator/generate/code_generator.py b/protocol_code_generator/generate/code_generator.py
@@ -275,6 +275,10 @@ def _generate_packet(self, protocol_packet):
             .unindent()
         )
 
+        object_code_generator.data.docstring.add_code_block(
+            generate_docstring(get_comment(protocol_packet))
+        )
+
         code_block = object_code_generator.code
         code_block.add_import("Packet", "eolib.protocol.net.packet")
         code_block.add_import("PacketFamily", "eolib.protocol._generated.net.packet_family")