from __future__ import annotations

from sqladmin import ModelView

from app.models.mariadb.device import Device, DeviceGroup


class DeviceAdmin(ModelView, model=Device):
    column_list = [
        Device.id, Device.device_uid, Device.name, Device.device_type,
        Device.status, Device.last_seen_at, Device.created_at,
    ]
    column_searchable_list = [Device.device_uid, Device.name]
    column_sortable_list = [Device.id, Device.name, Device.status, Device.created_at]
    column_default_sort = ("id", True)
    can_create = True
    can_edit = True
    can_delete = False
    name = "Device"
    name_plural = "Devices"
    icon = "fa-solid fa-microchip"


class DeviceGroupAdmin(ModelView, model=DeviceGroup):
    column_list = [DeviceGroup.id, DeviceGroup.name, DeviceGroup.description]
    column_searchable_list = [DeviceGroup.name]
    can_create = True
    can_edit = True
    can_delete = True
    name = "Device Group"
    name_plural = "Device Groups"
    icon = "fa-solid fa-layer-group"