Merge pull request #795 from uw-it-aca/feature/msca-shared-drive-delete

Feature/msca shared drive delete

Author: mikeseibel

endorsement/dao/shared_drive.py

@@ -84,25 +84,28 @@ def sync_quota_from_subscription(drive_id):
         raise SharedDriveRecordNotFound(drive_id)
 
 
-def expire_shared_drives(gracetime, lifetime):
+def expire_shared_drive(shared_drive_record):
     """
-    Expire shared drives that have exceeded their lifetime.
-    """
-    for drive in shared_drives_to_expire(gracetime, lifetime):
-        shared_drive_lifecycle_expired(drive)
+        Mark lifecycle expired shared drive record to deleted
 
+        Actions:
+            - mark expired drive for deletion in MSCA
+            - set shared_drive_record datetime_deleted to now
+        """
+    try:
+        logger.info(
+            "Expire shared drive "
+            f"{shared_drive_record.shared_drive.drive_id}, "
+            "mark for deletion")
 
-def shared_drive_lifecycle_expired(shared_drive):
-    """
-    Set lifecycle to expired for shared drive
+        mark_drive_for_deletion(shared_drive_record.shared_drive.drive_id)
 
-    Actions:
-       - set shared_drive quota to 0 (org_unit_name "None"? pending delete?)
-       - set subscription end_date to today using:
-            - expire_subscription(drive_record)
-    """
-    logger.info(f"Shared drive {shared_drive.drive_id} marked for deletion")
-    mark_drive_for_deletion(shared_drive.drive_id)
+        shared_drive_record.datetime_deleted = timezone.now()
+        shared_drive_record.save()
+    except DataFailureException as ex:
+        logger.error(
+            f"Expire {shared_drive_record.shared_drive.drive_id} "
+            f"failure: {ex}")
 
 
 def rescue_shared_drive_from_deletion(shared_drive):

endorsement/management/commands/expire_shared_drives.py

@@ -2,10 +2,9 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from django.core.management.base import BaseCommand
-from django.utils import timezone
 from restclients_core.exceptions import DataFailureException
 from endorsement.policy.shared_drive import SharedDrivePolicy
-from endorsement.dao.shared_drive import shared_drive_lifecycle_expired
+from endorsement.dao.shared_drive import expire_shared_drive
 import csv
 import logging
 
@@ -52,16 +51,7 @@ def handle(self, *args, **options):
                         f'"{record.shared_drive.drive_name}"')
 
             if actually_mark_for_deletion:
-                self.expire_drive(record)
-
-    def expire_drive(self, record):
-        try:
-            shared_drive_lifecycle_expired(record.shared_drive)
-            record.datetime_deleted = timezone.now()
-            record.save()
-        except DataFailureException as ex:
-            logger.error(
-                "Cannot expire drive {record.shared_drive.drive_id} : {ex}")
+                expire_shared_drive(record)
 
     def _record_csv(self, record):
         self.drive_writer.writerow([

endorsement/views/api/google/shared_drive.py

@@ -6,7 +6,7 @@
 from endorsement.dao.itbill import update_itbill_subscription
 from endorsement.dao.shared_drive import (
     sync_quota_from_subscription,
-    shared_drive_lifecycle_expired,
+    expire_shared_drive,
     rescue_shared_drive_from_deletion)
 from endorsement.dao.pws import get_person
 from endorsement.util.auth import is_support_user
@@ -71,7 +71,7 @@ def put(self, request, *args, **kwargs):
                 if drive.datetime_deleted:
                     rescue_shared_drive_from_deletion(drive.shared_drive)
             else:
-                shared_drive_lifecycle_expired(drive.shared_drive)
+                expire_shared_drive(drive.shared_drive)
 
             drive.set_acceptance(netid, accept, acted_as)