Merge pull request #2212 from laws-africa/comment-tasks

Tasks linked to annotations (comments) have code = 'comment'

Author: longhotsummer

indigo_api/migrations/0042_backfill_task_comment_code.py

@@ -0,0 +1,24 @@
+# Generated by Django 4.2.15 on 2024-08-28 06:40
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('indigo_api', '0041_citationalias'),
+    ]
+
+    operations = [
+        migrations.RunSQL(
+            """
+UPDATE indigo_api_task
+SET code = 'comment'
+WHERE id IN (
+    SELECT task_id
+    FROM indigo_api_annotation
+)
+            """,
+            migrations.RunSQL.noop,
+        ),
+    ]

indigo_api/models/documents.py

@@ -711,6 +711,7 @@ def create_task(self, user):
 
         if not self.task:
             task = Task()
+            task.code = 'comment'
             task.country = self.document.work.country
             task.locality = self.document.work.locality
             task.work = self.document.work

indigo_api/models/tasks.py

@@ -102,6 +102,7 @@ class Task(models.Model):
         ('check-update-primary', _('Check / update primary work')),
         ('check-update-repeal', _('Check / update repeal')),
         ('commences-on-date-missing', _("'Commences on' date missing")),
+        ('comment', _("Comment")),
         ('link-amendment-active', _('Link amendment (active)')),
         ('link-amendment-passive', _('Link amendment (passive)')),
         ('link-amendment-pending-commencement', _('Link amendment (pending commencement)')),

indigo_api/serializers.py

@@ -464,6 +464,7 @@ class Meta:
         fields = (
             'id',
             'title',
+            'code',
             'description',
             'country',
             'locality',

indigo_api/tests/test_annotations_api.py

@@ -122,6 +122,7 @@ def test_create_annotation_task(self):
         assert_equal(response.status_code, 201)
         assert_equal(response.data['title'], '"Section 1.": hello')
         assert_equal(response.data['state'], 'open')
+        assert_equal(response.data['code'], 'comment')
         assert_is_none(response.data.get('anchor_id'))
 
     def test_create_annotation_task_anchor_missing(self):