feat(datawarehouse): add shard_count params (scaleway#1570)

Author: scaleway-bot

scaleway-async/scaleway_async/datawarehouse/v1beta1/api.py

@@ -391,6 +391,7 @@ async def create_deployment(
         region: Optional[ScwRegion] = None,
         project_id: Optional[str] = None,
         tags: Optional[list[str]] = None,
+        shard_count: Optional[int] = None,
         endpoints: Optional[list[EndpointSpec]] = None,
     ) -> Deployment:
         """
@@ -406,6 +407,7 @@ async def create_deployment(
         :param region: Region to target. If none is passed will use default region from the config.
         :param project_id: The Project ID on which the deployment will be created.
         :param tags: Tags to apply to the deployment.
+        :param shard_count: Number of shard for the deployment.
         :param endpoints: Endpoints to associate with the deployment.
         :return: :class:`Deployment <Deployment>`
 
@@ -442,6 +444,7 @@ async def create_deployment(
                     region=region,
                     project_id=project_id,
                     tags=tags,
+                    shard_count=shard_count,
                     endpoints=endpoints,
                 ),
                 self.client,

scaleway-async/scaleway_async/datawarehouse/v1beta1/marshalling.py

@@ -198,12 +198,6 @@ def unmarshal_Deployment(data: Any) -> Deployment:
     else:
         args["tags"] = []
 
-    field = data.get("version", None)
-    if field is not None:
-        args["version"] = field
-    else:
-        args["version"] = None
-
     field = data.get("created_at", None)
     if field is not None:
         args["created_at"] = parser.isoparse(field) if isinstance(field, str) else field
@@ -216,12 +210,24 @@ def unmarshal_Deployment(data: Any) -> Deployment:
     else:
         args["updated_at"] = None
 
+    field = data.get("version", None)
+    if field is not None:
+        args["version"] = field
+    else:
+        args["version"] = None
+
     field = data.get("replica_count", None)
     if field is not None:
         args["replica_count"] = field
     else:
         args["replica_count"] = 0
 
+    field = data.get("shard_count", None)
+    if field is not None:
+        args["shard_count"] = field
+    else:
+        args["shard_count"] = 0
+
     field = data.get("cpu_min", None)
     if field is not None:
         args["cpu_min"] = field
@@ -374,6 +380,12 @@ def unmarshal_Preset(data: Any) -> Preset:
     else:
         args["replica_count"] = 0
 
+    field = data.get("shard_count", None)
+    if field is not None:
+        args["shard_count"] = field
+    else:
+        args["shard_count"] = 0
+
     return Preset(**args)
 
 
@@ -570,6 +582,9 @@ def marshal_CreateDeploymentRequest(
     if request.tags is not None:
         output["tags"] = request.tags
 
+    if request.shard_count is not None:
+        output["shard_count"] = request.shard_count
+
     if request.endpoints is not None:
         output["endpoints"] = [
             marshal_EndpointSpec(item, defaults) for item in request.endpoints

scaleway-async/scaleway_async/datawarehouse/v1beta1/types.py

@@ -185,6 +185,11 @@ class Deployment:
     Number of replicas for the deployment.
     """
 
+    shard_count: int
+    """
+    Number of shards for the deployment.
+    """
+
     cpu_min: int
     """
     Minimum CPU count for the deployment.
@@ -253,6 +258,11 @@ class Preset:
     Number of replicas for the preset.
     """
 
+    shard_count: int
+    """
+    Number of shards for the preset.
+    """
+
 
 @dataclass
 class User:
@@ -350,6 +360,11 @@ class CreateDeploymentRequest:
     Tags to apply to the deployment.
     """
 
+    shard_count: Optional[int] = 0
+    """
+    Number of shard for the deployment.
+    """
+
     endpoints: Optional[list[EndpointSpec]] = field(default_factory=list)
     """
     Endpoints to associate with the deployment.

scaleway/scaleway/datawarehouse/v1beta1/api.py

@@ -389,6 +389,7 @@ def create_deployment(
         region: Optional[ScwRegion] = None,
         project_id: Optional[str] = None,
         tags: Optional[list[str]] = None,
+        shard_count: Optional[int] = None,
         endpoints: Optional[list[EndpointSpec]] = None,
     ) -> Deployment:
         """
@@ -404,6 +405,7 @@ def create_deployment(
         :param region: Region to target. If none is passed will use default region from the config.
         :param project_id: The Project ID on which the deployment will be created.
         :param tags: Tags to apply to the deployment.
+        :param shard_count: Number of shard for the deployment.
         :param endpoints: Endpoints to associate with the deployment.
         :return: :class:`Deployment <Deployment>`
 
@@ -440,6 +442,7 @@ def create_deployment(
                     region=region,
                     project_id=project_id,
                     tags=tags,
+                    shard_count=shard_count,
                     endpoints=endpoints,
                 ),
                 self.client,

scaleway/scaleway/datawarehouse/v1beta1/marshalling.py

@@ -198,12 +198,6 @@ def unmarshal_Deployment(data: Any) -> Deployment:
     else:
         args["tags"] = []
 
-    field = data.get("version", None)
-    if field is not None:
-        args["version"] = field
-    else:
-        args["version"] = None
-
     field = data.get("created_at", None)
     if field is not None:
         args["created_at"] = parser.isoparse(field) if isinstance(field, str) else field
@@ -216,12 +210,24 @@ def unmarshal_Deployment(data: Any) -> Deployment:
     else:
         args["updated_at"] = None
 
+    field = data.get("version", None)
+    if field is not None:
+        args["version"] = field
+    else:
+        args["version"] = None
+
     field = data.get("replica_count", None)
     if field is not None:
         args["replica_count"] = field
     else:
         args["replica_count"] = 0
 
+    field = data.get("shard_count", None)
+    if field is not None:
+        args["shard_count"] = field
+    else:
+        args["shard_count"] = 0
+
     field = data.get("cpu_min", None)
     if field is not None:
         args["cpu_min"] = field
@@ -374,6 +380,12 @@ def unmarshal_Preset(data: Any) -> Preset:
     else:
         args["replica_count"] = 0
 
+    field = data.get("shard_count", None)
+    if field is not None:
+        args["shard_count"] = field
+    else:
+        args["shard_count"] = 0
+
     return Preset(**args)
 
 
@@ -570,6 +582,9 @@ def marshal_CreateDeploymentRequest(
     if request.tags is not None:
         output["tags"] = request.tags
 
+    if request.shard_count is not None:
+        output["shard_count"] = request.shard_count
+
     if request.endpoints is not None:
         output["endpoints"] = [
             marshal_EndpointSpec(item, defaults) for item in request.endpoints

scaleway/scaleway/datawarehouse/v1beta1/types.py

@@ -185,6 +185,11 @@ class Deployment:
     Number of replicas for the deployment.
     """
 
+    shard_count: int
+    """
+    Number of shards for the deployment.
+    """
+
     cpu_min: int
     """
     Minimum CPU count for the deployment.
@@ -253,6 +258,11 @@ class Preset:
     Number of replicas for the preset.
     """
 
+    shard_count: int
+    """
+    Number of shards for the preset.
+    """
+
 
 @dataclass
 class User:
@@ -350,6 +360,11 @@ class CreateDeploymentRequest:
     Tags to apply to the deployment.
     """
 
+    shard_count: Optional[int] = 0
+    """
+    Number of shard for the deployment.
+    """
+
     endpoints: Optional[list[EndpointSpec]] = field(default_factory=list)
     """
     Endpoints to associate with the deployment.