[#2646] Enable Organization in API v2

carpentries · May 22, 2024 · e2cf382 · e2cf382
1 parent c094d7f
commit e2cf382
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 0 deletions.
diff --git a/amy/api/v2/serializers.py b/amy/api/v2/serializers.py
@@ -6,6 +6,7 @@
     Award,
     Event,
     Membership,
+    Organization,
     Person,
     TrainingProgress,
     TrainingRequirement,
@@ -30,6 +31,25 @@ class Meta:
         )
 
 
+class OrganizationSerializer(serializers.ModelSerializer):
+    country = serializers.CharField()
+    affiliated_organizations = serializers.SlugRelatedField(
+        many=True, read_only=True, slug_field="domain"
+    )
+
+    class Meta:
+        model = Organization
+        fields = (
+            "pk",
+            "domain",
+            "fullname",
+            "country",
+            "latitude",
+            "longitude",
+            "affiliated_organizations",
+        )
+
+
 class EventSerializer(serializers.ModelSerializer):
     # start = serializers.DateField()
     # end = serializers.DateField()

diff --git a/amy/api/v2/urls.py b/amy/api/v2/urls.py
@@ -7,6 +7,7 @@
 
 router = routers.DefaultRouter(trailing_slash=False)
 router.register("award", views.AwardViewSet)
+router.register("organization", views.OrganizationViewSet)
 router.register("event", views.EventViewSet)
 router.register("instructorrecruitmentsignup", views.InstructorRecruitmentSignupViewSet)
 router.register("membership", views.MembershipViewSet)

diff --git a/amy/api/v2/views.py b/amy/api/v2/views.py
@@ -13,6 +13,7 @@
     EventSerializer,
     InstructorRecruitmentSignupSerializer,
     MembershipSerializer,
+    OrganizationSerializer,
     PersonSerializer,
     ScheduledEmailLogDetailsSerializer,
     ScheduledEmailSerializer,
@@ -26,6 +27,7 @@
     Award,
     Event,
     Membership,
+    Organization,
     Person,
     TrainingProgress,
     TrainingRequirement,
@@ -56,6 +58,24 @@ class AwardViewSet(viewsets.ReadOnlyModelViewSet):
     pagination_class = StandardResultsSetPagination
 
 
+class OrganizationViewSet(viewsets.ReadOnlyModelViewSet):
+    authentication_classes = (
+        TokenAuthentication,
+        SessionAuthentication,
+    )
+    permission_classes = (
+        IsAuthenticated,
+        ApiAccessPermission,
+    )
+    queryset = (
+        Organization.objects.prefetch_related("affiliated_organizations")
+        .order_by("pk")
+        .all()
+    )
+    serializer_class = OrganizationSerializer
+    pagination_class = StandardResultsSetPagination
+
+
 class EventViewSet(viewsets.ReadOnlyModelViewSet):
     authentication_classes = (
         TokenAuthentication,