Do not delete Model if the corresponsing Project is removed; set stat…

…us to Archived

components/studio/models/models.py

@@ -16,9 +16,11 @@ class Model(models.Model):
 
     CREATED = 'CR'
     DEPLOYED = 'DP'
+    ARCHIVED = 'AR'
     STATUS = [
         (CREATED, 'Created'),
         (DEPLOYED, 'Deployed'),
+        (ARCHIVED, 'Archived'),
     ]
     uid = models.CharField(max_length=255)
     name = models.CharField(max_length=255)
@@ -27,7 +29,11 @@ class Model(models.Model):
     resource = models.URLField(max_length=2048, null=True, blank=True)
     url = models.URLField(max_length=512, null=True, blank=True)
     uploaded_at = models.DateTimeField(auto_now_add=True)
-    project = models.ForeignKey('projects.Project', on_delete=models.CASCADE, related_name='model_owner')
+    project = models.ForeignKey(
+        'projects.Project',
+        on_delete=models.SET_NULL,
+        related_name='model_owner',
+        null=True)
     status = models.CharField(max_length=2, choices=STATUS, default=CREATED)
     tag = models.CharField(max_length=10, default='latest')
 

components/studio/projects/views.py

@@ -10,6 +10,7 @@
 import markdown
 from .forms import TransferProjectOwnershipForm
 from django.db.models import Q
+from models.models import Model
 
 logger = logging.getLogger(__name__)
 
@@ -172,6 +173,11 @@ def delete(request, user, project_slug):
 
     print("PROJECT RESOURCES DELETED SUCCESFULLY!")
 
+    models = Model.objects.filter(project=project)
+    for model in models:
+        model.status = 'AR'
+        model.save()
+
     project.delete()
 
     return HttpResponseRedirect(next_page, {'message': 'Deleted project successfully.'})