django-unfold 0.48.0__py3-none-any.whl → 0.49.1__py3-none-any.whl

unfold/mixins/action_model_admin.py

@@ -0,0 +1,330 @@
+from typing import Any, Callable, Optional, Union
+
+from django.db.models import Model
+from django.forms import Form
+from django.http import HttpRequest
+from django.template.response import TemplateResponse
+from django.urls import reverse
+
+from unfold.dataclasses import UnfoldAction
+from unfold.exceptions import UnfoldException
+
+
+class ActionModelAdminMixin:
+    """
+    Adds support for various ModelAdmin actions (list, detail, row, submit line)
+    """
+
+    actions_list = ()  # Displayed in changelist at the top
+    actions_row = ()  # Displayed in changelist for each row in the table
+    actions_detail = ()  # Displayed in changeform at the top
+    actions_submit_line = ()  # Displayed in changeform in the submit line (form buttons)
+
+    def changelist_view(
+        self, request: HttpRequest, extra_context: Optional[dict[str, str]] = None
+    ) -> TemplateResponse:
+        """
+        Changelist contains `actions_list` and `actions_row` custom actions. In case of `actions_row` they
+        are displayed in the each row of the table.
+        """
+        extra_context = extra_context or {}
+
+        actions_row = [
+            {
+                "title": action.description,
+                "icon": action.icon,
+                "attrs": action.method.attrs,
+                # This is just a path name as string and in template is used in {% url %} tag
+                # with custom instance pk value
+                "raw_path": f"{self.admin_site.name}:{action.action_name}",
+            }
+            for action in self.get_actions_row(request)
+        ]
+
+        # `actions_list` may contain custom structure with dropdowns so it is needed
+        # to use `_get_actions_navigation` to build the final structure for the template
+        actions_list = self._get_actions_navigation(
+            self.actions_list, self.get_actions_list(request)
+        )
+
+        extra_context.update(
+            {
+                "actions_list": actions_list,
+                "actions_row": actions_row,
+            }
+        )
+
+        return super().changelist_view(request, extra_context)
+
+    def changeform_view(
+        self,
+        request: HttpRequest,
+        object_id: Optional[str] = None,
+        form_url: str = "",
+        extra_context: Optional[dict[str, bool]] = None,
+    ) -> Any:
+        """
+        Changeform contains `actions_submit_line` and `actions_detail` custom actions.
+        """
+        extra_context = extra_context or {}
+
+        if object_id:
+            # `actions_submit_line` is a list of actions that are displayed in the submit line they
+            # are displayed as form buttons
+            actions_submit_line = self.get_actions_submit_line(request, object_id)
+
+            # `actions_detail` may contain custom structure with dropdowns so it is needed
+            # to use `_get_actions_navigation` to build the final structure for the template
+            actions_detail = self._get_actions_navigation(
+                self.actions_detail,
+                self.get_actions_detail(request, object_id),
+                object_id,
+            )
+
+            extra_context.update(
+                {
+                    "actions_submit_line": actions_submit_line,
+                    "actions_detail": actions_detail,
+                }
+            )
+
+        return super().changeform_view(request, object_id, form_url, extra_context)
+
+    def save_model(
+        self, request: HttpRequest, obj: Model, form: Form, change: Any
+    ) -> None:
+        """
+        When saving object, run all appropriate actions from `actions_submit_line`
+        """
+        super().save_model(request, obj, form, change)
+
+        # After saving object, check if any button from `actions_submit_line` was pressed
+        # and call the corresponding method
+        for action in self.get_actions_submit_line(request, obj.pk):
+            if action.action_name not in request.POST:
+                continue
+
+            action.method(request, obj)
+
+    def get_unfold_action(self, action: str) -> UnfoldAction:
+        """
+        Converts action name into UnfoldAction object.
+        """
+        method = self._get_instance_method(action)
+
+        return UnfoldAction(
+            action_name=f"{self.model._meta.app_label}_{self.model._meta.model_name}_{action}",
+            method=method,
+            description=self._get_action_description(method, action),
+            path=getattr(method, "url_path", action),
+            attrs=method.attrs if hasattr(method, "attrs") else None,
+            icon=method.icon if hasattr(method, "icon") else None,
+        )
+
+    def get_actions_list(self, request: HttpRequest) -> list[UnfoldAction]:
+        """
+        Filters `actions_list` by permissions and returns list of UnfoldAction objects.
+        """
+        return self._filter_unfold_actions_by_permissions(
+            request, self._get_base_actions_list()
+        )
+
+    def get_actions_detail(
+        self, request: HttpRequest, object_id: int
+    ) -> list[UnfoldAction]:
+        """
+        Filters `actions_detail` by permissions and returns list of UnfoldAction objects.
+        """
+        return self._filter_unfold_actions_by_permissions(
+            request, self._get_base_actions_detail(), object_id
+        )
+
+    def get_actions_row(self, request: HttpRequest) -> list[UnfoldAction]:
+        """
+        Filters `actions_row` by permissions and returns list of UnfoldAction objects.
+        """
+        return self._filter_unfold_actions_by_permissions(
+            request, self._get_base_actions_row()
+        )
+
+    def get_actions_submit_line(
+        self, request: HttpRequest, object_id: int
+    ) -> list[UnfoldAction]:
+        """
+        Filters `actions_submit_line` by permissions and returns list of UnfoldAction objects.
+        """
+        return self._filter_unfold_actions_by_permissions(
+            request, self._get_base_actions_submit_line(), object_id
+        )
+
+    def _extract_action_names(self, actions: list[Union[str, dict]]) -> list[str]:
+        """
+        Gets the list of only actions names from the actions structure provided in ModelAdmin
+        """
+        results = []
+
+        for action in actions or []:
+            if isinstance(action, dict) and "items" in action:
+                results.extend(action["items"])
+            else:
+                results.append(action)
+
+        return results
+
+    def _get_base_actions_list(self) -> list[UnfoldAction]:
+        """
+        Returns list of UnfoldAction objects for `actions_list`.
+        """
+        return [
+            self.get_unfold_action(action)
+            for action in self._extract_action_names(self.actions_list)
+        ]
+
+    def _get_base_actions_detail(self) -> list[UnfoldAction]:
+        """
+        Returns list of UnfoldAction objects for `actions_detail`.
+        """
+        return [
+            self.get_unfold_action(action)
+            for action in self._extract_action_names(self.actions_detail) or []
+        ]
+
+    def _get_base_actions_row(self) -> list[UnfoldAction]:
+        """
+        Returns list of UnfoldAction objects for `actions_row`.
+        """
+        return [
+            self.get_unfold_action(action)
+            for action in self._extract_action_names(self.actions_row) or []
+        ]
+
+    def _get_base_actions_submit_line(self) -> list[UnfoldAction]:
+        """
+        Returns list of UnfoldAction objects for `actions_submit_line`.
+        """
+        return [
+            self.get_unfold_action(action)
+            for action in self._extract_action_names(self.actions_submit_line) or []
+        ]
+
+    def _get_instance_method(self, method_name: str) -> Callable:
+        """
+        Searches for method on self instance based on method_name and returns it if it exists.
+        If it does not exist or is not callable, it raises UnfoldException
+        """
+        try:
+            method = getattr(self, method_name)
+        except AttributeError as e:
+            raise UnfoldException(
+                f"Method {method_name} specified does not exist on current object"
+            ) from e
+
+        if not callable(method):
+            raise UnfoldException(f"{method_name} is not callable")
+
+        return method
+
+    def _get_actions_navigation(
+        self,
+        provided_actions: list[Union[str, dict]],
+        allowed_actions: list[UnfoldAction],
+        object_id: Optional[str] = None,
+    ) -> list[Union[str, dict]]:
+        """
+        Builds navigation structure for the actions which is going to be provided to the template.
+        """
+        navigation = []
+
+        def get_action_by_name(name: str) -> UnfoldAction:
+            """
+            Searches for an action in allowed_actions by its name.
+            """
+            for action in allowed_actions:
+                full_action_name = (
+                    f"{self.model._meta.app_label}_{self.model._meta.model_name}_{name}"
+                )
+
+                if action.action_name == full_action_name:
+                    return action
+
+        def get_action_path(action: UnfoldAction) -> str:
+            """
+            Returns the URL path for an action.
+            """
+            path_name = f"{self.admin_site.name}:{action.action_name}"
+
+            if object_id:
+                return reverse(path_name, args=(object_id,))
+
+            return reverse(path_name)
+
+        def get_action_attrs(action: UnfoldAction) -> dict:
+            """
+            Returns the attributes for an action which will be used in the template.
+            """
+            return {
+                "title": action.description,
+                "icon": action.icon,
+                "attrs": action.method.attrs,
+                "path": get_action_path(action),
+            }
+
+        def build_dropdown(nav_item: dict) -> Optional[dict]:
+            """
+            Builds a dropdown structure for the action.
+            """
+            dropdown = {
+                "title": nav_item["title"],
+                "icon": nav_item.get("icon"),
+                "items": [],
+            }
+
+            for child in nav_item["items"]:
+                if action := get_action_by_name(child):
+                    dropdown["items"].append(get_action_attrs(action))
+
+            if len(dropdown["items"]) > 0:
+                return dropdown
+
+        for nav_item in provided_actions:
+            if isinstance(nav_item, str):
+                if action := get_action_by_name(nav_item):
+                    navigation.append(get_action_attrs(action))
+            elif isinstance(nav_item, dict):
+                if dropdown := build_dropdown(nav_item):
+                    navigation.append(dropdown)
+
+        return navigation
+
+    def _filter_unfold_actions_by_permissions(
+        self,
+        request: HttpRequest,
+        actions: list[UnfoldAction],
+        object_id: Optional[Union[int, str]] = None,
+    ) -> list[UnfoldAction]:
+        """
+        Filters out actions that the user doesn't have access to.
+        """
+        filtered_actions = []
+
+        for action in actions:
+            if not hasattr(action.method, "allowed_permissions"):
+                filtered_actions.append(action)
+                continue
+
+            permission_checks = (
+                getattr(self, f"has_{permission}_permission")
+                for permission in action.method.allowed_permissions
+            )
+
+            if object_id:
+                if all(
+                    has_permission(request, object_id)
+                    for has_permission in permission_checks
+                ):
+                    filtered_actions.append(action)
+            else:
+                if all(has_permission(request) for has_permission in permission_checks):
+                    filtered_actions.append(action)
+
+        return filtered_actions

