Обновление файла
This commit is contained in:
+120
-135
@@ -1,4 +1,12 @@
|
||||
"""Обёртка над Proxmox REST API.
|
||||
|
||||
Все методы — синхронные (proxmoxer работает через requests). Используйте
|
||||
внутри FastAPI через run_in_threadpool, если эндпоинт async.
|
||||
"""
|
||||
|
||||
import logging
|
||||
import random
|
||||
import re
|
||||
import string
|
||||
import time
|
||||
|
||||
@@ -6,6 +14,8 @@ from proxmoxer import ProxmoxAPI
|
||||
|
||||
from .config import settings
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def _client() -> ProxmoxAPI:
|
||||
"""Создаёт клиент Proxmox API, аутентифицированный по API-токену."""
|
||||
@@ -20,17 +30,19 @@ def _client() -> ProxmoxAPI:
|
||||
|
||||
|
||||
def gen_password(length: int = 14) -> str:
|
||||
"""Генерирует криптостойкий пароль из букв и цифр."""
|
||||
alphabet = string.ascii_letters + string.digits
|
||||
return "".join(random.choice(alphabet) for _ in range(length))
|
||||
|
||||
|
||||
def get_next_vmid() -> int:
|
||||
"""Берёт свободный VMID у самого Proxmox (гарантированно не занят)."""
|
||||
"""Берёт следующий свободный VMID у самого Proxmox (гарантированно не занят)."""
|
||||
px = _client()
|
||||
return int(px.cluster.nextid.get())
|
||||
|
||||
|
||||
def clone_vm(source_vmid: int, new_vmid: int, name: str, node: str = None, storage: str = None) -> None:
|
||||
"""Клонирует VM-шаблон в новую VM."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
upid = px.nodes(node).qemu(source_vmid).clone.post(
|
||||
@@ -42,6 +54,7 @@ def clone_vm(source_vmid: int, new_vmid: int, name: str, node: str = None, stora
|
||||
|
||||
|
||||
def resize_vm(vmid: int, cores: int, memory_mb: int, node: str = None) -> None:
|
||||
"""Изменяет количество vCPU и RAM у VM."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
px.nodes(node).qemu(vmid).config.put(cores=cores, memory=memory_mb)
|
||||
@@ -77,172 +90,152 @@ def create_lxc(
|
||||
return password
|
||||
|
||||
|
||||
def _wait_task(px: ProxmoxAPI, node: str, upid: str, timeout: int = 1200) -> None:
|
||||
"""Ждёт завершения асинхронной задачи Proxmox (клонирование, создание и т.п.)."""
|
||||
def _wait_task(px: ProxmoxAPI, node: str, upid: str, timeout: int = 1200, poll: float = 2.0) -> None:
|
||||
"""Ждёт завершения асинхронной задачи Proxmox с экспоненциальной задержкой.
|
||||
|
||||
Начинаем с poll=1s и удваиваем до 5s — снижает нагрузку на Proxmox
|
||||
при долгом клонировании больших дисков.
|
||||
"""
|
||||
start = time.time()
|
||||
delay = poll
|
||||
while time.time() - start < timeout:
|
||||
status = px.nodes(node).tasks(upid).status.get()
|
||||
if status.get("status") == "stopped":
|
||||
if status.get("exitstatus") != "OK":
|
||||
raise RuntimeError(f"Задача Proxmox завершилась с ошибкой: {status}")
|
||||
return
|
||||
time.sleep(2)
|
||||
time.sleep(delay)
|
||||
delay = min(delay * 1.5, 5.0)
|
||||
raise TimeoutError("Превышено время ожидания задачи Proxmox")
|
||||
|
||||
|
||||
def list_vm_templates(node=None):
|
||||
_DISK_KEYS = ("scsi0", "virtio0", "ide0", "sata0")
|
||||
|
||||
|
||||
def _disk_size_from_config(cfg: dict) -> int:
|
||||
"""Извлекает размер диска (ГБ) из конфигурации VM, иначе 10."""
|
||||
for k in _DISK_KEYS:
|
||||
v = cfg.get(k, "")
|
||||
if ",size=" in v:
|
||||
m = re.search(r"size=(\d+)G", v)
|
||||
if m:
|
||||
return int(m.group(1))
|
||||
return 10
|
||||
|
||||
|
||||
def list_vm_templates(node: str = None) -> list:
|
||||
"""Возвращает список VM-шаблонов (template=1) с параметрами."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
result = []
|
||||
for vm in px.nodes(node).qemu.get():
|
||||
if vm.get("template") == 1:
|
||||
cfg = px.nodes(node).qemu(vm["vmid"]).config.get()
|
||||
disk = 10
|
||||
for k in ("scsi0","virtio0","ide0","sata0"):
|
||||
v = cfg.get(k,"")
|
||||
if ",size=" in v:
|
||||
import re as _re
|
||||
m = _re.search(r"size=(\d+)G", v)
|
||||
if m: disk = int(m.group(1)); break
|
||||
result.append({
|
||||
"vmid": vm["vmid"], "name": vm.get("name",""),
|
||||
"cores": int(cfg.get("cores",1)), "memory_mb": int(cfg.get("memory",1024)),
|
||||
"disk_gb": disk,
|
||||
"vmid": vm["vmid"],
|
||||
"name": vm.get("name", ""),
|
||||
"cores": int(cfg.get("cores", 1)),
|
||||
"memory_mb": int(cfg.get("memory", 1024)),
|
||||
"disk_gb": _disk_size_from_config(cfg),
|
||||
})
|
||||
return result
|
||||
|
||||
def list_lxc_templates(node=None):
|
||||
|
||||
def list_lxc_templates(node: str = None) -> list:
|
||||
"""Возвращает список LXC-шаблонов (.tar.zst/.tar.gz/.tar.xz) из всех хранилищ."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
result = []
|
||||
for st in px.nodes(node).storage.get():
|
||||
if "vztmpl" in st.get("content",""):
|
||||
if "vztmpl" not in st.get("content", ""):
|
||||
continue
|
||||
try:
|
||||
for item in px.nodes(node).storage(st["storage"]).content.get():
|
||||
if item.get("content") == "vztmpl":
|
||||
name = item["volid"].split("/")[-1]
|
||||
for ext in (".tar.zst",".tar.gz",".tar.xz"): name = name.replace(ext,"")
|
||||
result.append({"volid": item["volid"], "name": name, "storage": st["storage"], "size_mb": round(item.get("size",0)/(1024**2),1)})
|
||||
except: pass
|
||||
for ext in (".tar.zst", ".tar.gz", ".tar.xz"):
|
||||
name = name.replace(ext, "")
|
||||
result.append({
|
||||
"volid": item["volid"],
|
||||
"name": name,
|
||||
"storage": st["storage"],
|
||||
"size_mb": round(item.get("size", 0) / (1024 ** 2), 1),
|
||||
})
|
||||
except Exception as exc:
|
||||
logger.warning("Не удалось прочитать шаблоны с хранилища %s: %s", st["storage"], exc)
|
||||
return result
|
||||
|
||||
|
||||
def resize_disk(vmid: int, disk_gb: int, node: str = None) -> None:
|
||||
"""Увеличивает диск VM до нужного размера (только в большую сторону)."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
config = px.nodes(node).qemu(vmid).config.get()
|
||||
for key in ("scsi0","virtio0","ide0","sata0"):
|
||||
for key in _DISK_KEYS:
|
||||
val = config.get(key, "")
|
||||
if val:
|
||||
import re as _re
|
||||
m = _re.search(r"size=(\d+)G", val)
|
||||
if not val:
|
||||
continue
|
||||
m = re.search(r"size=(\d+)G", val)
|
||||
current = int(m.group(1)) if m else 0
|
||||
if disk_gb > current:
|
||||
px.nodes(node).qemu(vmid).resize.put(disk=key, size=f"+{disk_gb - current}G")
|
||||
break
|
||||
return
|
||||
logger.warning("У VM %s не найден ни один диск для resize", vmid)
|
||||
|
||||
|
||||
def configure_cloud_init(vmid: int, ciuser: str, cipassword: str, node: str = None) -> None:
|
||||
"""Задаёт cloud-init user/password и переключает сеть на DHCP."""
|
||||
node = node or settings.pve_node
|
||||
_client().nodes(node).qemu(vmid).config.put(ciuser=ciuser, cipassword=cipassword, ipconfig0="ip=dhcp")
|
||||
|
||||
def get_instance_ip(vmid: int, node: str = None) -> str:
|
||||
"""Возвращает IP-адрес VM/LXC из гостевого агента или сетевой конфигурации."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
# Пробуем через гостевой агент (QEMU Guest Agent)
|
||||
try:
|
||||
ifaces = px.nodes(node).qemu(vmid).agent.get("network-get-interfaces").get("result", [])
|
||||
for iface in ifaces:
|
||||
if iface.get("name") != "lo" and iface.get("ip-addresses"):
|
||||
for addr in iface["ip-addresses"]:
|
||||
if addr.get("ip-address-type") == "ipv4" and not addr.get("ip-address", "").startswith("127."):
|
||||
return addr["ip-address"]
|
||||
except Exception:
|
||||
pass
|
||||
# Пробуем через LXC (если контейнер)
|
||||
try:
|
||||
config = px.nodes(node).lxc(vmid).config.get()
|
||||
# Просто пробуем разные пути для LXC
|
||||
except Exception:
|
||||
pass
|
||||
# Пробуем через DHCP-лиз Proxmox
|
||||
try:
|
||||
config = px.nodes(node).qemu(vmid).config.get()
|
||||
net = config.get("net0", "")
|
||||
if "dhcp" in net.lower():
|
||||
return "dhcp (агент не установлен)"
|
||||
except Exception:
|
||||
pass
|
||||
return "неизвестен"
|
||||
_client().nodes(node).qemu(vmid).config.put(
|
||||
ciuser=ciuser,
|
||||
cipassword=cipassword,
|
||||
ipconfig0="ip=dhcp",
|
||||
)
|
||||
|
||||
|
||||
def get_instance_ip(vmid: int, node: str = None) -> str:
|
||||
"""Возвращает IP-адрес VM/LXC из гостевого агента или сетевой конфигурации."""
|
||||
"""Возвращает основной IPv4 инстанса.
|
||||
|
||||
Для VM используется QEMU Guest Agent. Если агент недоступен — возвращает
|
||||
строку-плейсхолдер, чтобы UI мог отличить «не знаю» от «ошибка».
|
||||
"""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
# Пробуем через гостевой агент (QEMU Guest Agent)
|
||||
# 1. QEMU Guest Agent (требует установленного qemu-guest-agent в гостевой ОС).
|
||||
try:
|
||||
ifaces = px.nodes(node).qemu(vmid).agent.get("network-get-interfaces").get("result", [])
|
||||
ifaces = (
|
||||
px.nodes(node).qemu(vmid).agent.get("network-get-interfaces").get("result", [])
|
||||
)
|
||||
for iface in ifaces:
|
||||
if iface.get("name") != "lo" and iface.get("ip-addresses"):
|
||||
for addr in iface["ip-addresses"]:
|
||||
if addr.get("ip-address-type") == "ipv4" and not addr.get("ip-address", "").startswith("127."):
|
||||
return addr["ip-address"]
|
||||
except Exception:
|
||||
pass
|
||||
# Пробуем через LXC (если контейнер)
|
||||
try:
|
||||
config = px.nodes(node).lxc(vmid).config.get()
|
||||
# Просто пробуем разные пути для LXC
|
||||
except Exception:
|
||||
pass
|
||||
# Пробуем через DHCP-лиз Proxmox
|
||||
try:
|
||||
config = px.nodes(node).qemu(vmid).config.get()
|
||||
net = config.get("net0", "")
|
||||
if "dhcp" in net.lower():
|
||||
return "dhcp (агент не установлен)"
|
||||
except Exception:
|
||||
pass
|
||||
return "неизвестен"
|
||||
if iface.get("name") == "lo":
|
||||
continue
|
||||
for addr in iface.get("ip-addresses", []) or []:
|
||||
ip = addr.get("ip-address", "")
|
||||
if addr.get("ip-address-type") == "ipv4" and not ip.startswith("127."):
|
||||
return ip
|
||||
except Exception as exc:
|
||||
logger.debug("VM %s: guest-agent недоступен (%s)", vmid, exc)
|
||||
|
||||
|
||||
def get_instance_ip(vmid: int, node: str = None) -> str:
|
||||
"""Возвращает IP-адрес VM/LXC из гостевого агента или сетевой конфигурации."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
# Пробуем через гостевой агент (QEMU Guest Agent)
|
||||
# 2. LXC — адрес можно увидеть через конфиг + интерфейсы внутри.
|
||||
try:
|
||||
ifaces = px.nodes(node).qemu(vmid).agent.get("network-get-interfaces").get("result", [])
|
||||
ifaces = px.nodes(node).lxc(vmid).interfaces.get() or []
|
||||
for iface in ifaces:
|
||||
if iface.get("name") != "lo" and iface.get("ip-addresses"):
|
||||
for addr in iface["ip-addresses"]:
|
||||
if addr.get("ip-address-type") == "ipv4" and not addr.get("ip-address", "").startswith("127."):
|
||||
return addr["ip-address"]
|
||||
except Exception:
|
||||
pass
|
||||
# Пробуем через LXC (если контейнер)
|
||||
try:
|
||||
config = px.nodes(node).lxc(vmid).config.get()
|
||||
# Просто пробуем разные пути для LXC
|
||||
except Exception:
|
||||
pass
|
||||
# Пробуем через DHCP-лиз Proxmox
|
||||
try:
|
||||
config = px.nodes(node).qemu(vmid).config.get()
|
||||
net = config.get("net0", "")
|
||||
if "dhcp" in net.lower():
|
||||
return "dhcp (агент не установлен)"
|
||||
except Exception:
|
||||
pass
|
||||
return "неизвестен"
|
||||
for ip in iface.get("ip-addresses", []) or []:
|
||||
if not ip.startswith("127."):
|
||||
return ip
|
||||
except Exception as exc:
|
||||
logger.debug("LXC %s: интерфейсы недоступны (%s)", vmid, exc)
|
||||
|
||||
# 3. Fallback: понятная заглушка вместо молчаливого «неизвестен».
|
||||
return "не определён (агент недоступен)"
|
||||
|
||||
|
||||
def get_live_stats(vmid: int, node: str = None) -> dict:
|
||||
"""Возвращает live-показатели: cpu%, ram_used, ram_total, uptime."""
|
||||
"""Live-показатели (cpu%, mem_used, mem_total, uptime, status) для VM и LXC."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
try:
|
||||
status = px.nodes(node).qemu(vmid).status.current.get()
|
||||
|
||||
def _shape(status: dict) -> dict:
|
||||
return {
|
||||
"cpu": round(status.get("cpu", 0) * 100, 1),
|
||||
"mem_used": status.get("mem", 0),
|
||||
@@ -250,43 +243,34 @@ def get_live_stats(vmid: int, node: str = None) -> dict:
|
||||
"uptime": status.get("uptime", 0),
|
||||
"status": status.get("status", "unknown"),
|
||||
}
|
||||
except Exception:
|
||||
|
||||
for endpoint in (px.nodes(node).qemu(vmid).status.current, px.nodes(node).lxc(vmid).status.current):
|
||||
try:
|
||||
status = px.nodes(node).lxc(vmid).status.current.get()
|
||||
return {
|
||||
"cpu": round(status.get("cpu", 0) * 100, 1),
|
||||
"mem_used": status.get("mem", 0),
|
||||
"mem_total": status.get("maxmem", 0),
|
||||
"uptime": status.get("uptime", 0),
|
||||
"status": status.get("status", "unknown"),
|
||||
}
|
||||
return _shape(endpoint.get())
|
||||
except Exception:
|
||||
continue
|
||||
return {"cpu": 0, "mem_used": 0, "mem_total": 0, "uptime": 0, "status": "unknown"}
|
||||
|
||||
|
||||
def get_live_stats_lxc(vmid: int, node: str = None) -> dict:
|
||||
"""Live-показатели LXC контейнера."""
|
||||
return get_live_stats(vmid, node)
|
||||
# Алиас для обратной совместимости со старым кодом.
|
||||
get_live_stats_lxc = get_live_stats
|
||||
|
||||
|
||||
_VALID_ACTIONS = {"start", "stop", "shutdown", "reboot"}
|
||||
|
||||
|
||||
def guest_action(guest_type: str, vmid: int, action: str, node: str = None) -> None:
|
||||
"""action: start | stop | shutdown | reboot"""
|
||||
"""Выполняет действие над VM/LXC: start | stop | shutdown | reboot."""
|
||||
if action not in _VALID_ACTIONS:
|
||||
raise ValueError(f"Неизвестное действие: {action}")
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
endpoint = px.nodes(node).qemu(vmid) if guest_type == "vm" else px.nodes(node).lxc(vmid)
|
||||
if action == "start":
|
||||
endpoint.status.start.post()
|
||||
elif action == "stop":
|
||||
endpoint.status.stop.post()
|
||||
elif action == "shutdown":
|
||||
endpoint.status.shutdown.post()
|
||||
elif action == "reboot":
|
||||
endpoint.status.reboot.post()
|
||||
else:
|
||||
raise ValueError(f"Неизвестное действие: {action}")
|
||||
getattr(endpoint.status, action).post()
|
||||
|
||||
|
||||
def delete_guest(guest_type: str, vmid: int, node: str = None) -> None:
|
||||
"""Удаляет VM или LXC (синхронно — Proxmox дожидается завершения)."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
if guest_type == "vm":
|
||||
@@ -296,6 +280,7 @@ def delete_guest(guest_type: str, vmid: int, node: str = None) -> None:
|
||||
|
||||
|
||||
def get_status(guest_type: str, vmid: int, node: str = None) -> dict:
|
||||
"""Текущий статус VM/LXC (running, stopped, …)."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
if guest_type == "vm":
|
||||
@@ -304,7 +289,7 @@ def get_status(guest_type: str, vmid: int, node: str = None) -> dict:
|
||||
|
||||
|
||||
def get_vnc_ticket(guest_type: str, vmid: int, node: str = None) -> dict:
|
||||
"""Запрашивает у Proxmox тикет для VNC/websocket-консоли."""
|
||||
"""Запрашивает у Proxmox одноразовый тикет для VNC/websocket-консоли."""
|
||||
node = node or settings.pve_node
|
||||
px = _client()
|
||||
endpoint = px.nodes(node).qemu(vmid) if guest_type == "vm" else px.nodes(node).lxc(vmid)
|
||||
|
||||
Reference in New Issue
Block a user