from django.core.exceptions import PermissionDenied
from django.utils.translation import ugettext as _

from pretix.base.models import EventPermission, OrganizerPermission


def event_permission_required(permission):
    """
    This view decorator rejects all requests with a 403 response which are not from
    users having the given permission for the event the request is associated with.
    """
    def decorator(function):
        def wrapper(request, *args, **kw):
            if not request.user.is_authenticated():  # NOQA
                # just a double check, should not ever happen
                raise PermissionDenied()
            try:
                perm = EventPermission.objects.get(
                    event=request.event,
                    user=request.user
                )
            except EventPermission.DoesNotExist:
                pass
            else:
                allowed = not permission
                try:
                    if permission:
                        allowed = getattr(perm, permission)
                except AttributeError:
                    pass
                if allowed:
                    return function(request, *args, **kw)
            raise PermissionDenied(_('You do not have permission to view this content.'))
        return wrapper
    return decorator


class EventPermissionRequiredMixin:
    """
    This mixin is equivalent to the event_permission_required view decorator but
    is in a form suitable for class-based views.
    """
    permission = ''

    @classmethod
    def as_view(cls, **initkwargs):
        view = super(EventPermissionRequiredMixin, cls).as_view(**initkwargs)
        return event_permission_required(cls.permission)(view)


def organizer_permission_required(permission):
    """
    This view decorator rejects all requests with a 403 response which are not from
    users having the given permission for the event the request is associated with.
    """
    def decorator(function):
        def wrapper(request, *args, **kw):
            if not request.user.is_authenticated():  # NOQA
                # just a double check, should not ever happen
                raise PermissionDenied()
            try:
                perm = OrganizerPermission.objects.get(
                    organizer=request.organizer,
                    user=request.user
                )
            except OrganizerPermission.DoesNotExist:
                pass
            else:
                allowed = not permission
                try:
                    if permission:
                        allowed = getattr(perm, permission)
                except AttributeError:
                    pass
                if allowed or request.user.is_superuser:
                    return function(request, *args, **kw)
            raise PermissionDenied(_('You do not have permission to view this content.'))
        return wrapper
    return decorator


class OrganizerPermissionRequiredMixin:
    """
    This mixin is equivalent to the event_permission_required view decorator but
    is in a form suitable for class-based views.
    """
    permission = ''

    @classmethod
    def as_view(cls, **initkwargs):
        view = super(OrganizerPermissionRequiredMixin, cls).as_view(**initkwargs)
        return organizer_permission_required(cls.permission)(view)