unfold/mixins/base_model_admin.py

@@ -0,0 +1,110 @@
+import copy
+from typing import Optional
+
+from django.contrib.admin.sites import AdminSite
+from django.contrib.admin.widgets import RelatedFieldWidgetWrapper
+from django.db import models
+from django.db.models.fields import Field
+from django.db.models.fields.related import ForeignKey, ManyToManyField
+from django.forms.fields import TypedChoiceField
+from django.forms.models import ModelChoiceField, ModelMultipleChoiceField
+from django.forms.widgets import SelectMultiple
+from django.http import HttpRequest
+from django.utils.translation import gettext_lazy as _
+
+from unfold import widgets
+from unfold.overrides import FORMFIELD_OVERRIDES
+
+
+class BaseModelAdminMixin:
+    def __init__(self, model: models.Model, admin_site: AdminSite) -> None:
+        overrides = copy.deepcopy(FORMFIELD_OVERRIDES)
+
+        for k, v in self.formfield_overrides.items():
+            overrides.setdefault(k, {}).update(v)
+
+        self.formfield_overrides = overrides
+
+        super().__init__(model, admin_site)
+
+    def formfield_for_choice_field(
+        self, db_field: Field, request: HttpRequest, **kwargs
+    ) -> TypedChoiceField:
+        if "widget" not in kwargs:
+            if db_field.name in self.radio_fields:
+                kwargs["widget"] = widgets.UnfoldAdminRadioSelectWidget(
+                    radio_style=self.radio_fields[db_field.name]
+                )
+            else:
+                kwargs["widget"] = widgets.UnfoldAdminSelectWidget()
+
+        if "choices" not in kwargs:
+            kwargs["choices"] = db_field.get_choices(
+                include_blank=db_field.blank, blank_choice=[("", _("Select value"))]
+            )
+
+        return super().formfield_for_choice_field(db_field, request, **kwargs)
+
+    def formfield_for_foreignkey(
+        self, db_field: ForeignKey, request: HttpRequest, **kwargs
+    ) -> Optional[ModelChoiceField]:
+        db = kwargs.get("using")
+
+        # Overrides widgets for all related fields
+        if "widget" not in kwargs:
+            if db_field.name in self.raw_id_fields:
+                kwargs["widget"] = widgets.UnfoldForeignKeyRawIdWidget(
+                    db_field.remote_field, self.admin_site, using=db
+                )
+            elif (
+                db_field.name not in self.get_autocomplete_fields(request)
+                and db_field.name not in self.radio_fields
+            ):
+                kwargs["widget"] = widgets.UnfoldAdminSelectWidget()
+                kwargs["empty_label"] = _("Select value")
+
+        return super().formfield_for_foreignkey(db_field, request, **kwargs)
+
+    def formfield_for_manytomany(
+        self,
+        db_field: ManyToManyField,
+        request: HttpRequest,
+        **kwargs,
+    ) -> ModelMultipleChoiceField:
+        if "widget" not in kwargs:
+            if db_field.name in self.raw_id_fields:
+                kwargs["widget"] = widgets.UnfoldAdminTextInputWidget()
+
+        form_field = super().formfield_for_manytomany(db_field, request, **kwargs)
+
+        # If M2M uses intermediary model, form_field will be None
+        if not form_field:
+            return None
+
+        if isinstance(form_field.widget, SelectMultiple):
+            form_field.widget.attrs["class"] = " ".join(widgets.SELECT_CLASSES)
+
+        return form_field
+
+    def formfield_for_nullboolean_field(
+        self, db_field: Field, request: HttpRequest, **kwargs
+    ) -> Optional[Field]:
+        if "widget" not in kwargs:
+            kwargs["widget"] = widgets.UnfoldAdminNullBooleanSelectWidget()
+
+        return db_field.formfield(**kwargs)
+
+    def formfield_for_dbfield(
+        self, db_field: Field, request: HttpRequest, **kwargs
+    ) -> Optional[Field]:
+        if isinstance(db_field, models.BooleanField) and db_field.null is True:
+            return self.formfield_for_nullboolean_field(db_field, request, **kwargs)
+
+        formfield = super().formfield_for_dbfield(db_field, request, **kwargs)
+
+        if formfield and isinstance(formfield.widget, RelatedFieldWidgetWrapper):
+            formfield.widget.template_name = (
+                "unfold/widgets/related_widget_wrapper.html"
+            )
+
+        return formfield

