Add WikimediaProject model and related API endpoints

WikiMovimentoBrasil · Jan 11, 2025 · ffeebbd · ffeebbd
1 parent 8f75585
commit ffeebbd
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/CapX/urls.py b/CapX/urls.py
@@ -23,7 +23,7 @@
 from skills.views import SkillViewSet, SkillByTypeViewSet
 from users.views import (
     ProfileViewSet, UsersViewSet, QuickListViewSet,
-    UsersBySkillViewSet, UsersByTagViewSet, TerritoryViewSet,
+    UsersBySkillViewSet, UsersByTagViewSet, TerritoryViewSet, WikimediaProjectViewSet
 )
 from bugs.views import BugViewSet, AttachmentViewSet
 from orgs.views import OrganizationViewSet, OrganizationTypeViewSet
@@ -38,6 +38,7 @@
 router.register('skill', SkillViewSet, basename='skill')
 router.register('users', UsersViewSet, basename='users')
 router.register('profile', ProfileViewSet, basename='profile')
+router.register('wikimedia_project', WikimediaProjectViewSet, basename='wikimedia_project')
 router.register('territory', TerritoryViewSet, basename='territory')
 router.register('organizations', OrganizationViewSet, basename='organizations')
 router.register('organization_type', OrganizationTypeViewSet, basename='organization_type')

diff --git a/users/serializers.py b/users/serializers.py
@@ -46,7 +46,7 @@ class WikimediaProjectSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = WikimediaProject
-        fields = ['id', 'wikimedia_project_name', 'wikimedia_project_code']
+        fields = ['id', 'wikimedia_project_name', 'wikimedia_project_code', 'wikimedia_project_picture']
 
 class OrganizationSerializer(serializers.ModelSerializer):
 

diff --git a/users/views.py b/users/views.py
@@ -142,6 +142,20 @@ def perform_destroy(self, instance):
         instance.delete()
         user.delete()
 
+@extend_schema_view(
+    list=extend_schema(
+        summary='List all Wikimedia projects.',
+        description='This endpoint lists all Wikimedia projects.',
+    ),
+    retrieve=extend_schema(
+        summary='Retrieve a Wikimedia project by ID.',
+        description='This endpoint retrieves a Wikimedia project by its ID.',
+    ),
+)
+class WikimediaProjectViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = WikimediaProject.objects.all()
+    serializer_class = WikimediaProjectSerializer
+
 @extend_schema_view(
     list=extend_schema(
         summary='List all territories.',