Added EuropeCountry.reports_can_be_published

Author: epou

tigaserver_app/migrations/0083_europecountry_reports_can_be_published.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.25 on 2025-07-04 14:07
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('tigaserver_app', '0082_remove_iascore'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='europecountry',
+            name='reports_can_be_published',
+            field=models.BooleanField(default=True),
+        ),
+    ]

tigaserver_app/models.py

@@ -662,6 +662,8 @@ class EuropeCountry(models.Model):
     pending_crisis_reports = models.IntegerField(blank=True, null=True, help_text='Number of reports in country assignable to non-supervisors')
     last_crisis_report_n_update = models.DateTimeField(help_text="Last time count was updated", null=True, blank=True)
 
+    reports_can_be_published = models.BooleanField(default=True)
+
     class Meta:
         managed = True
         ordering = ['name_engl']
@@ -2069,10 +2071,7 @@ def save(self, *args, **kwargs):
                 getattr(self, fname) for fname in bite_fieldnames
             )
 
-        if self.type not in self.PUBLISHABLE_TYPES:
-            self.published_at = None
-
-        if not self.is_browsable:
+        if (self.type not in self.PUBLISHABLE_TYPES) or (not self.is_browsable) or (self.country and not self.country.reports_can_be_published):
             self.published_at = None
 
         super(Report, self).save(*args, **kwargs)

tigaserver_app/tests/tests.py

@@ -1,4 +1,5 @@
 from datetime import datetime, timedelta
+import pytest
 import uuid
 
 # Create your tests here.
@@ -2027,6 +2028,29 @@ def test_bite_is_published_on_create(self):
         self.assertEqual(report.published_at, timezone.now())
         self.assertEqual(report.published, True)
 
+    @time_machine.travel("2024-01-01 00:00:00", tick=False)
+    def test_bite_is_not_published_if_country_does_not_allow_public_on_create(self):
+        disabled_publish_country = EuropeCountry.objects.create(
+            cntr_id="RD", name_engl="Random", iso3_code="RND", fid="RD",
+            geom=MultiPolygon(Polygon.from_bbox((-10.0, 35.0, 3.5, 44.0))),
+            reports_can_be_published=False,
+        )
+        point_on_surface = disabled_publish_country.geom.point_on_surface
+        report = Report.objects.create(
+            user=TigaUser.objects.create(),
+            report_id='1234',
+            phone_upload_time=timezone.now(),
+            creation_time=timezone.now(),
+            version_time=timezone.now(),
+            type=Report.TYPE_BITE,
+            location_choice=Report.LOCATION_CURRENT,
+            current_location_lon=point_on_surface.x,
+            current_location_lat=point_on_surface.y,
+        )
+        self.assertEqual(report.country, disabled_publish_country)
+        self.assertIsNone(report.published_at)
+        self.assertEqual(report.published, False)
+
     def test_breeding_site_with_picture_is_published_in_two_days_on_create(self):
         with time_machine.travel("2024-01-01 00:00:00", tick=False) as traveller:
             report = Report.objects.create(
@@ -2065,6 +2089,29 @@ def test_breeding_site_without_picture_is_published_on_create(self):
         self.assertEqual(report.published_at, timezone.now())
         self.assertEqual(report.published, True)
 
+    @time_machine.travel("2024-01-01 00:00:00", tick=False)
+    def test_breeding_site_without_picture_is_not_published_if_country_does_not_allow_public_on_create(self):
+        disabled_publish_country = EuropeCountry.objects.create(
+            cntr_id="RD", name_engl="Random", iso3_code="RND", fid="RD",
+            geom=MultiPolygon(Polygon.from_bbox((-10.0, 35.0, 3.5, 44.0))),
+            reports_can_be_published=False,
+        )
+        point_on_surface = disabled_publish_country.geom.point_on_surface
+        report = Report.objects.create(
+            user=TigaUser.objects.create(),
+            report_id='1234',
+            phone_upload_time=timezone.now(),
+            creation_time=timezone.now(),
+            version_time=timezone.now(),
+            type=Report.TYPE_SITE,
+            location_choice=Report.LOCATION_CURRENT,
+            current_location_lon=point_on_surface.x,
+            current_location_lat=point_on_surface.y,
+        )
+        self.assertEqual(report.country, disabled_publish_country)
+        self.assertIsNone(report.published_at)
+        self.assertEqual(report.published, False)
+
     def test_adult_without_picture_is_published_on_create(self):
         report = Report.objects.create(
             user=TigaUser.objects.create(),
@@ -2079,6 +2126,29 @@ def test_adult_without_picture_is_published_on_create(self):
         )
         self.assertEqual(report.published, True)
 
+    @time_machine.travel("2024-01-01 00:00:00", tick=False)
+    def test_adult_without_picture_is_not_published_if_country_does_not_allow_public_on_create(self):
+        disabled_publish_country = EuropeCountry.objects.create(
+            cntr_id="RD", name_engl="Random", iso3_code="RND", fid="RD",
+            geom=MultiPolygon(Polygon.from_bbox((-10.0, 35.0, 3.5, 44.0))),
+            reports_can_be_published=False,
+        )
+        point_on_surface = disabled_publish_country.geom.point_on_surface
+        report = Report.objects.create(
+            user=TigaUser.objects.create(),
+            report_id='1234',
+            phone_upload_time=timezone.now(),
+            creation_time=timezone.now(),
+            version_time=timezone.now(),
+            type=Report.TYPE_ADULT,
+            location_choice=Report.LOCATION_CURRENT,
+            current_location_lon=point_on_surface.x,
+            current_location_lat=point_on_surface.y,
+        )
+        self.assertEqual(report.country, disabled_publish_country)
+        self.assertIsNone(report.published_at)
+        self.assertEqual(report.published, False)
+
     def test_adult_with_picture_is_not_published_on_create(self):
         report = Report.objects.create(
             user=TigaUser.objects.create(),