Merge branch 'projectnumber-resetcounter' into 'main'

Plugin projectnumber: Add command to reset projectnumber

See merge request reportcreator/reportcreator!792

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## Upcoming
+* Plugin projectnumber: Add manage.py command to reset projectnumber
+
+
 ## v2024.96 - 2024-12-04
 * Plugin: webhooks at certain events
 * Plugin: automatically assign project numbers

plugins/projectnumber/management/commands/resetprojectnumber.py

@@ -0,0 +1,23 @@
+from typing import Any
+
+from django.core.management.base import BaseCommand, CommandError, CommandParser
+
+from ...models import ProjectNumber
+
+
+class Command(BaseCommand):
+    help = 'Reset the project number counter'
+
+    def add_arguments(self, parser: CommandParser) -> None:
+        parser.add_argument('value', nargs='?', type=int, default=0, help='New project number counter value')
+
+    def handle(self, *args: Any, value=0, **options: Any) -> str | None:
+        if value < 0:
+            raise CommandError('Project number counter must be a non-negative integer')
+
+        counter, _ = ProjectNumber.objects.update_or_create(
+            pk=1,
+            defaults={'current_id': value},
+            create_defaults={'current_id': value},
+        )
+        print(f'Project number was reset to {counter.current_id}')

plugins/projectnumber/tests/test_plugin_api.py

@@ -3,6 +3,7 @@
 
 import pytest
 from django.apps import apps
+from django.core.management import call_command
 from django.template.defaultfilters import date
 from django.template.defaulttags import NowNode
 from django.utils import dateparse
@@ -19,6 +20,7 @@
 )
 
 from ..app import ProjectNumberPluginConfig
+from ..management.commands import resetprojectnumber
 from ..models import ProjectNumber
 
 
@@ -99,3 +101,14 @@ def test_on_project_saved(self, template, expected):
             # Check project counter increment
             counter.refresh_from_db()
             assert counter.current_id == 1
+
+    def test_command_resetprojectnumber(self):
+        # Reset project counter to a specific value
+        call_command(resetprojectnumber.Command(), '10')
+        counter = ProjectNumber.objects.get(pk=1)
+        assert counter.current_id == 10
+
+        # Reset to 0
+        call_command(resetprojectnumber.Command())
+        counter.refresh_from_db()
+        assert counter.current_id == 0