PUBLICAPI-1013: support reset mask annotation (#136)

Author: complynx

src/nebius/api/nebius/maintenance/v1alpha1/__init__.py

@@ -17,6 +17,7 @@
 import nebius.api.nebius.common.v1.metadata_pb2 as metadata_pb2
 import nebius.api.nebius.maintenance.v1alpha1.maintenance_pb2 as maintenance_pb2
 import nebius.api.nebius.maintenance.v1alpha1.maintenance_service_pb2 as maintenance_service_pb2
+import nebius.base.fieldmask_protobuf as fieldmask_protobuf
 import nebius.base.protos.descriptor as descriptor
 import nebius.base.protos.pb_classes as pb_classes
 import nebius.base.protos.pb_enum as pb_enum
@@ -673,6 +674,7 @@ def update(self,
             :class:`nebius.api.nebius.maintenance.v1alpha1.UpdateMaintenanceResponse`.
         """
 
+        kwargs['metadata'] = fieldmask_protobuf.ensure_reset_mask_in_metadata(request, kwargs.get('metadata', None))
         return super().request(
             method="Update",
             request=request,

src/nebius/base/protos/compiler/generators.py

@@ -1,11 +1,13 @@
 from logging import getLogger
+from typing import Any, cast
 
 from nebius.api.nebius import (
     FieldBehavior,
     enum_value_deprecation_details,
     field_deprecation_details,
     message_deprecation_details,
     method_deprecation_details,
+    send_reset_mask,
     service_deprecation_details,
 )
 
@@ -755,6 +757,24 @@ def is_operation_output(method: Method) -> bool:
     )
 
 
+def _send_reset_mask_setting(method: Method) -> bool | None:
+    """
+    Returns True/False when annotation is set, otherwise None.
+    """
+    ext = cast(Any, send_reset_mask)
+    options = method.descriptor.options
+    if not options.HasExtension(ext):
+        return None
+    return bool(options.Extensions[ext])
+
+
+def _should_add_reset_mask(method: Method) -> bool:
+    setting = _send_reset_mask_setting(method)
+    if setting is None:
+        return method.name == "Update"
+    return setting
+
+
 def generate_service(srv: Service, g: PyGenFile) -> None:
     operation_type = None
     operation_source_method = None
@@ -882,7 +902,7 @@ def generate_service(srv: Service, g: PyGenFile) -> None:
                     g.p(", stack_info=True, stacklevel=2)")
                     g.p()
 
-                if method.name == "Update" and is_operation_output(method):
+                if _should_add_reset_mask(method):
                     g.p(
                         "kwargs['metadata'] = ",
                         ImportedSymbol(