ojuso-map/apps/profiles/views.py

try:
    from django.urls import reverse
except ImportError:
    from django.core.urlresolvers import reverse

from django.conf import settings

from django.views.generic import DetailView, ListView, RedirectView, UpdateView

from django.contrib.auth.mixins import LoginRequiredMixin

from django.contrib.auth import get_user_model

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 get_user_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.id)
        context['case_studies'] = case_studies

        context['enable_edit'] = settings.FF_CAN_EDIT

        return context


class UserRedirectView(LoginRequiredMixin, RedirectView):
    permanent = False

    def get_redirect_url(self):
        return reverse('profile:detail')


class UserUpdateView(LoginRequiredMixin, UpdateView):

    fields = ['first_name', 'last_name']

    # we already imported User in the view code above, remember?
    model = get_user_model()

    # send the user back to their own page after a successful update
    def get_success_url(self):
        return reverse('profile:detail')

    def get_object(self):
        # Only get the User record for the user making the request
        return get_user_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'