2019-03-07 09:15:18 +00:00
|
|
|
from django.conf import settings
|
2017-10-31 14:57:26 +00:00
|
|
|
from django.contrib.auth import get_user_model
|
2019-08-19 23:06:20 +00:00
|
|
|
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
|
2017-10-31 14:57:26 +00:00
|
|
|
|
2019-03-04 12:39:27 +00:00
|
|
|
from apps.map.models import CaseStudy
|
|
|
|
|
2017-10-31 14:57:26 +00:00
|
|
|
|
|
|
|
class UserDetailView(LoginRequiredMixin, DetailView):
|
|
|
|
model = get_user_model()
|
|
|
|
# These next two lines tell the view to index lookups by username
|
2019-08-19 21:37:32 +00:00
|
|
|
slug_field = "username"
|
|
|
|
slug_url_kwarg = "username"
|
2017-10-31 14:57:26 +00:00
|
|
|
|
|
|
|
def get_object(self):
|
2019-08-19 23:06:20 +00:00
|
|
|
return self.model.objects.get(username=self.request.user.username)
|
2017-10-31 14:57:26 +00:00
|
|
|
|
2019-03-04 12:39:27 +00:00
|
|
|
def get_context_data(self, **kwargs):
|
|
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
|
|
|
|
# Get case studies for this user
|
2019-08-19 23:06:20 +00:00
|
|
|
case_studies = CaseStudy.objects.filter(author=self.request.user)
|
2019-08-19 21:37:32 +00:00
|
|
|
context["case_studies"] = case_studies
|
2019-03-04 12:39:27 +00:00
|
|
|
|
2019-08-19 21:37:32 +00:00
|
|
|
context["enable_edit"] = settings.FF_CAN_EDIT
|
2019-03-07 09:15:18 +00:00
|
|
|
|
2019-03-04 12:39:27 +00:00
|
|
|
return context
|
|
|
|
|
2017-10-31 14:57:26 +00:00
|
|
|
|
|
|
|
class UserRedirectView(LoginRequiredMixin, RedirectView):
|
|
|
|
permanent = False
|
2019-08-19 23:06:20 +00:00
|
|
|
url = reverse_lazy("profile:detail")
|
2017-10-31 14:57:26 +00:00
|
|
|
|
|
|
|
|
|
|
|
class UserUpdateView(LoginRequiredMixin, UpdateView):
|
|
|
|
model = get_user_model()
|
2019-08-19 23:06:20 +00:00
|
|
|
fields = ["first_name", "last_name"]
|
|
|
|
success_url = reverse_lazy("profile:detail")
|
2017-10-31 14:57:26 +00:00
|
|
|
|
|
|
|
def get_object(self):
|
|
|
|
# Only get the User record for the user making the request
|
2019-08-19 23:06:20 +00:00
|
|
|
return self.model.objects.get(username=self.request.user.username)
|
2017-10-31 14:57:26 +00:00
|
|
|
|
|
|
|
|
|
|
|
class UserListView(LoginRequiredMixin, ListView):
|
|
|
|
model = get_user_model()
|
|
|
|
# These next two lines tell the view to index lookups by username
|
2019-08-19 21:37:32 +00:00
|
|
|
slug_field = "username"
|
|
|
|
slug_url_kwarg = "username"
|