import psutil
from apps.monitor.models import AuditLog
from apps.system.models import User
from datetime import datetime
from apps.utils.tools import compare_values
from apps.utils.models import get_model_info


def delete_auditlog(model, instance_id):
    """
    删除其对应的审计记录
    """
    model_name = get_model_info(model)
    AuditLog.objects.filter(model_name=model_name,
                            instance_id=instance_id).delete()


def create_auditlog(action: str, instance,  val_new: dict, val_old: dict = None, change_reason: str = '', delete_time: datetime = None, delete_user: User = None):
    """
    生成审计日志
    action: create/update/delete/其他action
    """
    app_label_model_name = get_model_info(instance)
    if val_old is None:
        val_old = {}
    difference = []
    has_changed = False
    if action == 'create':
        has_changed = True
        change_user = instance.create_by
        change_time = instance.create_time
    elif action == 'delete':
        has_changed = True
        change_user = delete_user if delete_user else instance.update_by
        change_time = delete_time if delete_time else instance.update_time
    else:
        change_user = instance.update_by
        change_time = instance.update_time
        for k, v in val_new.items():
            if k not in ['create_by', 'update_by', 'create_time', 'update_time', 'id']:
                if k not in val_old:
                    difference.append(
                        {'field': k, 'action': 'create', 'val_old': None, 'val_new': v})
                elif not compare_values(val_new.get(k), val_old.get(k), ignore_order=True):
                    difference.append(
                        {'field': k, 'action': 'update', 'val_old': val_old[k], 'val_new': v})
        if difference:
            has_changed = True
    if has_changed:
        AuditLog.objects.create(
            action=action,
            model_name=app_label_model_name,
            instance_id=instance.id,
            val_new=val_new,
            difference=difference,
            change_reason=change_reason,
            change_user=change_user,
            change_time=change_time
        )


class ServerService:
    @classmethod
    def get_memory_dict(cls):
        ret = {}
        memory = psutil.virtual_memory()
        ret['total'] = round(memory.total/1024/1024/1024, 2)
        ret['used'] = round(memory.used/1024/1024/1024, 2)
        ret['percent'] = memory.percent
        return ret

    @classmethod
    def get_cpu_dict(cls):
        ret = {}
        ret['lcount'] = psutil.cpu_count()
        ret['count'] = psutil.cpu_count(logical=False)
        ret['percent'] = psutil.cpu_percent(interval=1)
        return ret

    @classmethod
    def get_disk_dict(cls):
        ret = {}
        disk = psutil.disk_usage('/')
        ret['total'] = round(disk.total/1024/1024/1024, 2)
        ret['used'] = round(disk.used/1024/1024/1024, 2)
        ret['percent'] = disk.percent
        return ret

    @classmethod
    def get_full(cls):
        return {'cpu': cls.get_cpu_dict(), 'memory': cls.get_memory_dict(), 'disk': cls.get_disk_dict()}