Remove __post_init__ from MessageCompiler

Author: AdrienVannson

src/betterproto2_compiler/plugin/models.py

@@ -269,15 +269,6 @@ class MessageCompiler(ProtoContentBase):
     oneofs: list["OneofCompiler"] = field(default_factory=list)
     builtins_types: set[str] = field(default_factory=set)
 
-    def __post_init__(self) -> None:
-        # Add message to output file
-        if isinstance(self.parent, OutputTemplate):
-            if isinstance(self, EnumDefinitionCompiler):
-                self.output_file.enums[self.proto_name] = self
-            else:
-                self.output_file.messages[self.proto_name] = self
-        super().__post_init__()
-
     @property
     def proto_name(self) -> str:
         return self.proto_obj.name

src/betterproto2_compiler/plugin/parser.py

@@ -192,6 +192,8 @@ def read_protobuf_type(
             path=path,
             typing_compiler=output_package.typing_compiler,
         )
+        output_package.messages[message_data.proto_name] = message_data
+
         for index, field in enumerate(item.field):
             if is_map(field, item):
                 MapEntryCompiler(
@@ -235,13 +237,14 @@ def read_protobuf_type(
 
     elif isinstance(item, EnumDescriptorProto):
         # Enum
-        EnumDefinitionCompiler(
+        enum = EnumDefinitionCompiler(
             source_file=source_file,
             parent=output_package,
             proto_obj=item,
             path=path,
             typing_compiler=output_package.typing_compiler,
         )
+        output_package.enums[enum.proto_name] = enum
 
 
 def read_protobuf_service(