squashed migrations; added Bulk update

EBI-Metagenomics · Sep 13, 2023 · cd0345f · cd0345f
1 parent df9e5c4
commit cd0345f
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 110 deletions.
diff --git a/.../0013_analysisjob_analysis_summary_two.py → ...0011_analysisjob_analysis_summary_json.py b/.../0013_analysisjob_analysis_summary_two.py → ...0011_analysisjob_analysis_summary_json.py
@@ -1,18 +1,18 @@
-# Generated by Django 3.2.18 on 2023-09-12 17:40
+# Generated by Django 3.2.18 on 2023-09-13 10:24
 
 from django.db import migrations, models
 
 
 class Migration(migrations.Migration):
 
     dependencies = [
-        ('emgapi', '0012_remove_analysisjob_job_operator_2'),
+        ('emgapi', '0010_runextraannotation'),
     ]
 
     operations = [
         migrations.AddField(
             model_name='analysisjob',
-            name='analysis_summary_two',
-            field=models.JSONField(blank=True, db_column='ANALYSIS_SUMMARY_TWO', null=True),
+            name='analysis_summary_json',
+            field=models.JSONField(blank=True, db_column='ANALYSIS_SUMMARY_JSON', null=True),
         ),
     ]
diff --git a/emgapi/migrations/0011_analysisjob_job_operator_2.py b/emgapi/migrations/0011_analysisjob_job_operator_2.py
diff --git a/emgapi/migrations/0012_remove_analysisjob_job_operator_2.py b/emgapi/migrations/0012_remove_analysisjob_job_operator_2.py
diff --git a/emgapi/migrations/0014_auto_20230912_1741.py b/emgapi/migrations/0014_auto_20230912_1741.py
diff --git a/emgapi/migrations/0015_auto_20230912_1748.py b/emgapi/migrations/0015_auto_20230912_1748.py
diff --git a/emgapi/migrations/0016_auto_20230912_1749.py b/emgapi/migrations/0016_auto_20230912_1749.py
diff --git a/emgapianns/management/commands/import_analysis_summaries.py b/emgapianns/management/commands/import_analysis_summaries.py
@@ -10,24 +10,26 @@ def add_arguments(self, parser):
 
     def handle(self, *args, **options):
         batch_number = options['batch_number']
-        batch_size = 10000  # Set your desired batch size here
+        batch_size = 10000
 
         try:
-            # Calculate the starting and ending index for the batch
             start_index = (batch_number - 1) * batch_size
             end_index = batch_number * batch_size
 
-            # Get AnalysisJob records for the specified batch
             analysis_jobs = AnalysisJob.objects.all()[start_index:end_index]
 
-            # Print the number of records in the batch
             self.stdout.write(self.style.SUCCESS(f'Processing batch {batch_number} of {len(analysis_jobs)} records.'))
 
+            updated_records = []
+
             for analysis_job in analysis_jobs:
                 analysis_summary = analysis_job.analysis_summary
                 if analysis_summary:
                     analysis_job.analysis_summary_json = analysis_summary
-                    analysis_job.save()
+                    updated_records.append(analysis_job)
+
+            if updated_records:
+                AnalysisJob.objects.bulk_update(updated_records, ['analysis_summary_json'])
 
             self.stdout.write(self.style.SUCCESS(f'Values copied successfully for batch {batch_number}.'))
         except AnalysisJob.DoesNotExist: