Adds a fallback for the first migrations with JSONField

django-notifications · Apr 20, 2024 · 21c0a63 · 21c0a63
1 parent 21a0b95
commit 21c0a63
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/notifications/migrations/0003_notification_data.py b/notifications/migrations/0003_notification_data.py
@@ -1,6 +1,11 @@
 # -*- coding: utf-8 -*-
 from django.db import migrations, models
 
+try:
+    from jsonfield.fields import JSONField
+except ImportError:
+    from django.db.models import JSONField
+
 
 class Migration(migrations.Migration):
     dependencies = [
@@ -11,7 +16,7 @@ class Migration(migrations.Migration):
         migrations.AddField(
             model_name="notification",
             name="data",
-            field=models.JSONField(null=True, blank=True),
+            field=JSONField(null=True, blank=True),
             preserve_default=True,
         ),
     ]
diff --git a/notifications/migrations/0009_alter_notification_options_and_more.py b/notifications/migrations/0009_alter_notification_options_and_more.py
@@ -5,6 +5,11 @@
 from django.conf import settings
 from django.db import migrations, models
 
+try:
+    from jsonfield.fields import JSONField
+except ImportError:
+    from django.db.models import JSONField
+
 
 class Migration(migrations.Migration):
     dependencies = [
@@ -62,7 +67,7 @@ class Migration(migrations.Migration):
         migrations.AlterField(
             model_name="notification",
             name="data",
-            field=models.JSONField(blank=True, null=True, verbose_name="data"),
+            field=JSONField(blank=True, null=True, verbose_name="data"),
         ),
         migrations.AlterField(
             model_name="notification",