From 1cc8b149c55ebf9601bf20e54d3f1d71a911bb5a Mon Sep 17 00:00:00 2001
From: fmareuil <fmareuil@pasteur.fr>
Date: Wed, 10 Mar 2021 19:15:34 +0100
Subject: [PATCH] too many interactfiles, use filter and update

---
 .../migrations/0068_auto_20210310_1052.py     | 41 ++++++++-----------
 1 file changed, 18 insertions(+), 23 deletions(-)

diff --git a/ippisite/ippidb/migrations/0068_auto_20210310_1052.py b/ippisite/ippidb/migrations/0068_auto_20210310_1052.py
index 526cd0ae..c5a9d853 100644
--- a/ippisite/ippidb/migrations/0068_auto_20210310_1052.py
+++ b/ippisite/ippidb/migrations/0068_auto_20210310_1052.py
@@ -8,30 +8,25 @@ def init_isolevel(apps, schema_editor):
     InteractFile = apps.get_model("ippidb", "InteractFile")
     Chains = Chain.objects.all()
     if Chains:
-        for Chain in Chains:
-            Chain.default_isolevel = 0.5
-            Chain.save()
-    InteractFiles = InteractFile.objects.all()
+        Chain.update(default_isolevel=0.5)
+    InteractFiles = InteractFile.objects.filter(label="all")
     if InteractFiles:
-        for InteractFile in InteractFiles:
-            if InteractFile.label == "all":
-                InteractFile.default_isolevel = 0.7
-                InteractFile.save()
-            if InteractFile.label == "hydrophobic":
-                InteractFile.default_isolevel = 0.3
-                InteractFile.save()
-            if InteractFile.label == "hbond donor-acceptor":
-                InteractFile.default_isolevel = 0.3
-                InteractFile.save()
-            if InteractFile.label == "negative":
-                InteractFile.default_isolevel = 0.08
-                InteractFile.save()
-            if InteractFile.label == "positive":
-                InteractFile.default_isolevel = 0.05
-                InteractFile.save()
-            if InteractFile.label == "backbone":
-                InteractFile.default_isolevel = 0.18
-                InteractFile.save()
+        InteractFiles.update(default_isolevel=0.7)
+    InteractFiles = InteractFile.objects.filter(label="hydrophobic")
+    if InteractFiles:
+        InteractFiles.update(default_isolevel=0.3)
+    InteractFiles = InteractFile.objects.filter(label="hbond donor-acceptor")
+    if InteractFiles:
+        InteractFiles.update(default_isolevel=0.3)
+    InteractFiles = InteractFile.objects.filter(label="negative")
+    if InteractFiles:
+        InteractFiles.update(default_isolevel=0.08)
+    InteractFiles = InteractFile.objects.filter(label="positive")
+    if InteractFiles:
+        InteractFiles.update(default_isolevel=0.05)
+    InteractFiles = InteractFile.objects.filter(label="backbone")
+    if InteractFiles:
+        InteractFiles.update(default_isolevel=0.18)
 
 
 class Migration(migrations.Migration):
-- 
GitLab