🎨 [#103] made number_generator to util func

Author: bart-maykin

src/openklant/components/klantinteracties/api/tests/test_internetaken.py

@@ -94,7 +94,8 @@ def test_create_internetaak(self):
             response_data = response.json()
             self.assertEqual(
                 response_data["detail"],
-                "Nummer mag maximaal 10 characters bevatten.",
+                "Er kon niet automatisch een opvolgend nummer worden gegenereerd. "
+                "Het maximaal aantal tekens is bereikt.",
             )
 
     def test_update_internetaak(self):

src/openklant/components/klantinteracties/api/tests/test_klantcontacten.py

@@ -113,7 +113,8 @@ def test_create_klantcontact(self):
             response_data = response.json()
             self.assertEqual(
                 response_data["detail"],
-                "Nummer mag maximaal 10 characters bevatten.",
+                "Er kon niet automatisch een opvolgend nummer worden gegenereerd. "
+                "Het maximaal aantal tekens is bereikt.",
             )
 
     def test_create_klantcontact_with_reverse_lookup_fields(self):

src/openklant/components/klantinteracties/api/tests/test_partijen.py

@@ -125,24 +125,6 @@ def test_create_partij(self):
             },
         )
 
-        with self.subTest("voorkeurs_adres_must_be_given_digitaal_adres_validation"):
-            data["nummer"] = "1298329192"
-            data["voorkeursDigitaalAdres"] = {"uuid": str(digitaal_adres2.uuid)}
-            response = self.client.post(list_url, data)
-
-            self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
-            response_data = response.json()
-            self.assertEqual(
-                response_data["invalidParams"],
-                [
-                    {
-                        "name": "voorkeursDigitaalAdres",
-                        "code": "invalid",
-                        "reason": "Het voorkeurs adres moet een gelinkte digitaal adres zijn.",
-                    }
-                ],
-            )
-
         with self.subTest("create_partij_without_foreignkey_relations"):
             data["nummer"] = "1298329192"
             data["digitaleAdressen"] = []
@@ -224,7 +206,26 @@ def test_create_partij(self):
             response_data = response.json()
             self.assertEqual(
                 response_data["detail"],
-                "Nummer mag maximaal 10 characters bevatten.",
+                "Er kon niet automatisch een opvolgend nummer worden gegenereerd. "
+                "Het maximaal aantal tekens is bereikt.",
+            )
+
+        with self.subTest("voorkeurs_adres_must_be_given_digitaal_adres_validation"):
+            data["nummer"] = "1298329194"
+            data["voorkeursDigitaalAdres"] = {"uuid": str(digitaal_adres2.uuid)}
+            response = self.client.post(list_url, data)
+
+            self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+            response_data = response.json()
+            self.assertEqual(
+                response_data["invalidParams"],
+                [
+                    {
+                        "name": "voorkeursDigitaalAdres",
+                        "code": "invalid",
+                        "reason": "Het voorkeurs adres moet een gelinkte digitaal adres zijn.",
+                    }
+                ],
             )
 
     def test_create_persoon(self):

src/openklant/components/klantinteracties/models/internetaken.py

@@ -4,7 +4,7 @@
 from django.db import models
 from django.utils.translation import gettext_lazy as _
 
-from vng_api_common.exceptions import Conflict
+from openklant.components.utils.number_generator import number_generator
 
 from .actoren import Actor
 from .constants import Taakstatus
@@ -77,13 +77,5 @@ class Meta:
         verbose_name_plural = _("interne taken")
 
     def save(self, *args, **kwargs):
-        if not self.nummer:
-            max_nummer = (
-                int(InterneTaak.objects.aggregate(models.Max("nummer"))["nummer__max"])
-                or 0
-            )
-            self.nummer = str(max_nummer + 1).rjust(10, "0")
-            if len(self.nummer) > 10:
-                raise Conflict(_("Nummer mag maximaal 10 characters bevatten."))
-
+        number_generator(self, InterneTaak)
         return super().save(*args, **kwargs)

src/openklant/components/klantinteracties/models/klantcontacten.py

@@ -5,9 +5,8 @@
 from django.utils import timezone
 from django.utils.translation import gettext_lazy as _
 
-from vng_api_common.exceptions import Conflict
-
 from openklant.components.utils.mixins import APIMixin
+from openklant.components.utils.number_generator import number_generator
 
 from .actoren import Actor
 from .constants import Klantcontrol
@@ -105,15 +104,7 @@ class Meta:
         verbose_name_plural = _("klantcontacten")
 
     def save(self, *args, **kwargs):
-        if not self.nummer:
-            max_nummer = (
-                int(Klantcontact.objects.aggregate(models.Max("nummer"))["nummer__max"])
-                or 0
-            )
-            self.nummer = str(max_nummer + 1).rjust(10, "0")
-            if len(self.nummer) > 10:
-                raise Conflict(_("Nummer mag maximaal 10 characters bevatten."))
-
+        number_generator(self, Klantcontact)
         return super().save(*args, **kwargs)
 
     def __str__(self):

src/openklant/components/klantinteracties/models/partijen.py

@@ -6,9 +6,9 @@
 from django.utils.translation import gettext_lazy as _
 
 from vng_api_common.descriptors import GegevensGroepType
-from vng_api_common.exceptions import Conflict
 
 from openklant.components.utils.mixins import APIMixin
+from openklant.components.utils.number_generator import number_generator
 
 from .constants import SoortPartij
 from .mixins import BezoekadresMixin, ContactnaamMixin, CorrespondentieadresMixin
@@ -102,14 +102,7 @@ def clean(self):
                 )
 
     def save(self, *args, **kwargs):
-        if not self.nummer:
-            max_nummer = (
-                int(Partij.objects.aggregate(models.Max("nummer"))["nummer__max"]) or 0
-            )
-            self.nummer = str(max_nummer + 1).rjust(10, "0")
-            if len(self.nummer) > 10:
-                raise Conflict(_("Nummer mag maximaal 10 characters bevatten."))
-
+        number_generator(self, Partij)
         return super().save(*args, **kwargs)
 
 

src/openklant/components/utils/number_generator.py

@@ -0,0 +1,23 @@
+from django.db.models import BigIntegerField, Max, Model
+from django.db.models.functions import Cast
+from django.utils.translation import gettext_lazy as _
+
+from vng_api_common.exceptions import Conflict
+
+
+def number_generator(self, model: Model) -> None:
+    if not self.nummer:
+        max_nummer = (
+            model.objects.annotate(
+                nummer_int=Cast("nummer", output_field=BigIntegerField())
+            ).aggregate(Max("nummer_int"))["nummer_int__max"]
+            or 0
+        )
+        self.nummer = str(max_nummer + 1).rjust(10, "0")
+        if len(self.nummer) > 10:
+            raise Conflict(
+                _(
+                    "Er kon niet automatisch een opvolgend nummer worden gegenereerd. "
+                    "Het maximaal aantal tekens is bereikt."
+                )
+            )