from django.contrib import admin
from apps.em.models import Ecate, Equipment
# Register your models here.


@admin.register(Ecate)
class EcateAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'code', 'type',
                    'is_for_safe', 'is_for_enp', 'is_car')


@admin.register(Equipment)
class EquipmentAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'type',
                    'cate', 'state', 'is_online', 'running_state')