Fix bug

AdrienVannson · AdrienVannson · commit 9a23d3f16791 · 2025-01-07T15:50:24.000+01:00
diff --git a/src/betterproto2_compiler/plugin/models.py b/src/betterproto2_compiler/plugin/models.py
@@ -477,7 +477,7 @@ def get_field_string(self) -> str:
             f"betterproto2.{self.proto_v_type}))"
         )
         if self.py_name in dir(builtins):
-            self.parent.builtins_types.add(self.py_name)
+            self.message.builtins_types.add(self.py_name)
         return f'{self.py_name}: "{self.annotation}" = {betterproto_field_type}'
 
     @property

Original file line number	Diff line number	Diff line change
`@@ -477,7 +477,7 @@ def get_field_string(self) -> str:`
`477`	`477`	`f"betterproto2.{self.proto_v_type}))"`
`478`	`478`	`)`
`479`	`479`	`if self.py_name in dir(builtins):`
`480`		`- self.parent.builtins_types.add(self.py_name)`
	`480`	`+ self.message.builtins_types.add(self.py_name)`
`481`	`481`	`return f'{self.py_name}: "{self.annotation}" = {betterproto_field_type}'`
`482`	`482`
`483`	`483`	`@property`