Add Edly site config api - EDLY-6656

course_discovery/apps/edly_discovery_app/api/v1/urls.py

@@ -6,5 +6,6 @@
 app_name = 'v1'
 urlpatterns = [
     url(r'^edly_sites/', edly_sites.EdlySiteViewSet.as_view(), name='edly_sites'),
+    url(r'^edly_site_config/', edly_sites.EdlySiteConfigViewset.as_view(), name='edly_site_config'),
     url(r'^dataloader/', dataloader_api.EdlyDataLoaderView.as_view(), name='edly_dataloader'),
 ]

course_discovery/apps/edly_discovery_app/api/v1/views/edly_sites.py

@@ -2,8 +2,7 @@
 Views for Edly Sites API.
 """
 from django.contrib.sites.models import Site
-from rest_framework import status, viewsets, filters
-from rest_framework.authentication import SessionAuthentication
+from rest_framework import status
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
 from rest_framework.views import APIView
@@ -96,3 +95,24 @@ def get_updated_site_partner(self, discovery_site):
         partner.save()
 
         return partner
+
+
+class EdlySiteConfigViewset(APIView):
+    """
+    Create Default Site and Partner Configuration.
+    """
+    permission_classes = [IsAuthenticated, CanAccessSiteCreation]
+
+    def post(self, request):
+        """
+        POST /edly_api/v1/edly_site_config/
+        """
+        site = request.site
+        site_partner = Partner.objects.get(site=site)
+        request_data = request.data.get('discovery', {})
+        current_plan = request_data.get('DJANGO_SETTINGS_OVERRIDE', {}).get('CURRENT_PLAN')
+        if current_plan == 'deactivated':
+            site_partner.is_disabled = True
+            site_partner.save()
+
+        return Response('Discovery site updated successfully', status=status.HTTP_200_OK)