Refactor backend: Split models and proxmox_client into modules, add alembic

backend/app/proxmox/mixins/vm.py

@@ -0,0 +1,241 @@
+from typing import List, Dict, Optional, Union, Any
+import time
+import urllib3
+from loguru import logger
+
+class VmMixin:
+        def start_vm(self, node: str, vmid: int) -> Optional[str]:
+            """Запустить VM. Возвращает UPID задачи или None при ошибке."""
+            if not self.proxmox:
+                return None
+            
+            try:
+                upid = self.proxmox.nodes(node).qemu(vmid).status.start.post()
+                return upid if isinstance(upid, str) else None
+            except Exception as e:
+                logger.error(f"Ошибка запуска VM {vmid} на {node}: {e}")
+                return None
+
+        def stop_vm(self, node: str, vmid: int, force: bool = False) -> Optional[str]:
+            """Остановить VM. Возвращает UPID задачи или None при ошибке."""
+            if not self.proxmox:
+                return None
+            
+            try:
+                upid = self.proxmox.nodes(node).qemu(vmid).status.stop.post()
+                return upid if isinstance(upid, str) else None
+            except Exception as e:
+                logger.error(f"Ошибка остановки VM {vmid} на {node}: {e}")
+                return None
+
+        def restart_vm(self, node: str, vmid: int) -> Optional[str]:
+            """Перезапустить VM. Возвращает UPID задачи или None при ошибке."""
+            if not self.proxmox:
+                return None
+            
+            try:
+                upid = self.proxmox.nodes(node).qemu(vmid).status.reboot.post()
+                return upid if isinstance(upid, str) else None
+            except Exception as e:
+                logger.error(f"Ошибка перезапуска VM {vmid} на {node}: {e}")
+                return None
+
+        def shutdown_vm(self, node: str, vmid: int, timeout: int = 60) -> Optional[str]:
+            """Корректно выключить VM (ACPI). Возвращает UPID задачи."""
+            if not self.proxmox:
+                return None
+            try:
+                upid = self.proxmox.nodes(node).qemu(vmid).status.shutdown.post(timeout=timeout)
+                return upid if isinstance(upid, str) else None
+            except Exception as e:
+                logger.error(f"Ошибка корректного выключения VM {vmid} на {node}: {e}")
+                return None
+
+        def clone_vm(self, node: str, vmid: int, new_vmid: int, name: str,
+                     full: bool = True, target_node: Optional[str] = None,
+                     target_storage: Optional[str] = None,
+                     description: Optional[str] = None) -> Optional[str]:
+            """Клонировать VM (qemu). Работает и для шаблонов, и для обычных VM."""
+            if not self.proxmox:
+                return None
+            try:
+                params = {'newid': new_vmid, 'name': name, 'full': 1 if full else 0}
+                if target_storage:
+                    params['storage'] = target_storage
+                if target_node and target_node != node:
+                    params['target'] = target_node
+                if description:
+                    params['description'] = description
+                upid = self.proxmox.nodes(node).qemu(vmid).clone.post(**params)
+                return upid if isinstance(upid, str) else None
+            except Exception as e:
+                logger.error(f"Ошибка клонирования VM {vmid} -> {new_vmid}: {e}")
+                return None
+
+        def change_vm_password(self, node: str, vmid: int, username: str, password: str) -> Dict:
+            """
+            Сменить пароль пользователя на VM через QEMU guest agent.
+            Требует установленный qemu-guest-agent в VM.
+            """
+            if not self.proxmox:
+                return {'success': False, 'error': 'Proxmox connection not initialized'}
+            try:
+                # Proxmox API: POST /nodes/{node}/qemu/{vmid}/agent/set-user-password
+                self.proxmox.nodes(node).qemu(vmid).agent('set-user-password').post(
+                    username=username,
+                    password=password,
+                )
+                return {'success': True}
+            except Exception as e:
+                logger.error(f"Не удалось сменить пароль на VM {vmid}: {e}")
+                return {'success': False, 'error': str(e)}
+
+        def set_vm_notes(self, node: str, vmid: int, description: str) -> bool:
+            """Установить description (заметки) у VM."""
+            if not self.proxmox:
+                return False
+            try:
+                self.proxmox.nodes(node).qemu(vmid).config.put(description=description or '')
+                return True
+            except Exception as e:
+                logger.error(f"Ошибка обновления заметок VM {vmid}: {e}")
+                return False
+
+        def attach_iso(self, node: str, vmid: int, iso_volid: str, device: str = 'ide2') -> bool:
+            """Подключить ISO образ к VM (только KVM)."""
+            if not self.proxmox:
+                return False
+            try:
+                value = f"{iso_volid},media=cdrom"
+                self.proxmox.nodes(node).qemu(vmid).config.put(**{device: value})
+                return True
+            except Exception as e:
+                logger.error(f"Ошибка подключения ISO {iso_volid} к VM {vmid}: {e}")
+                return False
+
+        def detach_iso(self, node: str, vmid: int, device: str = 'ide2') -> bool:
+            """Отключить ISO образ (CD-ROM) у VM."""
+            if not self.proxmox:
+                return False
+            try:
+                self.proxmox.nodes(node).qemu(vmid).config.put(**{device: 'none,media=cdrom'})
+                return True
+            except Exception as e:
+                logger.error(f"Ошибка отключения ISO у VM {vmid}: {e}")
+                return False
+
+        def force_stop_vm(self, node: str, vmid: int) -> bool:
+            """Принудительно остановить ВМ (через SSH - аналог kill -9)"""
+            if not self.proxmox or not self.host:
+                return False
+            
+            try:
+                # Пробуем остановить через SSH команду
+                # Сначала получаем информацию о ВМ
+                vm_info = self.proxmox.nodes(node).qemu(vmid).status.current.get()
+                if vm_info.get('status') != 'running':
+                    logger.info(f"VM {vmid} на {node} уже остановлена")
+                    return True
+                
+                # Используем qm sendkey или qm stop с timeout=0
+                try:
+                    # Пробуем обычный stop - может сработать
+                    self.proxmox.nodes(node).qemu(vmid).status.stop.post()
+                    logger.info(f"Принудительная остановка VM {vmid} на {node} выполнена")
+                    return True
+                except Exception as api_error:
+                    logger.warning(f"Прямой stop не сработал, ошибка: {api_error}")
+                    # Если прямой stop не сработал, это просто значит что ВМ не бежит или API не поддерживает
+                    return False
+                    
+            except Exception as e:
+                logger.error(f"Ошибка принудительной остановки VM {vmid}: {e}")
+                return False
+
+        def get_vm_status(self, node: str, vmid: int) -> Optional[Dict]:
+            """Получить статус конкретной VM"""
+            if not self.proxmox:
+                return None
+            
+            try:
+                status = self.proxmox.nodes(node).qemu(vmid).status.current.get()
+                return status
+            except Exception as e:
+                logger.error(f"Ошибка получения статуса VM {vmid} на {node}: {e}")
+                return None
+
+        def get_vm_stats(self, node: str, vmid: int) -> Optional[Dict]:
+            """Получить статистику VM (CPU, память, диск)"""
+            return self.get_vm_status(node, vmid)
+
+        def get_vm_rrddata(self, node: str, vmid: int, timeframe: str = "hour") -> Dict:
+            """
+            Получить исторические данные VM для графиков (CPU, Memory, Network, Disk IO)
+            
+            Args:
+                node: Имя ноды
+                vmid: ID виртуальной машины
+                timeframe: Период времени (hour, day, week, month, year)
+            
+            Returns:
+                Dict с временными рядами данных
+            """
+            if not self.proxmox:
+                return {}
+            
+            try:
+                rrddata = self.proxmox.nodes(node).qemu(vmid).rrddata.get(timeframe=timeframe)
+                return {'data': rrddata, 'timeframe': timeframe}
+            except Exception as e:
+                logger.error(f"Ошибка получения RRD данных VM {vmid} на {node}: {e}")
+                return {}
+
+        def get_vm_vnc(self, node: str, vmid: int) -> Dict:
+            """
+            Получить данные для VNC подключения к VM
+            
+            Args:
+                node: Имя ноды
+                vmid: ID виртуальной машины
+            
+            Returns:
+                Dict с данными VNC (port, ticket, cert, upid)
+            """
+            if not self.proxmox:
+                return {}
+            
+            try:
+                # Создать VNC прокси сессию
+                vnc_data = self.proxmox.nodes(node).qemu(vmid).vncproxy.post(websocket=1)
+                return vnc_data
+            except Exception as e:
+                logger.error(f"Ошибка получения VNC для VM {vmid} на {node}: {e}")
+                return {}
+
+        def resize_vm_disk(self, node: str, vmid: int, disk: str, size: str) -> bool:
+            """
+            Изменить размер диска VM
+            
+            Args:
+                node: Имя ноды
+                vmid: ID виртуальной машины
+                disk: Имя диска (scsi0, virtio0 и т.д.)
+                size: Новый размер (+10G, 50G)
+            
+            Returns:
+                True при успехе
+            """
+            if not self.proxmox:
+                return False
+            
+            try:
+                resize_upid = self.proxmox.nodes(node).qemu(vmid).resize.put(disk=disk, size=size)
+                # Wait for resize task to complete to avoid lock conflicts
+                if isinstance(resize_upid, str):
+                    self.wait_for_task(node, resize_upid, timeout=120)
+                logger.info(f"Размер диска {disk} VM {vmid} изменен на {size}")
+                return True
+            except Exception as e:
+                logger.error(f"Ошибка изменения размера диска VM {vmid}: {e}")
+                return False
+