AIRAVATA-3698 management command to fix up the content type id in pag…

…e revision json
apache · Sep 25, 2023 · dd41406 · dd41406
1 parent 18cd1ab
commit dd41406
Show file tree

Hide file tree

Showing 15 changed files with 30 additions and 304 deletions.
diff --git a/django_airavata/wagtailapps/base/management/commands/fix_page_revision_content_type.py b/django_airavata/wagtailapps/base/management/commands/fix_page_revision_content_type.py
@@ -0,0 +1,23 @@
+
+import json
+
+from django.core.management.base import BaseCommand
+from wagtail.core.models import PageRevision
+
+
+class Command(BaseCommand):
+    help = "Fix the content_type id in the page revisions content_type which may be correct due to being imported from a different Django instance"
+
+    def handle(self, **options):
+        fixed_count = 0
+        for pr in PageRevision.objects.all():
+            content_json = json.loads(pr.content_json)
+            if content_json['content_type'] != pr.page.content_type.id:
+                content_json['content_type'] = pr.page.content_type.id
+                pr.content_json = json.dumps(content_json)
+                pr.save()
+                fixed_count = fixed_count + 1
+        if fixed_count > 0:
+            self.stdout.write(
+                self.style.SUCCESS(f"Successfully fixed the content type of {fixed_count} page revisions")
+            )
diff --git a/django_airavata/wagtailapps/base/management/commands/load_cms_data.py b/django_airavata/wagtailapps/base/management/commands/load_cms_data.py
@@ -35,5 +35,8 @@ def handle(self, **options):
 
         call_command('loaddata', fixture_file, verbosity=0)
         call_command('set_wagtail_site')
+        call_command('fix_page_revision_content_type')
 
-        print(f"{options['filename']} is loaded successfully....!")
+        self.stdout.write(
+            self.style.SUCCESS(f"{options['filename']} is loaded successfully....!")
+        )
diff --git a/django_airavata/wagtailapps/base/management/commands/load_cybergateway_data.py b/django_airavata/wagtailapps/base/management/commands/load_cybergateway_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_default_gateway.py b/django_airavata/wagtailapps/base/management/commands/load_default_gateway.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_hpcgateway_data.py b/django_airavata/wagtailapps/base/management/commands/load_hpcgateway_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_nanoconfinement_data.py b/django_airavata/wagtailapps/base/management/commands/load_nanoconfinement_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_osu_data.py b/django_airavata/wagtailapps/base/management/commands/load_osu_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_phasta_data.py b/django_airavata/wagtailapps/base/management/commands/load_phasta_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_seagrid_data.py b/django_airavata/wagtailapps/base/management/commands/load_seagrid_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_searching_sra_data.py b/django_airavata/wagtailapps/base/management/commands/load_searching_sra_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_simccs_data.py b/django_airavata/wagtailapps/base/management/commands/load_simccs_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_simvascular_data.py b/django_airavata/wagtailapps/base/management/commands/load_simvascular_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_siu_data.py b/django_airavata/wagtailapps/base/management/commands/load_siu_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/load_south_dakota_data.py b/django_airavata/wagtailapps/base/management/commands/load_south_dakota_data.py
diff --git a/django_airavata/wagtailapps/base/management/commands/set_wagtail_site.py b/django_airavata/wagtailapps/base/management/commands/set_wagtail_site.py
@@ -25,16 +25,16 @@ def handle(self, **options):
                 if site_root is None:
                     raise Exception("Could not find site root page!")
                 else:
-                    print(f"Setting root page to {site_root.title}")
+                    self.stdout.write(f"Setting root page to {site_root.title}")
                 Site.objects.create(
                     hostname=hostname,
                     is_default_site=True,
                     site_name=settings.PORTAL_TITLE,
                     root_page=site_root
                 )
-                print(f"Created Site object for domain {hostname}")
+                self.stdout.write(f"Created Site object for domain {hostname}")
         else:
-            print(f"Site object for domain {hostname} already exists")
+            self.stdout.write(f"Site object for domain {hostname} already exists")
 
     def find_root_airavata_page(self, pages):
         for page in pages: