Sith/core/views/widgets/select.py

from django.contrib.staticfiles.storage import staticfiles_storage
from django.db.models import Model
from django.forms import Select, SelectMultiple
from ninja import ModelSchema

from core.models import Group, User
from core.schemas import GroupSchema, UserProfileSchema


class AutoCompleteSelectMixin:
    component_name = "autocomplete-select"
    template_name = "core/widgets/autocomplete_select.jinja"
    model: Model | None = None
    schema: ModelSchema | None = None
    pk = "id"

    def __init__(self, attrs=None, choices=()):
        if self.is_ajax:
            choices = ()  # Avoid computing anything when in ajax mode
        super().__init__(attrs=attrs, choices=choices)

    @property
    def is_ajax(self):
        return self.model and self.schema

    def optgroups(self, name, value, attrs=None):
        """Don't create option groups when doing ajax"""
        if self.is_ajax:
            return []
        return super().optgroups(name, value, attrs=attrs)

    def get_context(self, name, value, attrs):
        context = super().get_context(name, value, attrs)
        context["component"] = self.component_name
        context["statics"] = {
            "js": staticfiles_storage.url(
                "webpack/core/components/ajax-select-index.ts"
            ),
            "csss": [
                staticfiles_storage.url(
                    "webpack/core/components/ajax-select-index.css"
                ),
                staticfiles_storage.url("core/components/ajax-select.scss"),
            ],
        }
        if self.is_ajax:
            context["selected"] = [
                self.schema.from_orm(obj).json()
                for obj in self.model.objects.filter(
                    **{f"{self.pk}__in": context["widget"]["value"]}
                ).all()
            ]
        return context


class AutoCompleteSelect(AutoCompleteSelectMixin, Select): ...


class AutoCompleteSelectMultiple(AutoCompleteSelectMixin, SelectMultiple): ...


class AutoCompleteSelectUser(AutoCompleteSelectMixin, Select):
    component_name = "user-ajax-select"
    model = User
    schema = UserProfileSchema


class AutoCompleteSelectMultipleUser(AutoCompleteSelectMixin, SelectMultiple):
    component_name = "user-ajax-select"
    model = User
    schema = UserProfileSchema


class AutoCompleteSelectGroup(AutoCompleteSelectMixin, Select):
    component_name = "group-ajax-select"
    model = Group
    schema = GroupSchema


class AutoCompleteSelectMultipleGroup(AutoCompleteSelectMixin, SelectMultiple):
    component_name = "group-ajax-select"
    model = Group
    schema = GroupSchema