Mirrors/ProxMenux
1
0
Fork 0
mirror of https://github.com/MacRimi/ProxMenux.git synced 2025-11-18 03:26:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update flask_server.py

Browse Source
This commit is contained in:
MacRimi
2025-10-17 17:04:55 +02:00
parent c40bb6a4d5
commit ef973df7c9
1 changed files with 74 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

132
AppImage/scripts/flask_server.py
View File

@@ -1169,78 +1169,86 @@ def get_smart_data(disk_name):
print(f"[v0] ===== Final SMART data for /dev/{disk_name}: {smart_data} =====") print(f"[v0] ===== Final SMART data for /dev/{disk_name}: {smart_data} =====")
return smart_data return smart_data
# START OF CHANGES FOR get_proxmox_storage
def get_proxmox_storage(): def get_proxmox_storage():
"""Get Proxmox storage information using pvesm status""" """Get Proxmox storage information using pvesh (filtered by local node)"""
try: try:
print("[v0] Getting Proxmox storage with pvesm status...") local_node = socket.gethostname()
result = subprocess.run(['pvesm', 'status'], capture_output=True, text=True, timeout=10) print(f"[v0] Getting Proxmox storage for local node: {local_node}")
result = subprocess.run(['pvesh', 'get', '/cluster/resources', '--type', 'storage', '--output-format', 'json'],
capture_output=True, text=True, timeout=10)
if result.returncode != 0: if result.returncode != 0:
print(f"[v0] pvesm status failed with return code {result.returncode}") print(f"[v0] pvesh command failed with return code {result.returncode}")
print(f"[v0] stderr: {result.stderr}") print(f"[v0] stderr: {result.stderr}")
return { return {
'error': 'pvesm command not available or failed', 'error': 'pvesh command not available or failed',
'storage': [] 'storage': []
} }
storage_list = [] storage_list = []
lines = result.stdout.strip().split('\n') resources = json.loads(result.stdout)
# Parse each storage line for resource in resources:
for line in lines[1:]: # Skip header node = resource.get('node', '')
parts = line.split()
if len(parts) >= 6: # Filtrar solo storage del nodo local
name = parts[0] if node != local_node:
storage_type = parts[1] print(f"[v0] Skipping storage {resource.get('storage')} from remote node: {node}")
status = parts[2] continue
try: name = resource.get('storage', 'unknown')
total = int(parts[3]) storage_type = resource.get('plugintype', 'unknown')
used = int(parts[4]) status = resource.get('status', 'unknown')
available = int(parts[5])
except (ValueError, IndexError): try:
print(f"[v0] Skipping storage {name} - invalid numeric data") total = int(resource.get('maxdisk', 0))
continue used = int(resource.get('disk', 0))
available = total - used if total > 0 else 0
try: except (ValueError, TypeError):
percent = float(parts[6].rstrip('%')) if len(parts) > 6 and parts[6] != 'N/A' else 0.0 print(f"[v0] Skipping storage {name} - invalid numeric data")
except (ValueError, IndexError): continue
percent = 0.0
# Si total es 0, significa que hay un error de conexión o el datastore no está disponible
# Si total es 0, significa que hay un error de conexión o el datastore no está disponible if total == 0:
if total == 0: print(f"[v0] Skipping storage {name} - invalid data (total=0, likely connection error)")
print(f"[v0] Skipping storage {name} - invalid data (total=0, likely connection error)") continue
continue
# Si el status es "inactive", también lo omitimos
# Si el status es "inactive", también lo omitimos if status.lower() != "available":
if status.lower() == "inactive": print(f"[v0] Skipping storage {name} - status is not available: {status}")
print(f"[v0] Skipping storage {name} - status is inactive") continue
continue
# Calcular porcentaje
# Convert bytes to GB percent = (used / total * 100) if total > 0 else 0.0
total_gb = round(total / (1024**2), 2)
used_gb = round(used / (1024**2), 2) # Convert bytes to GB
available_gb = round(available / (1024**2), 2) total_gb = round(total / (1024**3), 2)
used_gb = round(used / (1024**3), 2)
storage_info = { available_gb = round(available / (1024**3), 2)
'name': name,
'type': storage_type, storage_info = {
'status': status, 'name': name,
'total': total_gb, 'type': storage_type,
'used': used_gb, 'status': 'active', # Normalizar status para compatibilidad con frontend
'available': available_gb, 'total': total_gb,
'percent': round(percent, 2) 'used': used_gb,
} 'available': available_gb,
'percent': round(percent, 2),
print(f"[v0] Found storage: {name} ({storage_type}) - {used_gb}/{total_gb} GB ({percent}%)") 'node': node # Incluir información del nodo
storage_list.append(storage_info) }
print(f"[v0] Found storage on {node}: {name} ({storage_type}) - {used_gb}/{total_gb} GB ({percent:.2f}%)")
storage_list.append(storage_info)
print(f"[v0] Total storage entries on local node {local_node}: {len(storage_list)}")
return {'storage': storage_list} return {'storage': storage_list}
except FileNotFoundError: except FileNotFoundError:
print("[v0] pvesm command not found - Proxmox not installed or not in PATH") print("[v0] pvesh command not found - Proxmox not installed or not in PATH")
return { return {
'error': 'pvesm command not found - Proxmox not installed', 'error': 'pvesh command not found - Proxmox not installed',
'storage': [] 'storage': []
} }
except Exception as e: except Exception as e:
@@ -1251,6 +1259,7 @@ def get_proxmox_storage():
'error': f'Unable to get Proxmox storage: {str(e)}', 'error': f'Unable to get Proxmox storage: {str(e)}',
'storage': [] 'storage': []
} }
# END OF CHANGES FOR get_proxmox_storage
def get_interface_type(interface_name): def get_interface_type(interface_name):
"""Detect the type of network interface""" """Detect the type of network interface"""
@@ -1814,7 +1823,10 @@ def get_ups_info():
ups_data['status'] = value ups_data['status'] = value
elif key == 'battery.charge': elif key == 'battery.charge':
ups_data['battery_charge'] = f"{value}%" ups_data['battery_charge'] = f"{value}%"
ups_data['battery_charge_raw'] = float(value) try:
ups_data['battery_charge_raw'] = float(value)
except ValueError:
ups_data['battery_charge_raw'] = None
elif key == 'battery.runtime': elif key == 'battery.runtime':
try: try:
runtime_sec = int(value) runtime_sec = int(value)
@@ -1823,13 +1835,17 @@ def get_ups_info():
ups_data['time_left_seconds'] = runtime_sec ups_data['time_left_seconds'] = runtime_sec
except ValueError: except ValueError:
ups_data['time_left'] = value ups_data['time_left'] = value
ups_data['time_left_seconds'] = None
elif key == 'battery.voltage': elif key == 'battery.voltage':
ups_data['battery_voltage'] = f"{value}V" ups_data['battery_voltage'] = f"{value}V"
elif key == 'battery.date': elif key == 'battery.date':
ups_data['battery_date'] = value ups_data['battery_date'] = value
elif key == 'ups.load': elif key == 'ups.load':
ups_data['load_percent'] = f"{value}%" ups_data['load_percent'] = f"{value}%"
ups_data['load_percent_raw'] = float(value) try:
ups_data['load_percent_raw'] = float(value)
except ValueError:
ups_data['load_percent_raw'] = None
elif key == 'input.voltage': elif key == 'input.voltage':
ups_data['input_voltage'] = f"{value}V" ups_data['input_voltage'] = f"{value}V"
elif key == 'input.frequency': elif key == 'input.frequency':