from django.contrib import admin
from import_export.admin import ImportExportMixin
from .models import *
from simple_history.admin import SimpleHistoryAdmin
from import_export.admin import ImportExportMixin
from . import models


@admin.register(models.Branch)
class BranchAdmin(ImportExportMixin, SimpleHistoryAdmin, admin.ModelAdmin):
    list_display = (
        "name",
        "description",
    )


@admin.register(models.Area)
class AreaAdmin(ImportExportMixin, SimpleHistoryAdmin, admin.ModelAdmin):
    list_display = (
        "name",
        "description",
    )
    list_filter = ("branch",)


@admin.register(models.InventoryTaking)
class InventoryTaking(ImportExportMixin, SimpleHistoryAdmin, admin.ModelAdmin):
    list_display = (
        "name",
        "description",
        "begin",
        "end",
    )


@admin.register(models.Item)
class ItemAmin(ImportExportMixin, SimpleHistoryAdmin, admin.ModelAdmin):
    list_display = ("inventory_id", "label_id", "name", "serial_nbr")
    list_filter = ("inventory_taking_id",)