feat: Added site deletion endpoint.

Author: manan-memon

ecommerce/extensions/edly_ecommerce_app/api/v1/constants.py

@@ -13,6 +13,8 @@
     'DJANGO_SETTINGS_UPDATE_FAILURE': _('Django settings update failed.'),
     'SITE_CONFIGURATIONS_UPDATE_SUCCESS': _('Site Configurations updated successfully.'),
     'SITE_CONFIGURATIONS_UPDATE_FAILURE': _('Site Configurations update failed.'),
+    'SITE_DELETION_SUCCESS': _('Ecommerce site deletion was successful.'),
+    'SITE_DELETION_FAILURE': _('Ecommerce site deletion failed.'),
 }
 
 CLIENT_SITE_SETUP_FIELDS = [

ecommerce/extensions/edly_ecommerce_app/api/v1/urls.py

@@ -8,5 +8,6 @@
     url(r'site_themes/', views.SiteThemesActions.as_view(), name='site_themes'),
     url(r'csrf_token/', views.CSRFTokenInfo.as_view(), name='get_csrf_token'),
     url(r'edly_sites/', views.EdlySiteViewSet.as_view(), name='edly_sites'),
+    url(r'delete_site/', views.EdlySiteDeletionViewSet.as_view(), name='edly_delete_site'),
     url(r'edly_site_config/', views.EdlySiteConfigViewset.as_view(), name='edly_site_config'),
 ]

ecommerce/extensions/edly_ecommerce_app/api/v1/views.py

@@ -5,6 +5,8 @@
 from django.contrib.auth.decorators import login_required
 from django.contrib.sites.models import Site
 from django.contrib.sites.shortcuts import get_current_site
+from django.contrib.auth import get_user_model
+from django.db import transaction
 from django.http import Http404
 from django.middleware import csrf
 from django.utils.decorators import method_decorator
@@ -18,6 +20,7 @@
 
 from ecommerce.core.constants import ENABLE_SUBSCRIPTIONS_ON_RUNTIME_SWITCH
 from ecommerce.core.models import SiteConfiguration
+from ecommerce.courses.models import Course
 from ecommerce.extensions.edly_ecommerce_app.api.v1.constants import ERROR_MESSAGES
 from ecommerce.extensions.edly_ecommerce_app.helpers import (
     user_is_course_creator,
@@ -27,9 +30,13 @@
     validate_site_configurations_for_self_service_api,
     validate_site_theme,
 )
-from ecommerce.extensions.edly_ecommerce_app.permissions import CanAccessSiteCreation
+from ecommerce.extensions.edly_ecommerce_app.permissions import CanAccessSiteCreation, CanAccessSiteDeletion
 from ecommerce.extensions.partner.models import Partner
 from ecommerce.theming.models import SiteTheme
+import logging
+
+logger = logging.getLogger(__name__)
+User = get_user_model()
 
 
 class SiteThemesActions(APIView):
@@ -205,6 +212,116 @@ def get_oauth2_credentials(self):
         )
         return oauth2_values
 
+class EdlySiteDeletionViewSet(APIView):
+    """
+    Delete the ecommerce site and its data.
+    """
+    permission_classes = [IsAuthenticated, CanAccessSiteDeletion] 
+
+    def post(self, request):
+        """
+        POST /edly_ecommerce_api/delete_site/
+        """
+        try:
+            with transaction.atomic():
+                self.process_site_deletion(request)
+            
+            return Response(
+                {'success': ERROR_MESSAGES.get('SITE_DELETION_SUCCESS')},
+                status=status.HTTP_200_OK
+            )
+        except Exception as e:
+            logger.error(f"Site deletion failed: {str(e)}", exc_info=True)
+            return Response(
+                {
+                    'error': ERROR_MESSAGES.get('SITE_DELETION_FAILURE'),
+                    'detail': str(e)
+                },
+                status=status.HTTP_400_BAD_REQUEST
+            )
+    
+    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
+        ).exclude(id=request.user.id)
+        from ecommerce.extensions.order.models import MarkOrdersStatusCompleteConfig
+        MarkOrdersStatusCompleteConfig.objects.filter(changed_by__in=users).delete()
+        users.delete()
+        logger.info(f"Successfully deleted user for {request.site}")
+    
+    def delete_courses(self, request):
+        """Delete the courses for the site."""
+        partner = request.site.partner
+        courses = Course.objects.filter(partner=partner)
+
+        course_ids = courses.values_list('id', flat=True)
+
+        HistoricalCourse = Course.history.model
+        HistoricalCourse.objects.filter(id__in=course_ids).delete()
+        courses.delete()
+
+    def delete_partner_additional_data(self, request):
+        """
+        Delete all related data referencing the partner that is not automatically
+        deleted due to on_delete settings.
+        """
+        from ecommerce.extensions.partner.models import StockRecord
+        from ecommerce.extensions.order.models import Order, Line
+        from ecommerce.extensions.offer.models import ConditionalOffer
+
+        site = request.site
+        partner = Partner.objects.get(default_site=site)
+        
+        HistoricalConditionalOffer = ConditionalOffer.history.model
+        count, _ = HistoricalConditionalOffer.objects.filter(partner=partner).delete()
+        logger.info(f"Deleted {count} HistoricalConditionalOffer record(s) for partner {partner}")
+
+        HistoricalOrder = Order.history.model
+        count, _ = HistoricalOrder.objects.filter(partner=partner).delete()
+        logger.info(f"Deleted {count} HistoricalOrder record(s) for partner {partner}")
+
+        count, _ = Line.objects.filter(partner=partner).delete()
+        logger.info(f"Deleted {count} Line record(s) for partner {partner}")
+        HistoricalLine = Line.history.model
+        count, _ = HistoricalLine.objects.filter(partner=partner).delete()
+        logger.info(f"Deleted {count} HistoricalLine record(s) for partner {partner}")
+
+        HistoricalStockRecord = StockRecord.history.model
+        count, _ = HistoricalStockRecord.objects.filter(partner=partner).delete()
+        logger.info(f"Deleted {count} HistoricalStockRecord record(s) for partner {partner}")
+
+        partner.history.all().delete()
+
+    
+    def delete_site(self, request):
+        """Delete the site and partner for a given site."""
+        site = request.site
+        try:
+            site_partner = Partner.objects.get(default_site=site)
+            site_partner.delete()
+            site.delete()
+            logger.info(f"Successfully deleted site {site} and its partner")
+        except Partner.DoesNotExist:
+            logger.info(f'Partner Not Found for site {request.site}')
+
+    def process_site_deletion(self, request):
+        """
+        Process deletion of a site.
+        """
+        self.delete_courses(request)
+        self.delete_partner_additional_data(request)
+        self.delete_users(request)
+        self.delete_site(request)
+
+
 
 class EdlySiteConfigViewset(APIView):
     """

ecommerce/extensions/edly_ecommerce_app/permissions.py

@@ -24,3 +24,14 @@ def has_permission(self, request, view):
         Checks for user's permission for current site.
         """
         return request.user.is_staff and request.user.username == EDLY_PANEL_WORKER_USER
+
+class CanAccessSiteDeletion(BasePermission):
+    """
+    Checks if a user has the access to create and update methods for sites.
+    """
+
+    def has_permission(self, request, view):
+        """
+        Checks for user's permission for current site.
+        """
+        return request.user.username == EDLY_PANEL_WORKER_USER