set_server_user_data

Author: Laure-di

scaleway/scaleway/instance/custom_api.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Dict, Any, Optional
+from typing import Dict, Any, Optional, BinaryIO
 
 from scaleway_core.bridge import Zone
 from scaleway_core.profile import ProfileDefaults
@@ -71,9 +71,57 @@ def get_server_user_data(self, server_id: str, key: str, zone: Optional[Zone] =
 
 @dataclass
 class SetServerUserDataRequest:
+    zone: Optional[Zone]
+    """
+    Zone of the user data to set
+    """
+
+    server_id: str
 
+    key: str
+    """
+    Key defines the user data key to set
+    """
 
-def set_server_user_data():
+    content: BinaryIO
+    """
+    Content defines the data to set
+    """
+
+def marshal_SetServerUserDataRequest(request: SetServerUserDataRequest, defaults: ProfileDefaults) -> Dict[str, Any]:
+    output: Dict[str, Any] = {}
+
+    if request.server_id is not None:
+        output["server_id"] = request.server_id
+    if request.key is not None:
+        output["key"] = request.key
+    if request.zone is not None:
+        output["zone"] = request.zone
+    if request.content is not None:
+        output["content"] = request.content
+
+    return output
+
+def set_server_user_data(self, server_id: str, key: str, content: BinaryIO, zone: Optional[Zone] = None):
+    param_zone = validate_path_param("zone", zone or self.client.default_zone)
+    param_server_id = validate_path_param("server_id", server_id)
+
+    res = self._request(
+        "PATCH",
+        f"/instance/v1/zones/{param_zone}/servers/{param_server_id}/user_data/{key}",
+        body=marshal_SetServerUserDataRequest(
+            SetServerUserDataRequest(
+                zone= zone,
+                server_id= server_id,
+                key=key,
+                content=content,
+            ),
+            self.client,
+        ),
+    )
+
+    self._throw_on_error(res)
+    return res.json()