Add API endpoint for organization type

core/serializers.py

@@ -17,7 +17,7 @@
 from core.email_utils import send_email, send_email_body
 
 from core.models import CoreUser, CoreGroup, EmailTemplate, LogicModule, Organization, PERMISSIONS_ORG_ADMIN, \
-    TEMPLATE_RESET_PASSWORD, Consortium
+    TEMPLATE_RESET_PASSWORD, Consortium, OrganizationType
 
 
 class LogicModuleSerializer(serializers.ModelSerializer):
@@ -358,3 +358,11 @@ class ConsortiumSerializer(serializers.ModelSerializer):
     class Meta:
         model = Consortium
         fields = '__all__'
+
+
+class OrganizationTypeSerializer(serializers.ModelSerializer):
+    id = serializers.ReadOnlyField()
+
+    class Meta:
+        model = OrganizationType
+        fields = '__all__'

core/urls.py

@@ -3,7 +3,7 @@
 from django.contrib.staticfiles.urls import staticfiles_urlpatterns
 from core.views.consortium import ConsortiumViewSet
 from rest_framework import routers
-
+from core.views.organization import OrganizationTypeViewSet
 from core import views
 from core.views.web import IndexView, oauth_complete
 
@@ -22,7 +22,7 @@
 router.register(r'organization', views.OrganizationViewSet)
 router.register(r'logicmodule', views.LogicModuleViewSet)
 router.register(r'consortium', ConsortiumViewSet)
-
+router.register(r'organization_type', OrganizationTypeViewSet)
 
 urlpatterns = [
     path('', IndexView.as_view(), name='index'),

core/views/organization.py

@@ -1,14 +1,15 @@
 import logging
-
+from django_filters.rest_framework import DjangoFilterBackend
+from core.permissions import IsSuperUser
 import django_filters
 from rest_framework import viewsets
 from rest_framework.response import Response
 from rest_framework.decorators import action
-from core.models import Organization
-from core.serializers import OrganizationSerializer
+from core.models import Organization, OrganizationType
+from core.serializers import OrganizationSerializer, OrganizationTypeSerializer
 from core.permissions import IsOrgMember
 from django.views.decorators.csrf import csrf_exempt
-
+from rest_framework.permissions import AllowAny
 
 logger = logging.getLogger(__name__)
 
@@ -50,7 +51,8 @@ def list(self, request, *args, **kwargs):
     serializer_class = OrganizationSerializer
 
     @csrf_exempt
-    @action(detail=False, methods=['get'], name='Fetch Already existing Organization', url_path='fetch_orgs')
+    @action(detail=False, methods=['get'], permission_classes=[AllowAny],
+            name='Fetch Already existing Organization', url_path='fetch_orgs')
     def fetch_existing_orgs(self, request, pk=None, *args, **kwargs):
         """
         Fetch Already existing Organizations in Buildly Core,
@@ -63,3 +65,39 @@ def fetch_existing_orgs(self, request, pk=None, *args, **kwargs):
             names.append(record.name)
 
         return Response(names)
+
+
+class OrganizationTypeViewSet(viewsets.ModelViewSet):
+    """
+    Organization type  is associated with an organization which defines type of organization.
+
+    title:
+    Organization Type
+
+    description:
+    An organization type are custodian and producer
+
+    They are associated with an organization.
+    Only admin has access to organization type.
+
+    retrieve:
+    Return the  Organization Type.
+
+    list:
+    Return a list of all the existing  Organization Types.
+
+    create:
+    Create a new Organization Type instance.
+
+    update:
+    Update a Organization Type instance.
+
+    delete:
+    Delete a Organization Type instance.
+    """
+
+    filter_fields = ('name',)
+    filter_backends = (DjangoFilterBackend,)
+    permission_classes = (IsSuperUser,)
+    queryset = OrganizationType.objects.all()
+    serializer_class = OrganizationTypeSerializer