unfold/overrides.py

@@ -0,0 +1,73 @@
+import copy
+
+from django import forms
+from django.db import models
+
+from unfold import widgets
+
+FORMFIELD_OVERRIDES = {
+    models.DateTimeField: {
+        "form_class": forms.SplitDateTimeField,
+        "widget": widgets.UnfoldAdminSplitDateTimeWidget,
+    },
+    models.DateField: {"widget": widgets.UnfoldAdminSingleDateWidget},
+    models.TimeField: {"widget": widgets.UnfoldAdminSingleTimeWidget},
+    models.EmailField: {"widget": widgets.UnfoldAdminEmailInputWidget},
+    models.CharField: {"widget": widgets.UnfoldAdminTextInputWidget},
+    models.URLField: {"widget": widgets.UnfoldAdminURLInputWidget},
+    models.GenericIPAddressField: {"widget": widgets.UnfoldAdminTextInputWidget},
+    models.UUIDField: {"widget": widgets.UnfoldAdminUUIDInputWidget},
+    models.TextField: {"widget": widgets.UnfoldAdminTextareaWidget},
+    models.NullBooleanField: {"widget": widgets.UnfoldAdminNullBooleanSelectWidget},
+    models.BooleanField: {"widget": widgets.UnfoldBooleanSwitchWidget},
+    models.IntegerField: {"widget": widgets.UnfoldAdminIntegerFieldWidget},
+    models.BigIntegerField: {"widget": widgets.UnfoldAdminBigIntegerFieldWidget},
+    models.DecimalField: {"widget": widgets.UnfoldAdminDecimalFieldWidget},
+    models.FloatField: {"widget": widgets.UnfoldAdminDecimalFieldWidget},
+    models.FileField: {"widget": widgets.UnfoldAdminFileFieldWidget},
+    models.ImageField: {"widget": widgets.UnfoldAdminImageFieldWidget},
+    models.JSONField: {"widget": widgets.UnfoldAdminTextareaWidget},
+    models.DurationField: {"widget": widgets.UnfoldAdminTextInputWidget},
+}
+
+######################################################################
+# Postgres
+######################################################################
+try:
+    from django.contrib.postgres.fields import ArrayField, IntegerRangeField
+    from django.contrib.postgres.search import SearchVectorField
+
+    FORMFIELD_OVERRIDES.update(
+        {
+            ArrayField: {"widget": widgets.UnfoldAdminTextareaWidget},
+            SearchVectorField: {"widget": widgets.UnfoldAdminTextareaWidget},
+            IntegerRangeField: {"widget": widgets.UnfoldAdminIntegerRangeWidget},
+        }
+    )
+except ImportError:
+    pass
+
+######################################################################
+# Django Money
+######################################################################
+try:
+    from djmoney.models.fields import MoneyField
+
+    FORMFIELD_OVERRIDES.update(
+        {
+            MoneyField: {"widget": widgets.UnfoldAdminMoneyWidget},
+        }
+    )
+except ImportError:
+    pass
+
+######################################################################
+# Inlines
+######################################################################
+FORMFIELD_OVERRIDES_INLINE = copy.deepcopy(FORMFIELD_OVERRIDES)
+
+FORMFIELD_OVERRIDES_INLINE.update(
+    {
+        models.ImageField: {"widget": widgets.UnfoldAdminImageSmallFieldWidget},
+    }
+)

unfold/sites.py

@@ -71,11 +71,11 @@ class UnfoldAdminSite(AdminSite):
                 "text_input": INPUT_CLASSES,
                 "checkbox": CHECKBOX_CLASSES,
             },
-            "site_title": self._get_config("SITE_TITLE", request),
-            "site_header": self._get_config("SITE_HEADER", request),
+            "site_title": self._get_config("SITE_TITLE", request) or self.site_title,
+            "site_header": self._get_config("SITE_HEADER", request) or self.site_header,
+            "site_url": self._get_config("SITE_URL", request) or self.site_url,
             "site_subheader": self._get_config("SITE_SUBHEADER", request),
             "site_dropdown": self._get_site_dropdown_items("SITE_DROPDOWN", request),
-            "site_url": self._get_config("SITE_URL", request),
             "site_logo": self._get_theme_images("SITE_LOGO", request),
             "site_icon": self._get_theme_images("SITE_ICON", request),
             "site_symbol": self._get_config("SITE_SYMBOL", request),