Source code for salesman.admin.wagtail.helpers

from typing import Any

from django.contrib.auth.models import AbstractBaseUser
from django.utils.translation import gettext_lazy as _
from wagtail.contrib.modeladmin.helpers import (
    AdminURLHelper,
    ButtonHelper,
    PermissionHelper,
)


class OrderPermissionHelper(PermissionHelper):
    def user_can_create(self, user: AbstractBaseUser) -> bool:
        return False

    def user_can_delete_obj(self, user: AbstractBaseUser, obj: Any) -> bool:
        return False


class OrderButtonHelper(ButtonHelper):
    def edit_button(self, *args: Any, **kwargs: Any) -> Any:
        button = super().edit_button(*args, **kwargs)
        button.update({"label": _("View"), "title": _("View this Order")})
        return button


class OrderAdminURLHelper(AdminURLHelper):
    pass