from Controller import mysql_connect
import psutil
import platform
import math

ipadress = "'172.18.1.6'"

def get_cpu_sql(cpu):
    query = f"SELECT `Prozessor-Anzahl` FROM `CPU-Liste` WHERE `CPU-Name`='{cpu}'"
    return mysql_connect.get_cpu(query,"Stines-GmbH",cpu)

def get_ram_info():
    ram_info = psutil.virtual_memory()
    total_ram = ram_info.total / (1024 ** 3)  # In Gigabytes
    return math.ceil(total_ram)

def get_hdd_info():
    total_disk_size = 0
    disk_partitions = psutil.disk_partitions()
    for partition in disk_partitions:
    # Wenn du das Laufwerk C: ignorieren möchtest
        if partition.device != 'C:\\':
            partition_info = psutil.disk_usage(partition.mountpoint)
            total_disk_size += partition_info.total / (1024 ** 3)
    return math.ceil(total_disk_size)

# Informationen über die CPU
def set_system_info(name,cpu,ipadress):
    query = f"UPDATE {name} SET RAM={get_ram_info()}, Prozessor={get_cpu_sql()}, CPU='{platform.processor()}' WHERE `IP-Adresse`={ipadress}"
    return mysql_connect.update_cpu(query,"Stines-GmbH",get_cpu_sql())