110 lines
3.3 KiB
Python
110 lines
3.3 KiB
Python
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.urls import reverse
|
|
from django.views.generic import DetailView, RedirectView, UpdateView
|
|
from django.contrib import messages
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
|
|
from geeksbot_v2.users.serializers import UserSerializer
|
|
from geeksbot_v2.users.serializers import UserLogSerializer
|
|
from geeksbot_v2.users.models import UserLog
|
|
from geeksbot_v2.utils.api_utils import PaginatedAPIView
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class UserDetailView(LoginRequiredMixin, DetailView):
|
|
|
|
model = User
|
|
slug_field = "username"
|
|
slug_url_kwarg = "username"
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.object = self.get_object()
|
|
context = self.get_context_data(object=self.object, user=request.user)
|
|
return self.render_to_response(context)
|
|
|
|
|
|
user_detail_view = UserDetailView.as_view()
|
|
|
|
|
|
class UserUpdateView(LoginRequiredMixin, UpdateView):
|
|
|
|
model = User
|
|
fields = ["name"]
|
|
|
|
def get_success_url(self):
|
|
return reverse("users:detail", kwargs={"username": self.request.user.username})
|
|
|
|
def get_object(self):
|
|
return User.objects.get(username=self.request.user.username)
|
|
|
|
def form_valid(self, form):
|
|
messages.add_message(
|
|
self.request, messages.INFO, _("Infos successfully updated")
|
|
)
|
|
return super().form_valid(form)
|
|
|
|
|
|
user_update_view = UserUpdateView.as_view()
|
|
|
|
|
|
class UserRedirectView(LoginRequiredMixin, RedirectView):
|
|
|
|
permanent = False
|
|
|
|
def get_redirect_url(self):
|
|
return reverse("users:detail", kwargs={"username": self.request.user.username})
|
|
|
|
|
|
user_redirect_view = UserRedirectView.as_view()
|
|
|
|
# API Views
|
|
|
|
|
|
class UsersAPI(PaginatedAPIView):
|
|
def get(self, request, guild, format=None):
|
|
users = User.objects.filter(guilds__id=guild)
|
|
page = self.paginate_queryset(users)
|
|
if page is not None:
|
|
serialized_users = UserSerializer(users, many=True)
|
|
return self.get_paginated_response(serialized_users.data)
|
|
|
|
serialized_users = UserSerializer(users, many=True)
|
|
return Response(serialized_users.data)
|
|
|
|
|
|
class UserDetail(APIView):
|
|
def get(self, request, guild, id, format=None):
|
|
user = User.objects.filter(guilds__id=guild).get(id=id)
|
|
return Response(UserSerializer(user).data)
|
|
|
|
|
|
class UserLogList(PaginatedAPIView):
|
|
def get(self, request, user, action=None, format=None):
|
|
if action:
|
|
user_logs = (
|
|
UserLog.objects.filter(user=user)
|
|
.filter(action=action)
|
|
.order_by("-time")
|
|
)
|
|
else:
|
|
user_logs = UserLog.objects.filter(user=user).order_by("-time")
|
|
|
|
page = self.paginate_queryset(user_logs)
|
|
if page is not None:
|
|
serialized_logs = UserLogSerializer(page, many=True)
|
|
return self.get_paginated_response(serialized_logs.data)
|
|
|
|
serialized_logs = UserLogSerializer(user_logs, many=True)
|
|
return Response(serialized_logs.data)
|
|
|
|
|
|
class UserLogDetail(APIView):
|
|
def get(self, request, id, format=None):
|
|
user_log = UserLog.objects.get(id=id)
|
|
return Response(UserLogSerializer(user_log).data)
|