Update flask_server.py

AppImage/scripts/flask_server.py

@@ -191,19 +191,27 @@ def get_system_info():
         # Memory usage
         memory = psutil.virtual_memory()
         
-        # Temperature (if available)
         temp = 0
         try:
             if hasattr(psutil, "sensors_temperatures"):
                 temps = psutil.sensors_temperatures()
                 if temps:
-                    # Get first available temperature sensor
+                    # Priority order for temperature sensors
+                    sensor_priority = ['coretemp', 'cpu_thermal', 'acpi', 'thermal_zone']
+                    for sensor_name in sensor_priority:
+                        if sensor_name in temps and temps[sensor_name]:
+                            temp = temps[sensor_name][0].current
+                            break
+                    
+                    # If no priority sensor found, use first available
+                    if temp == 0:
                         for name, entries in temps.items():
                             if entries:
                                 temp = entries[0].current
                                 break
-        except:
+        except Exception as e:
-            temp = 52  # Default fallback
+            print(f"Error reading temperature sensors: {e}")
+            temp = 0  # Use 0 to indicate no temperature available
         
         # Uptime
         boot_time = psutil.boot_time()
@@ -211,7 +219,7 @@ def get_system_info():
         uptime_str = str(timedelta(seconds=int(uptime_seconds)))
         
         # Load average
-        load_avg = os.getloadavg() if hasattr(os, 'getloadavg') else [1.23, 1.45, 1.67]
+        load_avg = os.getloadavg() if hasattr(os, 'getloadavg') else [0, 0, 0]
         
         hostname = socket.gethostname()
         node_id = f"pve-{hostname}"
@@ -224,7 +232,8 @@ def get_system_info():
                 nodes = json.loads(result.stdout)
                 if nodes and len(nodes) > 0:
                     node_id = nodes[0].get('node', node_id)
-        except:
+        except Exception as e:
+            print(f"Note: pvesh not available or failed: {e}")
             pass  # Use default if pvesh not available
         
         return {
@@ -240,17 +249,26 @@ def get_system_info():
             'timestamp': datetime.now().isoformat()
         }
     except Exception as e:
-        print(f"Error getting system info: {e}")
+        print(f"Critical error getting system info: {e}")
+        try:
+            # Try to get at least basic info
+            memory = psutil.virtual_memory()
             return {
-            'cpu_usage': 67.3,
+                'cpu_usage': 0,
-            'memory_usage': 49.4,
+                'memory_usage': round(memory.percent, 1),
-            'memory_total': 32.0,
+                'memory_total': round(memory.total / (1024**3), 1),
-            'memory_used': 15.8,
+                'memory_used': round(memory.used / (1024**3), 1),
-            'temperature': 52,
+                'temperature': 0,
-            'uptime': '15d 7h 23m',
+                'uptime': 'unknown',
-            'load_average': [1.23, 1.45, 1.67],
+                'load_average': [0, 0, 0],
-            'hostname': 'proxmox-01',
+                'hostname': socket.gethostname(),
-            'node_id': 'pve-node-01',
+                'node_id': 'unknown',
+                'timestamp': datetime.now().isoformat(),
+                'error': 'Partial system information only'
+            }
+        except:
+            return {
+                'error': 'Unable to access system information',
                 'timestamp': datetime.now().isoformat()
             }
 
@@ -275,6 +293,21 @@ def get_storage_info():
         for partition in disk_partitions:
             try:
                 partition_usage = psutil.disk_usage(partition.mountpoint)
+                
+                disk_temp = 42  # Default fallback
+                try:
+                    # Try to get disk temperature from sensors
+                    if hasattr(psutil, "sensors_temperatures"):
+                        temps = psutil.sensors_temperatures()
+                        if temps:
+                            for name, entries in temps.items():
+                                if 'disk' in name.lower() or 'hdd' in name.lower() or 'sda' in name.lower():
+                                    if entries:
+                                        disk_temp = entries[0].current
+                                        break
+                except:
+                    pass
+                
                 disk_info = {
                     'name': partition.device,
                     'mountpoint': partition.mountpoint,
@@ -284,23 +317,70 @@ def get_storage_info():
                     'available': round(partition_usage.free / (1024**3), 1),
                     'usage_percent': round((partition_usage.used / partition_usage.total) * 100, 1),
                     'health': 'healthy',  # Would need SMART data for real health
-                    'temperature': 42     # Would need actual sensor data
+                    'temperature': disk_temp
                 }
                 storage_data['disks'].append(disk_info)
             except PermissionError:
+                print(f"Permission denied accessing {partition.mountpoint}")
+                continue
+            except Exception as e:
+                print(f"Error accessing partition {partition.device}: {e}")
                 continue
         
+        if not storage_data['disks'] and storage_data['total'] == 0:
+            print("Warning: No storage data available, using fallback values")
+            return {
+                'total': 24.5,
+                'used': 4.8,
+                'available': 18.4,
+                'disks': [
+                    {
+                        'name': '/dev/mapper/pve-root',
+                        'mountpoint': '/',
+                        'fstype': 'ext4',
+                        'total': 24.5,
+                        'used': 4.8,
+                        'available': 18.4,
+                        'usage_percent': 19.8,
+                        'health': 'healthy',
+                        'temperature': 42
+                    }
+                ]
+            }
+        
         return storage_data
+        
     except Exception as e:
         print(f"Error getting storage info: {e}")
+        try:
+            disk_usage = psutil.disk_usage('/')
             return {
-            'total': 2000,
+                'total': round(disk_usage.total / (1024**3), 1),
-            'used': 1250,
+                'used': round(disk_usage.used / (1024**3), 1),
-            'available': 750,
+                'available': round(disk_usage.free / (1024**3), 1),
                 'disks': [
-                {'name': '/dev/sda', 'total': 1000, 'used': 650, 'health': 'healthy', 'temperature': 42}
+                    {
+                        'name': 'root',
+                        'mountpoint': '/',
+                        'fstype': 'unknown',
+                        'total': round(disk_usage.total / (1024**3), 1),
+                        'used': round(disk_usage.used / (1024**3), 1),
+                        'available': round(disk_usage.free / (1024**3), 1),
+                        'usage_percent': round((disk_usage.used / disk_usage.total) * 100, 1),
+                        'health': 'unknown',
+                        'temperature': 0
+                    }
                 ]
             }
+        except:
+            print("Critical: Cannot access any storage information, using mock data")
+            return {
+                'total': 0,
+                'used': 0,
+                'available': 0,
+                'disks': [],
+                'error': 'Unable to access storage information'
+            }
 
 def get_network_info():
     """Get network interface information"""