feat(rdb): add volume_type for snapshots (#396)

Author: scaleway-bot

scaleway-async/scaleway_async/rdb/v1/__init__.py

@@ -73,6 +73,7 @@
 from .types import SetInstanceACLRulesResponse
 from .types import SetInstanceSettingsResponse
 from .types import Snapshot
+from .types import SnapshotVolumeType
 from .types import UpgradableVersion
 from .types import UpgradeInstanceRequestMajorUpgradeWorkflow
 from .types import User
@@ -159,6 +160,7 @@
     "SetInstanceACLRulesResponse",
     "SetInstanceSettingsResponse",
     "Snapshot",
+    "SnapshotVolumeType",
     "UpgradableVersion",
     "UpgradeInstanceRequestMajorUpgradeWorkflow",
     "User",

scaleway-async/scaleway_async/rdb/v1/marshalling.py

@@ -66,6 +66,7 @@
     SetInstanceACLRulesResponse,
     SetInstanceSettingsResponse,
     Snapshot,
+    SnapshotVolumeType,
     UpgradableVersion,
     UpgradeInstanceRequestMajorUpgradeWorkflow,
     User,
@@ -419,6 +420,23 @@ def unmarshal_ReadReplica(data: Any) -> ReadReplica:
     return ReadReplica(**args)
 
 
+def unmarshal_SnapshotVolumeType(data: Any) -> SnapshotVolumeType:
+    if type(data) is not dict:
+        raise TypeError(
+            f"Unmarshalling the type 'SnapshotVolumeType' failed as data isn't a dictionary."
+        )
+
+    args: Dict[str, Any] = {}
+
+    field = data.get("class", None)
+    args["class_"] = field
+
+    field = data.get("type", None)
+    args["type_"] = field
+
+    return SnapshotVolumeType(**args)
+
+
 def unmarshal_UpgradableVersion(data: Any) -> UpgradableVersion:
     if type(data) is not dict:
         raise TypeError(
@@ -854,6 +872,11 @@ def unmarshal_Snapshot(data: Any) -> Snapshot:
     field = data.get("updated_at", None)
     args["updated_at"] = parser.isoparse(field) if type(field) is str else field
 
+    field = data.get("volume_type", None)
+    args["volume_type"] = (
+        unmarshal_SnapshotVolumeType(field) if field is not None else None
+    )
+
     return Snapshot(**args)
 
 

scaleway-async/scaleway_async/rdb/v1/types.py

@@ -1491,12 +1491,24 @@ class Snapshot:
     Source node type.
     """
 
+    volume_type: Optional[SnapshotVolumeType]
+    """
+    Type of volume where data is stored (lssd, bssd or sbs).
+    """
+
     region: Region
     """
     Region of this snapshot.
     """
 
 
+@dataclass
+class SnapshotVolumeType:
+    type_: VolumeType
+
+    class_: StorageClass
+
+
 @dataclass
 class UpgradableVersion:
     id: str

scaleway/scaleway/rdb/v1/__init__.py

@@ -73,6 +73,7 @@
 from .types import SetInstanceACLRulesResponse
 from .types import SetInstanceSettingsResponse
 from .types import Snapshot
+from .types import SnapshotVolumeType
 from .types import UpgradableVersion
 from .types import UpgradeInstanceRequestMajorUpgradeWorkflow
 from .types import User
@@ -159,6 +160,7 @@
     "SetInstanceACLRulesResponse",
     "SetInstanceSettingsResponse",
     "Snapshot",
+    "SnapshotVolumeType",
     "UpgradableVersion",
     "UpgradeInstanceRequestMajorUpgradeWorkflow",
     "User",

scaleway/scaleway/rdb/v1/marshalling.py

@@ -66,6 +66,7 @@
     SetInstanceACLRulesResponse,
     SetInstanceSettingsResponse,
     Snapshot,
+    SnapshotVolumeType,
     UpgradableVersion,
     UpgradeInstanceRequestMajorUpgradeWorkflow,
     User,
@@ -419,6 +420,23 @@ def unmarshal_ReadReplica(data: Any) -> ReadReplica:
     return ReadReplica(**args)
 
 
+def unmarshal_SnapshotVolumeType(data: Any) -> SnapshotVolumeType:
+    if type(data) is not dict:
+        raise TypeError(
+            f"Unmarshalling the type 'SnapshotVolumeType' failed as data isn't a dictionary."
+        )
+
+    args: Dict[str, Any] = {}
+
+    field = data.get("class", None)
+    args["class_"] = field
+
+    field = data.get("type", None)
+    args["type_"] = field
+
+    return SnapshotVolumeType(**args)
+
+
 def unmarshal_UpgradableVersion(data: Any) -> UpgradableVersion:
     if type(data) is not dict:
         raise TypeError(
@@ -854,6 +872,11 @@ def unmarshal_Snapshot(data: Any) -> Snapshot:
     field = data.get("updated_at", None)
     args["updated_at"] = parser.isoparse(field) if type(field) is str else field
 
+    field = data.get("volume_type", None)
+    args["volume_type"] = (
+        unmarshal_SnapshotVolumeType(field) if field is not None else None
+    )
+
     return Snapshot(**args)
 
 

scaleway/scaleway/rdb/v1/types.py

@@ -1491,12 +1491,24 @@ class Snapshot:
     Source node type.
     """
 
+    volume_type: Optional[SnapshotVolumeType]
+    """
+    Type of volume where data is stored (lssd, bssd or sbs).
+    """
+
     region: Region
     """
     Region of this snapshot.
     """
 
 
+@dataclass
+class SnapshotVolumeType:
+    type_: VolumeType
+
+    class_: StorageClass
+
+
 @dataclass
 class UpgradableVersion:
     id: str