Add size and resize on vdi

Signed-off-by: Damien Thenot <[email protected]>

Author: Nambrok

lib/vdi.py

@@ -47,6 +47,13 @@ def clone(self):
     def readonly(self) -> bool:
         return strtobool(self.param_get("read-only"))
 
+    def size(self) -> int:
+        return int(self.param_get("virtual-size"))
+
+    def resize(self, new_size: int) -> None:
+        logging.info(f"Resizing VDI {self.uuid} to {new_size}")
+        self.sr.pool.master.xe("vdi-resize", {"uuid": self.uuid, "disk-size": str(new_size)})
+
     def __str__(self):
         return f"VDI {self.uuid} on SR {self.sr.uuid}"