feat: reassign number

Author: shulcsm

.github/workflows/ci.yml

@@ -56,4 +56,4 @@ jobs:
 
       - name: Run tests
         # For example, using `pytest`
-        run: uv run pytest test
+        run: uv run pytest tests

README.rst

@@ -27,3 +27,9 @@ ____________
 
 3. Run `python manage.py migrate` to create the enumeration models.
 
+
+
+Development
+____________
+
+DJANGO_SETTINGS_MODULE=enumeration.tests.settings django-admin makemigrations

pytest.ini

@@ -1,3 +1,3 @@
 [pytest]
-DJANGO_SETTINGS_MODULE = test.settings
-pythonpath = . src
+DJANGO_SETTINGS_MODULE = enumeration.tests.settings
+pythonpath = src

src/enumeration/migrations/0002_alter_counter_id_alter_gap_id_alter_sequence_id.py

@@ -0,0 +1,34 @@
+# Generated by Django 4.2.17 on 2024-12-09 04:14
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("enumeration", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="counter",
+            name="id",
+            field=models.BigAutoField(
+                auto_created=True, primary_key=True, serialize=False, verbose_name="ID"
+            ),
+        ),
+        migrations.AlterField(
+            model_name="gap",
+            name="id",
+            field=models.BigAutoField(
+                auto_created=True, primary_key=True, serialize=False, verbose_name="ID"
+            ),
+        ),
+        migrations.AlterField(
+            model_name="sequence",
+            name="id",
+            field=models.BigAutoField(
+                auto_created=True, primary_key=True, serialize=False, verbose_name="ID"
+            ),
+        ),
+    ]

src/enumeration/mixins.py

@@ -0,0 +1,70 @@
+from django.db import models
+from typing import Dict, TYPE_CHECKING
+from enumeration.manager import get_number, format_number, ResetPeriod, truncate_date
+
+if TYPE_CHECKING:
+    from enumeration.models import Sequence
+
+
+class EnumeratedModelMixin(models.Model):
+    sequence = models.ForeignKey(
+        "enumeration.Sequence", null=True, on_delete=models.PROTECT
+    )
+
+    number = models.CharField(max_length=255, null=True, blank=True, db_index=True)
+    position = models.PositiveIntegerField(null=True)
+    counter = models.ForeignKey(
+        "enumeration.Counter", null=True, on_delete=models.PROTECT
+    )
+
+    def get_sequence(self) -> "Sequence":
+        raise NotImplementedError
+
+    def get_enumeration_context(self) -> Dict:
+        return {}
+
+    def assign_number(self):
+        if self.number:
+            raise RuntimeError("Number present")
+
+        if not self.sequence:
+            self.sequence = self.get_sequence()
+
+        self.number, self.position, self.counter_id = get_number(
+            sequence=self.sequence, **self.get_enumeration_context()
+        )
+
+    def reassign_number(self):
+        assert self.sequence and self.position and self.counter
+        ctx = self.get_enumeration_context()
+
+        # Sequecne never resets
+        if self.sequence.reset_period == ResetPeriod.NEVER:
+            # And correct counter
+            if self.counter.period is None:
+                self.number = format_number(
+                    self.sequence.format,
+                    position=self.position,
+                    **ctx,
+                )
+                return
+        # sequnce resets
+        else:
+            # and correct counter period
+            if self.counter.period == truncate_date(
+                self.sequence.reset_period, ctx["date"]
+            ):
+                self.number = format_number(
+                    self.sequence.format,
+                    position=self.position,
+                    **ctx,
+                )
+                return
+
+        # get new number
+        self.number, self.position, self.counter_id = get_number(
+            sequence=self.sequence, **ctx
+        )
+
+    class Meta:
+        abstract = True

src/enumeration/models.py

@@ -1,6 +1,5 @@
 from django.core.exceptions import ValidationError
 from django.db import models
-
 from enumeration.const import ResetPeriod
 from enumeration.validators import validate_format
 
@@ -43,12 +42,3 @@ class Gap(models.Model):
     position = models.PositiveIntegerField()
 
     unique_together = ("counter", "position")
-
-
-# class EnumeratedDocumentMixin(models.Model):
-#     sequence = models.ForeignKey(Sequence)
-#     position = models.PositiveIntegerField(null=True)
-#     number = models.CharField(max_length=255, null=True, blank=True)
-#
-#     class Meta:
-#         abstract = True

test/__init__.py renamed to src/enumeration/tests/__init__.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class TestConfig(AppConfig):
+    name = "enumeration.tests"
+    label = "enumeration_tests"

src/enumeration/tests/apps.py

@@ -0,0 +1,57 @@
+# Generated by Django 4.2.17 on 2024-12-09 04:14
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ("enumeration", "0002_alter_counter_id_alter_gap_id_alter_sequence_id"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Document",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                (
+                    "number",
+                    models.CharField(
+                        blank=True, db_index=True, max_length=255, null=True
+                    ),
+                ),
+                ("position", models.PositiveIntegerField(null=True)),
+                ("date", models.DateField()),
+                (
+                    "counter",
+                    models.ForeignKey(
+                        null=True,
+                        on_delete=django.db.models.deletion.PROTECT,
+                        to="enumeration.counter",
+                    ),
+                ),
+                (
+                    "sequence",
+                    models.ForeignKey(
+                        null=True,
+                        on_delete=django.db.models.deletion.PROTECT,
+                        to="enumeration.sequence",
+                    ),
+                ),
+            ],
+            options={
+                "abstract": False,
+            },
+        ),
+    ]