54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from django.conf import settings
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.urls import reverse_lazy
|
|
from django.views.generic import DetailView
|
|
from django.views.generic import ListView
|
|
from django.views.generic import RedirectView
|
|
from django.views.generic import UpdateView
|
|
|
|
from apps.map.models import CaseStudy
|
|
|
|
|
|
class UserDetailView(LoginRequiredMixin, DetailView):
|
|
model = get_user_model()
|
|
# These next two lines tell the view to index lookups by username
|
|
slug_field = "username"
|
|
slug_url_kwarg = "username"
|
|
|
|
def get_object(self):
|
|
return self.model.objects.get(username=self.request.user.username)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
# Get case studies for this user
|
|
case_studies = CaseStudy.objects.filter(author=self.request.user)
|
|
context["case_studies"] = case_studies
|
|
|
|
context["enable_edit"] = settings.FF_CAN_EDIT
|
|
|
|
return context
|
|
|
|
|
|
class UserRedirectView(LoginRequiredMixin, RedirectView):
|
|
permanent = False
|
|
url = reverse_lazy("profile:detail")
|
|
|
|
|
|
class UserUpdateView(LoginRequiredMixin, UpdateView):
|
|
model = get_user_model()
|
|
fields = ["first_name", "last_name"]
|
|
success_url = reverse_lazy("profile:detail")
|
|
|
|
def get_object(self):
|
|
# Only get the User record for the user making the request
|
|
return self.model.objects.get(username=self.request.user.username)
|
|
|
|
|
|
class UserListView(LoginRequiredMixin, ListView):
|
|
model = get_user_model()
|
|
# These next two lines tell the view to index lookups by username
|
|
slug_field = "username"
|
|
slug_url_kwarg = "username"
|