Add EditTeamAPIView

Author: Oksamies

django/thunderstore/api/cyberstorm/views/__init__.py

@@ -6,7 +6,12 @@
     NamespacePackageListAPIView,
     PackageDependantsListAPIView,
 )
-from .team import TeamDetailAPIView, TeamMembersAPIView, TeamServiceAccountsAPIView
+from .team import (
+    EditTeamAPIView,
+    TeamDetailAPIView,
+    TeamMembersAPIView,
+    TeamServiceAccountsAPIView,
+)
 
 __all__ = [
     "CommunityDetailAPIView",
@@ -16,6 +21,7 @@
     "NamespacePackageListAPIView",
     "PackageDependantsListAPIView",
     "TeamDetailAPIView",
+    "EditTeamAPIView",
     "TeamMembersAPIView",
     "TeamServiceAccountsAPIView",
 ]

django/thunderstore/api/cyberstorm/views/team.py

@@ -1,8 +1,12 @@
 from django.db.models import Q, QuerySet
-from rest_framework.exceptions import PermissionDenied
+from drf_yasg.utils import swagger_auto_schema
+from rest_framework import serializers
+from rest_framework.exceptions import PermissionDenied, ValidationError
 from rest_framework.generics import ListAPIView, RetrieveAPIView, get_object_or_404
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.request import Request
+from rest_framework.response import Response
+from rest_framework.views import APIView
 
 from thunderstore.account.models.service_account import ServiceAccount
 from thunderstore.api.cyberstorm.serializers import (
@@ -12,6 +16,7 @@
 )
 from thunderstore.api.ordering import StrictOrderingFilter
 from thunderstore.api.utils import CyberstormAutoSchemaMixin
+from thunderstore.repository.forms import DonationLinkTeamForm
 from thunderstore.repository.models.team import Team, TeamMember
 
 
@@ -39,6 +44,41 @@ def check_permissions(self, request: Request) -> None:
             raise PermissionDenied()
 
 
+class CyberstormEditTeamRequestSerialiazer(serializers.Serializer):
+    donation_link = serializers.CharField(
+        max_length=Team._meta.get_field("donation_link").max_length,
+        validators=Team._meta.get_field("donation_link").validators,
+    )
+
+
+class CyberstormEditTeamResponseSerialiazer(serializers.Serializer):
+    donation_link = serializers.CharField()
+
+
+class EditTeamAPIView(APIView):
+    @swagger_auto_schema(
+        request_body=CyberstormEditTeamRequestSerialiazer,
+        responses={200: CyberstormEditTeamResponseSerialiazer},
+        operation_id="cyberstorm.team.edit",
+        tags=["cyberstorm"],
+    )
+    def post(self, request, team_name, format=None):
+        serializer = CyberstormEditTeamRequestSerialiazer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        team = get_object_or_404(Team, name=team_name)
+        form = DonationLinkTeamForm(
+            user=request.user,
+            instance=team,
+            data=request.data,
+        )
+
+        if form.is_valid():
+            team = form.save()
+            return Response(CyberstormEditTeamResponseSerialiazer(team).data)
+        else:
+            raise ValidationError(form.errors)
+
+
 class TeamMembersAPIView(CyberstormAutoSchemaMixin, TeamRestrictedAPIView):
     serializer_class = CyberstormTeamMemberSerializer
     filter_backends = [StrictOrderingFilter]

django/thunderstore/api/urls.py

@@ -5,6 +5,7 @@
     CommunityFiltersAPIView,
     CommunityListAPIView,
     CommunityPackageListAPIView,
+    EditTeamAPIView,
     NamespacePackageListAPIView,
     PackageDependantsListAPIView,
     TeamDetailAPIView,
@@ -48,6 +49,11 @@
         TeamDetailAPIView.as_view(),
         name="cyberstorm.team.detail",
     ),
+    path(
+        "team/<str:team_id>/edit/",
+        EditTeamAPIView.as_view(),
+        name="cyberstorm.team.edit",
+    ),
     path(
         "team/<str:team_id>/members/",
         TeamMembersAPIView.as_view(),