deevirt_schema/api/vm.py

213 lines
8.5 KiB
Python

from typing import List, Optional
from uuid import uuid4
from pydantic import UUID4, Field, ConfigDict
from pydantic import BaseModel as _BaseModel
from ..libvirt.dom import config as configDom
class BaseModel(_BaseModel):
class Config:
@staticmethod
def json_schema_extra(schema: dict, _):
props = {}
for k, v in schema.get('properties', {}).items():
if not v.get("hidden", False):
props[k] = v
schema["properties"] = props
"""Common"""
class Resource(BaseModel):
"""Experimental"""
name:Optional[str] = Field(default=None, description="Nom du pool")
class VCPU(BaseModel):
value:int = Field(description="Nombre de VCPU")
dedicated:Optional[bool] = Field(default=False, description="VCPU dédié")
hotplug:Optional[bool] = Field(default=False, description="Ajout de VCPU à chaud.")
class Memory(BaseModel):
unit:Optional[configDom.UnitEnum] = Field(default=configDom.UnitEnum.GiB, description="Unité de mesure.")
value:int = Field(description="Taille de la mémoire.")
hotplug:Optional[bool] = Field(default=False,description="Ajout à chaud de la mémoire.")
class OSBootMenu(BaseModel):
enable:Optional[bool] = Field(default=False,description="Activation du menu de démarrage.")
delay:Optional[int] = Field(default=0, description="Delai maximal pour entrer dans le menu démarrage en millisecondes")
class OSBootOrder(BaseModel):
dev:Optional[configDom.OSBoot.bootEnum] = Field(default=None, description="Périphérique.")
class OSInfo(BaseModel):
model_config = ConfigDict(from_attributes=True)
name:str = Field(alias="os.name", serialization_alias="os_name")
version:str = Field(alias="os.version-id", serialization_alias="os_version")
machine:str = Field(alias="os.machine", serialization_alias="os_machine")
kernel:str = Field(alias="os.kernel-release", serialization_alias="os_kernel_release")
class OS(BaseModel):
firmware:configDom.OS.firmwareEnum = Field(default=configDom.OS.firmwareEnum.bios, description="Microprogramme")
enrolled_keys:Optional[bool] = Field(default=None, description="Vérification de la signature du chargeur de démarrage")
secure_boot:Optional[bool] = Field(default=None, description="Démarrage sécurisé")
boot_order:Optional[List[configDom.OSBoot.bootEnum]] = Field(default=None, description="Ordre de démarrage des périphériques")
boot_menu:OSBootMenu = Field(default=OSBootMenu(), description="Menu de démarrage")
class Clock(BaseModel):
offset:configDom.Clock.offsetEnum = Field(default=configDom.Clock.offsetEnum.utc, description="Sélection de l'offset ?")
class Event(BaseModel):
on_poweroff:configDom.Dom.on_poweroffEnum = Field(default=configDom.Dom.on_poweroffEnum.destroy, description="Action lors d'un poweroff")
on_reboot:configDom.Dom.on_rebootEnum = Field(default=configDom.Dom.on_rebootEnum.restart, description="Action lors d'un redémarrage")
on_crash:configDom.Dom.on_crashEnum = Field(default=configDom.Dom.on_crashEnum.restart, description="Action lors d'un crash de la VM")
on_failurelock:configDom.Dom.on_failurelockEnum = Field(default=configDom.Dom.on_failurelockEnum.ignore, description="A tester")
class DevicesDisk(BaseModel):
model_config = ConfigDict(from_attributes=True)
pool_id:Optional[UUID4] = Field(default=None)
id:UUID4 = Field(default_factory=uuid4)
size:int = Field()
unit:configDom.UnitEnum = Field(default=configDom.UnitEnum.GiB)
bus:configDom.DevicesDiskTarget.busEnum = Field(default=configDom.DevicesDiskTarget.busEnum.scsi)
class DevicesCDROMSource(BaseModel):
protocol:configDom.DevicesDiskSource.protocolEnum = Field(default=configDom.DevicesDiskSource.protocolEnum.https)
name:str = Field()
class DevicesCDROM(BaseModel):
model_config = ConfigDict(from_attributes=True)
device:configDom.DevicesDisk.deviceEnum = Field(default=configDom.DevicesDisk.deviceEnum.cdrom, json_schema_extra={"hidden": True})
source:DevicesCDROMSource = Field(default=None)
class DevicesInterfaceGuest(BaseModel):
model_config = ConfigDict(from_attributes=True)
addr:str = Field()
class DevicesInterface(BaseModel):
model_config = ConfigDict(from_attributes=True)
switch_id:UUID4
id:UUID4 = Field(default_factory=uuid4)
mac_addr:str = Field(default='')
class Graphics(BaseModel):
port:int = Field(default=-1)
autoport:bool = Field(default=False)
websocket:int = Field(default=-1)
password:Optional[str] = Field(default=None)
keymap:configDom.Graphics.keymapEnum = Field(default=configDom.Graphics.keymapEnum.fr)
power_control:bool = Field(default=False)
class Audio(BaseModel):
pass
class VideoModel(BaseModel):
type:configDom.VideoModel.typeEnum = Field(default=configDom.VideoModel.typeEnum.cirrus)
vram:int = Field(default=16384)
primary:bool = Field(default=True)
class Video(BaseModel):
model:VideoModel = Field()
class Watchdog(BaseModel):
model:configDom.Watchdog.modelEnum = Field(default=configDom.Watchdog.modelEnum.itco)
action:configDom.Watchdog.actionEnum = Field(default=configDom.Watchdog.actionEnum.reset)
class TPM(BaseModel):
model:configDom.TPM.modelEnum = Field(default=configDom.TPM.modelEnum.tpm_tis)
version:configDom.TPMBackend.versionEnum = Field(default=configDom.TPMBackend.versionEnum.v20)
class Devices(BaseModel):
disks:Optional[List[DevicesDisk]] = Field(default=None, description="Disques virtuel")
cdrom:Optional[DevicesCDROM] = Field(default=None, description="CDROM virtuel")
interfaces:Optional[List[DevicesInterface]] = Field(default=None, description="Interfaces réseau")
graphics:Optional[Graphics] = Field(default=None, description="Console VNC")
audio:Optional[Audio] = Field(default=None, description="Carte Audio")
video:Optional[Video] = Field(default=None, description="Carte Vidéo")
watchdog:Watchdog = Field(default=None, description="Chien de garde")
tpm:Optional[TPM] = Field(default=None, description="Puce de sécurité")
class VM(BaseModel):
model_config = ConfigDict(from_attributes=True)
# General metadata
id:UUID4 = Field(default_factory=uuid4)
datacenter_id:Optional[UUID4] = Field(default=None)
name:str = Field()
description:Optional[str] = Field(default=None)
# CPU Allocation
vcpu:VCPU = Field()
# Memory Allocation
memory:Memory = Field()
# Pool of resource (Experimental)
resource:Resource = Field(default=Resource(), description="Pool de resources")
# BIOS BootLoader
os:OS = Field(default=OS(), description="Options VM: Options de démarrage.")
# Clock configuration
clock:Clock = Field(default=Clock(), description="Options VM: Gestion de l'horloge")
# Event configuration
event:Event = Field(default=Event(), description="Options VM: Gestion des évènements")
# Devices
devices:Devices = Field(default=Devices(), description="Matériel virtuel: Périphériques")
"""VM LIST GET"""
class VMs(BaseModel):
model_config = ConfigDict(from_attributes=True)
class Datacenter(BaseModel):
model_config = ConfigDict(from_attributes=True)
id:UUID4
name:str
id:UUID4
name:str
state:int
datacenter:Optional[Datacenter] = Field(default=[])
"""VM GET"""
class VMGetVCPU(VCPU):
max:Optional[int] = Field(default=None, description="Maximum de VCPUs")
class VMGetOS(OS):
guest_info:Optional[OSInfo] = Field(default=None, description="Information de l'OS client")
class VMGetDevicesDisk(DevicesDisk):
allocated:float = Field(default=None)
class VMGetDevicesInterface(DevicesInterface):
guest_info:Optional[List[DevicesInterfaceGuest]] = Field(default=None, description="Information de configuration de la carte réseau.")
class VMGetDevices(Devices):
disks:Optional[List[VMGetDevicesDisk]] = Field(default=None, description="Disques virtuel")
interfaces:Optional[List[VMGetDevicesInterface]] = Field(default=None, description="Interfaces réseau")
class VMGet(VM):
vcpu:VMGetVCPU = Field()
gen_id:Optional[UUID4] = Field(default=None)
os:VMGetOS = Field(default=VMGetOS(), description="Options VM: Options de démarrage.")
devices:VMGetDevices = Field(default=VMGetDevices(), description="Matériel virtuel: Périphériques")
"""VM POST"""
class VMPost(VM):
id:UUID4 = Field(default_factory=uuid4)
datacenter_id:UUID4 = Field()
"""VM DELETE"""
class VMDelete(BaseModel):
id:UUID4 = Field()
class HardDisk(BaseModel):
pass
class NetInterfaceCreate(BaseModel):
pass