Master

ippisite/ippidb/management/commands/clean_targetcentric.py

@@ -34,7 +34,12 @@ class Command(AppCommand):
     help = "Load Cavities from global csv file and delete all Cavities not present in the csv"
 
     def __init__(
-        self, stdout=None, stderr=None, no_color=False, force_color=False, task=None,
+        self,
+        stdout=None,
+        stderr=None,
+        no_color=False,
+        force_color=False,
+        task=None,
     ):
         super(Command, self).__init__(
             stdout=stdout, stderr=stderr, no_color=no_color, force_color=force_color
@@ -113,5 +118,16 @@ class Command(AppCommand):
                 )
                 if delete:
                     pdb.delete()
+
+        pdbs = PDB.objects.filter(chain__cavity=None)
+        for pdb in pdbs:
+            if pdb.no_cavities():
+                clean = 1
+                self.stdout.write(
+                    self.style.WARNING("WARNING {} has no cavities".format(pdb.code))
+                )
+                if delete:
+                    pdb.delete()
+
         if not clean:
             self.stdout.write(self.style.SUCCESS("No PDB to remove"))