fix(Cluster):updated slrum version

deNBI · Oct 16, 2024 · e6c80db · e6c80db
1 parent 373af93
commit e6c80db
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/portal_client.thrift b/portal_client.thrift
@@ -170,6 +170,8 @@ struct Image{
 
     /** Title of the image */
     13:optional string title
+
+    14:optional string slurm_version
 }
 /**
  * This Struct defines a VirtualMachine.

diff --git a/simple_vm_client/ttypes.py b/simple_vm_client/ttypes.py
diff --git a/simple_vm_client/util/thrift_converter.py b/simple_vm_client/util/thrift_converter.py
@@ -18,6 +18,7 @@ def os_to_thrift_image(openstack_image: OpenStack_Image) -> Image:
     if not properties:
         properties = {}
     image_type = properties.get("image_type", "image")
+    slurm_version = properties.get("slurm_version", "")
     image = Image(
         name=openstack_image.name,
         min_disk=openstack_image.min_disk,
@@ -27,6 +28,7 @@ def os_to_thrift_image(openstack_image: OpenStack_Image) -> Image:
         updated_at=openstack_image.updated_at,
         os_version=openstack_image.get("os_version", ""),
         os_distro=openstack_image.get("os_distro", ""),
+        slurm_version=openstack_image.get("slurm_version", slurm_version),
         title=properties.get("title", ""),
         openstack_id=openstack_image.id,
         description=properties.get("description", ""),