mirror of
https://github.com/boostorg/website-v2.git
synced 2026-02-27 17:42:08 +00:00
112 lines
3.5 KiB
Python
112 lines
3.5 KiB
Python
from django.contrib import messages
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.http import HttpResponseRedirect
|
|
from django.urls import reverse_lazy
|
|
from django.utils.decorators import method_decorator
|
|
from django.views.generic import DetailView, UpdateView
|
|
from django.views.generic.edit import FormView
|
|
|
|
from rest_framework import generics
|
|
from rest_framework import viewsets
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
from .forms import UserProfilePhotoForm
|
|
from .models import User
|
|
from .permissions import CustomUserPermissions
|
|
from .serializers import UserSerializer, FullUserSerializer, CurrentUserSerializer
|
|
from . import tasks
|
|
|
|
|
|
class UserViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
Main User API ViewSet
|
|
"""
|
|
|
|
queryset = User.objects.all()
|
|
permission_classes = [CustomUserPermissions]
|
|
|
|
def get_serializer_class(self):
|
|
"""Pick the right serializer based on the user"""
|
|
if self.request.user.is_staff or self.request.user.is_superuser:
|
|
return FullUserSerializer
|
|
else:
|
|
return UserSerializer
|
|
|
|
|
|
class CurrentUserView(generics.RetrieveUpdateAPIView):
|
|
"""
|
|
This gives the current user a convenient way to retrieve or
|
|
update slightly more detailed information about themselves.
|
|
|
|
Typically set to a route of `/api/v1/user/me`
|
|
"""
|
|
|
|
serializer_class = CurrentUserSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def get_object(self):
|
|
return self.request.user
|
|
|
|
|
|
class ProfileViewSet(DetailView):
|
|
"""
|
|
ViewSet to show statistics about a user to include
|
|
stats, badges, forum posts, reviews, etc.
|
|
"""
|
|
|
|
model = User
|
|
queryset = User.objects.all()
|
|
template_name = "users/profile.html"
|
|
context_object_name = "user"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
user = self.get_object()
|
|
context["authored"] = user.authors.all()
|
|
context["maintained"] = user.maintainers.all().distinct()
|
|
return context
|
|
|
|
|
|
@method_decorator(login_required, name="dispatch")
|
|
class ProfilePhotoUploadView(FormView):
|
|
"""Allows a user to change their profile photo"""
|
|
|
|
template_name = "users/photo_upload.html"
|
|
form_class = UserProfilePhotoForm
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
user = self.request.user
|
|
context["user_has_gh_username"] = bool(user.github_username)
|
|
return context
|
|
|
|
def get_success_url(self, **kwargs):
|
|
return reverse_lazy("profile-user", args=[self.request.user.pk])
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
form = self.form_class(request.POST, request.FILES, instance=self.request.user)
|
|
if form.is_valid():
|
|
form.save()
|
|
messages.success(request, "Your profile photo has been updated")
|
|
return super().form_valid(form)
|
|
else:
|
|
return super().form_invalid()
|
|
|
|
|
|
@method_decorator(login_required, name="dispatch")
|
|
class ProfilePhotoGitHubUpdateView(UpdateView):
|
|
"""Allow a user to sync their profile photo to their current GitHub photo."""
|
|
|
|
http_method_names = ["post"]
|
|
|
|
def get_object(self, queryset=None):
|
|
return self.request.user
|
|
|
|
def get_success_url(self, **kwargs):
|
|
return reverse_lazy("profile-user", args=[self.request.user.pk])
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
user = self.get_object()
|
|
tasks.update_user_github_photo.delay(user.pk)
|
|
return HttpResponseRedirect(self.get_success_url())
|