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", "running_state")