oneof support

betterproto-extras/src/descriptor_pool.rs

@@ -4,7 +4,8 @@ use crate::{
 };
 use prost_reflect::{
     prost_types::{
-        field_descriptor_proto::{Type, Label}, DescriptorProto, FieldDescriptorProto, FileDescriptorProto,
+        field_descriptor_proto::{Label, Type},
+        DescriptorProto, FieldDescriptorProto, FileDescriptorProto, OneofDescriptorProto,
     },
     DescriptorPool, MessageDescriptor,
 };
@@ -69,6 +70,27 @@ fn create_cached_descriptor_in_pool<'py>(
                 field.set_label(Label::Repeated);
             }
 
+            if let Some(grp) = meta.oneof_group(field_name)? {
+                let oneof_index = message
+                    .oneof_decl
+                    .iter()
+                    .enumerate()
+                    .filter(|x| x.1.name() == grp)
+                    .map(|x| x.0)
+                    .last();
+
+                match oneof_index {
+                    Some(i) => field.oneof_index = Some(i as i32),
+                    None => {
+                        message.oneof_decl.push(OneofDescriptorProto {
+                            name: Some(grp),
+                            ..Default::default()
+                        });
+                        field.oneof_index = Some((message.oneof_decl.len() - 1) as i32)
+                    }
+                }
+            }
+
             field
         });
     }

betterproto-extras/src/merging.rs

@@ -9,11 +9,13 @@ pub fn merge_msg_into_pyobj(obj: &PyAny, msg: &DynamicMessage) -> Result<()> {
     for field in msg.descriptor().fields() {
         let field_name = field.name();
         let proto_meta = obj.get_proto_meta()?;
-        if let Some(field_value) = msg.get_field_by_name(field_name) {
-            obj.setattr(
-                field_name,
-                map_field_value(field_name, &field_value, proto_meta)?,
-            )?;
+        if msg.has_field_by_name(field_name) {
+            if let Some(field_value) = msg.get_field_by_name(field_name) {
+                obj.setattr(
+                    field_name,
+                    map_field_value(field_name, &field_value, proto_meta)?,
+                )?;
+            }
         }
     }
     Ok(())

betterproto-extras/src/py_any_extras.rs

@@ -1,14 +1,15 @@
 use crate::error::Result;
 use pyo3::PyAny;
 
-pub trait PyAnyExtras<'py> {
+pub trait PyAnyExtras {
     fn qualified_class_name(&self) -> Result<String>;
-    fn get_proto_meta(&self) -> Result<&'py PyAny>;
-    fn create_instance(&self, field_name: &str) -> Result<&'py PyAny>;
+    fn get_proto_meta(&self) -> Result<&PyAny>;
+    fn create_instance(&self, field_name: &str) -> Result<&PyAny>;
     fn is_list_field(&self, field_name: &str) -> Result<bool>;
+    fn oneof_group(&self, field_name: &str) -> Result<Option<String>>;
 }
 
-impl<'py> PyAnyExtras<'py> for &'py PyAny {
+impl PyAnyExtras for PyAny {
     fn qualified_class_name(&self) -> Result<String> {
         let class = self.getattr("__class__")?;
         let module = class.getattr("__module__")?;
@@ -16,11 +17,11 @@ impl<'py> PyAnyExtras<'py> for &'py PyAny {
         Ok(format!("{module}.{name}"))
     }
 
-    fn get_proto_meta(&self) -> Result<&'py PyAny> {
+    fn get_proto_meta(&self) -> Result<&PyAny> {
         Ok(self.getattr("_betterproto")?)
     }
 
-    fn create_instance(&self, field_name: &str) -> Result<&'py PyAny> {
+    fn create_instance(&self, field_name: &str) -> Result<&PyAny> {
         let res = self
             .getattr("cls_by_field")?
             .get_item(field_name)?
@@ -34,4 +35,12 @@ impl<'py> PyAnyExtras<'py> for &'py PyAny {
         let name = cls.getattr("__name__")?;
         Ok(module.to_string() == "builtins" && name.to_string() == "list")
     }
+
+    fn oneof_group(&self, field_name: &str) -> Result<Option<String>> {
+        let opt = self
+            .getattr("oneof_group_by_field")?
+            .call_method1("get", (field_name,))?
+            .extract()?;
+        Ok(opt)
+    }
 }