Reflect change in 25.10 API

truenas · Feb 12, 2025 · 2c037f9 · 2c037f9
1 parent e6eac40
commit 2c037f9
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/src/middlewared/middlewared/api/v25_10_0/virt_device.py b/src/middlewared/middlewared/api/v25_10_0/virt_device.py
@@ -2,13 +2,15 @@
 
 from pydantic import Field, field_validator
 
-from middlewared.api.base import BaseModel, LocalGID, LocalUID, NonEmptyString
+from middlewared.api.base import BaseModel, LocalGID, LocalUID, NonEmptyString, single_argument_args
 
 
 __all__ = [
     'DeviceType', 'InstanceType', 'VirtDeviceUSBChoicesArgs', 'VirtDeviceUSBChoicesResult',
     'VirtDeviceGPUChoicesArgs', 'VirtDeviceGPUChoicesResult', 'VirtDeviceDiskChoicesArgs',
     'VirtDeviceDiskChoicesResult', 'VirtDeviceNICChoicesArgs', 'VirtDeviceNICChoicesResult',
+    'VirtDeviceExportDiskImageArgs', 'VirtDeviceExportDiskImageResult', 'VirtDeviceImportDiskImageArgs',
+    'VirtDeviceImportDiskImageResult',
 ]
 
 
@@ -152,3 +154,24 @@ class VirtDeviceNICChoicesArgs(BaseModel):
 
 class VirtDeviceNICChoicesResult(BaseModel):
     result: dict[str, str]
+
+
+@single_argument_args('virt_device_import_disk_image')
+class VirtDeviceImportDiskImageArgs(BaseModel):
+    diskimg: NonEmptyString
+    zvol: NonEmptyString
+
+
+class VirtDeviceImportDiskImageResult(BaseModel):
+    result: bool
+
+
+@single_argument_args('virt_device_export_disk_image')
+class VirtDeviceExportDiskImageArgs(BaseModel):
+    format: NonEmptyString
+    directory: NonEmptyString
+    zvol: NonEmptyString
+
+
+class VirtDeviceExportDiskImageResult(BaseModel):
+    result: bool