kopia lustrzana https://github.com/jointakahe/takahe
26 wiersze
819 B
Python
26 wiersze
819 B
Python
from django.contrib.auth.decorators import login_required
|
|
from django.utils.decorators import method_decorator
|
|
|
|
from users.shortcuts import by_handle_for_user_or_404
|
|
|
|
|
|
@method_decorator(login_required, name="dispatch")
|
|
class IdentityViewMixin:
|
|
"""
|
|
A mixin that requires that the view has a "handle" kwarg that resolves
|
|
to a valid identity that the current user has.
|
|
"""
|
|
|
|
def dispatch(self, request, *args, **kwargs):
|
|
self.identity = by_handle_for_user_or_404(request, kwargs["handle"])
|
|
self.post_identity_setup()
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
def post_identity_setup(self):
|
|
pass
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context["identity"] = self.identity
|
|
return context
|