admin.py

from django.contrib import admin
from django.contrib.auth.models import User
from django.forms import ModelForm
from django.utils.translation import gettext_lazy as _
from openwisp_utils.admin import (
    AlwaysHasChangedMixin,
    HelpTextStackedInline,
    ReadOnlyAdmin,
    ReceiveUrlAdmin,
    Select2AdminMixin,
    TimeReadonlyAdminMixin,
    UUIDAdmin,
)
from openwisp_utils.admin_theme.filters import (
    AutocompleteFilter,
    InputFilter,
    SimpleInputFilter,
)

from .models import (
    Book,
    Operator,
    OrganizationRadiusSettings,
    Project,
    RadiusAccounting,
    Shelf,
)

admin.site.unregister(User)


class AutoShelfFilter(AutocompleteFilter):
    title = _("shelf")
    field_name = "shelf"
    parameter_name = "shelf__id"


@admin.register(User)
class UserAdmin(admin.ModelAdmin):
    list_display = ["username", "is_staff", "is_superuser", "is_active"]
    list_filter = [
        ("username", InputFilter),
        ("shelf", InputFilter),
        "is_staff",
        "is_superuser",
        "is_active",
    ]
    search_fields = ("username",)


@admin.register(Book)
class BookAdmin(admin.ModelAdmin):
    list_filter = [AutoShelfFilter, "name"]
    search_fields = ["name"]


@admin.register(Operator)
class OperatorAdmin(admin.ModelAdmin):
    list_display = ["first_name", "last_name"]
    list_filter = ["project__name"]  # DO NOT CHANGE: used for testing filters


@admin.register(RadiusAccounting)
class RadiusAccountingAdmin(ReadOnlyAdmin):
    list_display = ["session_id", "username"]
    fields = ["session_id", "username"]


class OperatorForm(AlwaysHasChangedMixin, ModelForm):
    pass


class OperatorInline(HelpTextStackedInline):
    model = Operator
    form = OperatorForm
    extra = 0
    help_text = {
        "text": _("Only added operators will have permission to access the project."),
        "documentation_url": "https://github.com/openwisp/openwisp-utils/",
    }


@admin.register(Project)
class ProjectAdmin(UUIDAdmin, ReceiveUrlAdmin):
    inlines = [OperatorInline]
    list_display = ("name",)
    fields = ("uuid", "name", "key", "receive_url")
    readonly_fields = ("uuid", "receive_url")
    receive_url_name = "receive_project"


class ShelfFilter(SimpleInputFilter):
    parameter_name = "shelf"
    title = _("Shelf")

    def queryset(self, request, queryset):
        if self.value() is not None:
            return queryset.filter(name__icontains=self.value())


class ReverseBookFilter(AutocompleteFilter):
    title = _("Book")
    field_name = "book"
    parameter_name = "book"


class AutoOwnerFilter(AutocompleteFilter):
    title = _("owner")
    field_name = "owner"
    parameter_name = "owner_id"


@admin.register(Shelf)
class ShelfAdmin(Select2AdminMixin, TimeReadonlyAdminMixin, admin.ModelAdmin):
    # DO NOT CHANGE: used for testing filters
    list_filter = [
        ShelfFilter,
        ["books_type", InputFilter],
        ["id", InputFilter],
        AutoOwnerFilter,
        "books_type",
        ReverseBookFilter,
    ]
    search_fields = ["name"]
    select2_fields = ("books_type",)


@admin.register(OrganizationRadiusSettings)
class OrganizationRadiusSettingsAdmin(admin.ModelAdmin):
    pass