Exclusive django 3.1 support

uptick · Oct 5, 2020 · ea73190 · ea73190
1 parent 093d9b4
commit ea73190
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 36 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
diff --git a/src/django_dramatiq_pg/migrations/0001_initial.py b/src/django_dramatiq_pg/migrations/0001_initial.py
@@ -1,18 +1,14 @@
 # Generated by Django 2.2.9 on 2020-02-16 23:57
 
-import django.contrib.postgres.fields.jsonb
 from django.db import migrations, models
 
 
 class Migration(migrations.Migration):
-
     initial = True
 
-    dependencies = []
-
     operations = [
         migrations.CreateModel(
-            name="QueuedJob",
+            name="BackgroundJob",
             fields=[
                 ("message_id", models.UUIDField(primary_key=True, serialize=False)),
                 ("queue_name", models.TextField(default="default")),
@@ -29,8 +25,8 @@ class Migration(migrations.Migration):
                     ),
                 ),
                 ("mtime", models.DateTimeField()),
-                ("message", django.contrib.postgres.fields.jsonb.JSONField()),
-                ("result", django.contrib.postgres.fields.jsonb.JSONField()),
+                ("message", models.jsonb.JSONField()),
+                ("result", models.jsonb.JSONField()),
                 ("result_ttl", models.DateTimeField()),
             ],
             options={"db_table": "queue", "managed": False},

diff --git a/src/django_dramatiq_pg/migrations/0002_auto_20200329_2103.py b/src/django_dramatiq_pg/migrations/0002_auto_20200329_2103.py
diff --git a/src/django_dramatiq_pg/models.py b/src/django_dramatiq_pg/models.py
@@ -1,10 +1,8 @@
-from django.contrib.postgres.fields import JSONField
 from django.db import models
 from django.utils import timezone
 
 
 class BackgroundJob(models.Model):
-
     class STATE(models.TextChoices):
         QUEUED = "queued"
         CONSUMED = "consumed"
@@ -15,8 +13,8 @@ class STATE(models.TextChoices):
     queue_name = models.TextField(default="default")
     state = models.CharField(max_length=16, default=STATE.QUEUED, choices=STATE.choices)
     mtime = models.DateTimeField(default=timezone.now)
-    message = JSONField(blank=True, null=True)
-    result = JSONField(blank=True, null=True)
+    message = models.JSONField(blank=True, null=True)
+    result = models.JSONField(blank=True, null=True)
     result_ttl = models.DateTimeField()
 
     class Meta: