Feat: Implement an API to delete site data.

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

@@ -6,6 +6,7 @@
 app_name = 'v1'
 urlpatterns = [
     url(r'^edly_sites/', edly_sites.EdlySiteViewSet.as_view(), name='edly_sites'),
+    url(r'^delete_site/', edly_sites.EdlySiteDeletionViewSet.as_view(), name='delete_site'),
     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

@@ -1,7 +1,11 @@
 """
 Views for Edly Sites API.
 """
+import logging
+
+from django.contrib.auth import get_user_model
 from django.contrib.sites.models import Site
+from django.db import transaction
 from rest_framework import status
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
@@ -12,6 +16,10 @@
 from edly_discovery_app.api.v1.constants import CoursePlans, DEFAULT_COURSE_ID, ERROR_MESSAGES
 from edly_discovery_app.api.v1.helpers import validate_partner_configurations
 from edly_discovery_app.api.v1.permissions import CanAccessSiteCreation
+from course_discovery.apps.course_metadata.models import Organization, Course
+
+logger = logging.getLogger(__name__)
+User = get_user_model()
 
 
 class EdlySiteViewSet(APIView):
@@ -94,6 +102,71 @@ def get_updated_site_partner(self, discovery_site):
         return partner
 
 
+class EdlySiteDeletionViewSet(APIView):
+    """
+    Create Default Site and Partner Configuration.
+    """
+    permission_classes = [IsAuthenticated, CanAccessSiteCreation]
+
+    def delete_users(self, request):
+        """Delete all the sync user for a given site."""
+        user_emails = request.data.get('emails')
+        user_names = request.data.get('usernames')
+        if not all([len(user_emails), len(user_names)]):
+            logger.info(f"No user deleted for given site : {(str(request.site))}")
+            return 
+
+        users = User.objects.filter(
+            email__in=user_emails,
+            username__in=user_names
+        )
+        users.delete()
+
+    def process_history_data(self, site, site_partner):
+        """Process the deletion of the historical data."""
+        OrganizationHistory = Organization.history.model
+        OrganizationHistory.objects.filter(partner=site_partner).delete()
+        CourseHistory = Course.history.model
+        CourseHistory.objects.filter(partner=site_partner).delete()
+
+
+    def delete_site(self, site, partner):
+        """Delete the site and partner for a given site."""
+        partner.delete()
+        site.delete()
+
+    def get_current_site(self, request):
+        """get current site value using domain value from request."""
+        site_domain = request.data.get('delete_site_url', '')
+        return Site.objects.get(domain=site_domain)
+
+    def process_deletion(self, request):
+        """Process the deletion request for a given site."""
+        with transaction.atomic():
+            site = self.get_current_site(request)
+            site_partner = Partner.objects.get(site=site)
+            self.process_history_data(site, site_partner)
+            self.delete_site(site, site_partner)
+            self.delete_users(request)
+
+    def post(self, request):
+        """
+        POST /edly_api/v1/delete_site/
+        """
+        try:
+            self.process_deletion(request)
+            return Response(
+                {'success': 'Discovery data deletion was successful'},
+                status=status.HTTP_200_OK
+            )
+        except Exception as err:
+            logger.info(f"Error deleting site: {str(err)}")
+            return Response(
+                {'error': f'{str(err)}'},
+                status=status.HTTP_400_BAD_REQUEST
+            )
+
+
 class EdlySiteConfigViewset(APIView):
     """
     Create Default Site and Partner Configuration.