Archived
72 lines
4.0 KiB
Python
72 lines
4.0 KiB
Python
import re
|
||
import time
|
||
from urllib.parse import urlparse
|
||
|
||
from proxmoxer import ProxmoxAPI
|
||
|
||
from .config import settings
|
||
|
||
|
||
def _client() -> ProxmoxAPI:
|
||
"""Создаёт Proxmox API client по отдельному токену панели."""
|
||
parsed = urlparse(settings.pve_host)
|
||
if not parsed.hostname:
|
||
raise RuntimeError("PVE_HOST имеет некорректный формат")
|
||
if "!" not in settings.pve_token_name:
|
||
raise RuntimeError("PVE_TOKEN_NAME должен иметь формат user@realm!token")
|
||
user, token_name = settings.pve_token_name.split("!", 1)
|
||
return ProxmoxAPI(parsed.hostname, port=parsed.port or 8006, user=user, token_name=token_name, token_value=settings.pve_token_value.get_secret_value(), verify_ssl=settings.pve_verify_ssl)
|
||
|
||
|
||
def _wait_task(client: ProxmoxAPI, node: str, upid: str, timeout: int = 1800) -> None:
|
||
"""Ожидает завершения задачи Proxmox с ограничением времени."""
|
||
deadline = time.monotonic() + timeout
|
||
while time.monotonic() < deadline:
|
||
task = client.nodes(node).tasks(upid).status.get()
|
||
if task.get("status") == "stopped":
|
||
if task.get("exitstatus") != "OK":
|
||
raise RuntimeError("Задача Proxmox завершилась с ошибкой")
|
||
return
|
||
time.sleep(2)
|
||
raise TimeoutError("Превышено время ожидания задачи Proxmox")
|
||
|
||
|
||
def next_vmid() -> int:
|
||
"""Запрашивает следующий свободный VMID у Proxmox."""
|
||
return int(_client().cluster.nextid.get())
|
||
|
||
|
||
def provision_vm(source_vmid: int, vmid: int, name: str, cores: int, memory_mb: int, disk_gb: int, username: str, password: str) -> None:
|
||
"""Клонирует cloud-init VM, настраивает ресурсы и запускает её."""
|
||
client = _client(); node = settings.pve_node
|
||
params = {"newid": vmid, "name": name, "full": 1}
|
||
if settings.pve_vm_storage: params["storage"] = settings.pve_vm_storage
|
||
_wait_task(client, node, client.nodes(node).qemu(source_vmid).clone.post(**params))
|
||
client.nodes(node).qemu(vmid).config.put(cores=cores, memory=memory_mb, ciuser=username, cipassword=password, ipconfig0="ip=dhcp")
|
||
config = client.nodes(node).qemu(vmid).config.get()
|
||
for key in ("scsi0", "virtio0", "sata0", "ide0"):
|
||
match = re.search(r"size=(\d+)G", config.get(key, ""))
|
||
if match and disk_gb > int(match.group(1)):
|
||
client.nodes(node).qemu(vmid).resize.put(disk=key, size=f"+{disk_gb - int(match.group(1))}G")
|
||
break
|
||
client.nodes(node).qemu(vmid).status.start.post()
|
||
|
||
|
||
def provision_lxc_clone(source_vmid: int, vmid: int, name: str, cores: int, memory_mb: int, disk_gb: int, password: str) -> None:
|
||
"""Клонирует готовый LXC-шаблон по CTID и запускает контейнер."""
|
||
client = _client(); node = settings.pve_node
|
||
params = {"newid": vmid, "hostname": name, "full": 1}
|
||
if settings.pve_lxc_storage: params["storage"] = settings.pve_lxc_storage
|
||
_wait_task(client, node, client.nodes(node).lxc(source_vmid).clone.post(**params))
|
||
client.nodes(node).lxc(vmid).config.put(cores=cores, memory=memory_mb, swap=memory_mb, password=password, net0=f"name=eth0,bridge={settings.pve_bridge},ip=dhcp")
|
||
client.nodes(node).lxc(vmid).resize.put(disk="rootfs", size=f"{disk_gb}G")
|
||
client.nodes(node).lxc(vmid).status.start.post()
|
||
|
||
|
||
def provision_lxc(source_template: str, vmid: int, name: str, cores: int, memory_mb: int, disk_gb: int, password: str) -> None:
|
||
"""Создаёт LXC из архивного CT-шаблона и запускает его."""
|
||
client = _client(); node = settings.pve_node
|
||
upid = client.nodes(node).lxc.post(vmid=vmid, hostname=name, ostemplate=source_template, cores=cores, memory=memory_mb, swap=memory_mb, rootfs=f"{settings.pve_lxc_storage}:{disk_gb}", password=password, net0=f"name=eth0,bridge={settings.pve_bridge},ip=dhcp", unprivileged=1)
|
||
_wait_task(client, node, upid)
|
||
client.nodes(node).lxc(vmid).status.start.post()
|