from django.contrib import admin
import nested_admin

from .models import (
    Device,
    DeviceCategory,
    DeviceInNet,
    DeviceManufacturer,
    HardwareModel,
    Warranty,
    WarrantyType,
)


class DeviceInNetInline(nested_admin.NestedStackedInline):  # pylint: disable=no-member
    model = DeviceInNet
    extra = 0
    verbose_name_plural = "Nets"


class DeviceAdmin(nested_admin.NestedModelAdmin):  # pylint: disable=no-member
    inlines = (DeviceInNetInline,)


admin.site.register(Device, DeviceAdmin)
admin.site.register(DeviceCategory)
admin.site.register(DeviceInNet)
admin.site.register(DeviceManufacturer)
admin.site.register(HardwareModel)
admin.site.register(Warranty)
admin.site.register(WarrantyType)