|
3 | 3 | from django.db.models import Q, QuerySet |
4 | 4 | from drf_spectacular.utils import extend_schema |
5 | 5 | from rest_framework import mixins, viewsets |
6 | | -from rest_framework.exceptions import NotFound |
| 6 | +from rest_framework.exceptions import APIException, NotFound |
7 | 7 | from rest_framework.permissions import IsAuthenticated |
8 | 8 | from rest_framework.request import Request |
9 | 9 | from rest_framework.response import Response |
10 | 10 |
|
11 | | -from api.public.v2.schema import owner_parameters, service_parameter |
| 11 | +from api.public.v2.schema import ( |
| 12 | + owner_parameters, |
| 13 | + service_parameter, |
| 14 | +) |
12 | 15 | from api.shared.owner.mixins import ( |
13 | 16 | OwnerViewSetMixin, |
14 | 17 | UserSessionViewSetMixin, |
15 | 18 | UserViewSetMixin, |
16 | 19 | ) |
17 | 20 | from codecov_auth.models import Owner, Service |
18 | 21 |
|
19 | | -from .serializers import OwnerSerializer, UserSerializer, UserSessionSerializer |
| 22 | +from .serializers import ( |
| 23 | + OwnerSerializer, |
| 24 | + UserSerializer, |
| 25 | + UserSessionSerializer, |
| 26 | + UserUpdateActivationSerializer, |
| 27 | +) |
| 28 | + |
| 29 | + |
| 30 | +class NotEnoughSeatsLeft(APIException): |
| 31 | + status_code = 400 |
| 32 | + default_detail = "Cannot activate user -- not enough seats left." |
| 33 | + default_code = "no_seats_left" |
20 | 34 |
|
21 | 35 |
|
22 | 36 | @extend_schema(parameters=owner_parameters, tags=["Users"]) |
@@ -53,6 +67,31 @@ def retrieve(self, request: Request, *args: Any, **kwargs: Any) -> Owner: |
53 | 67 | """ |
54 | 68 | return super().retrieve(request, *args, **kwargs) |
55 | 69 |
|
| 70 | + @extend_schema(summary="Update a user", request=UserUpdateActivationSerializer) |
| 71 | + def partial_update(self, request: Request, *args: Any, **kwargs: Any) -> Response: |
| 72 | + """ |
| 73 | + Updates a user for the specified owner_username or ownerid |
| 74 | +
|
| 75 | + Allowed fields |
| 76 | + - activated: boolean value to activate or deactivate the user |
| 77 | + """ |
| 78 | + instance = self.get_object() |
| 79 | + serializer = UserUpdateActivationSerializer( |
| 80 | + instance, |
| 81 | + data=request.data, |
| 82 | + ) |
| 83 | + serializer.is_valid(raise_exception=True) |
| 84 | + |
| 85 | + if serializer.validated_data["activated"]: |
| 86 | + if self.owner.can_activate_user(instance): |
| 87 | + self.owner.activate_user(instance) |
| 88 | + else: |
| 89 | + raise NotEnoughSeatsLeft() |
| 90 | + else: |
| 91 | + self.owner.deactivate_user(instance) |
| 92 | + |
| 93 | + return super().retrieve(request, *args, **kwargs) |
| 94 | + |
56 | 95 |
|
57 | 96 | @extend_schema(parameters=owner_parameters, tags=["Users"]) |
58 | 97 | class UserSessionViewSet(UserSessionViewSetMixin, mixins.ListModelMixin): |
|
0 commit comments