From 5783a0031857990b1d19e87d396d5f4cc3982fa6 Mon Sep 17 00:00:00 2001
From: Thomas Rusiecki <rusieckithomas@gmail.com>
Date: Fri, 20 Dec 2024 15:25:38 -0300
Subject: [PATCH] command for adding default states

---
 .../commands/create_default_states.py         | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 action/management/commands/create_default_states.py

diff --git a/action/management/commands/create_default_states.py b/action/management/commands/create_default_states.py
new file mode 100644
index 0000000..6daccb2
--- /dev/null
+++ b/action/management/commands/create_default_states.py
@@ -0,0 +1,43 @@
+#!/usr/bin/env python3
+import logging
+from django.core.management.base import BaseCommand
+from action.models import StateDefinition, Institution
+from django.utils.translation import gettext as _
+
+logger = logging.getLogger('django')
+
+class Command(BaseCommand):
+    help = 'Create default StateDefinitions for a given institution. "'
+
+    def add_arguments(self, parser):
+        parser.add_argument('institution_name', type=str, help='The name of the institution')
+
+    def handle(self, *args, **kwargs):
+        default_states = [
+            _("INBOX"),
+            _("VISUAL INSPECTION"),
+            _("REPAIR"),
+            _("INSTALL"),
+            _("TEST"),
+            _("PACKAGING"),         
+            _("DONATION"),
+            _("DISMANTLE")
+        ]
+
+        institution_name = kwargs['institution_name']
+        institution = Institution.objects.filter(name=institution_name).first()
+
+        if not institution:
+            txt = "No institution found for: %s. Please create an institution first"
+            logger.error(txt, institution.name)
+            return
+
+        for state in default_states:
+            state_def, created = StateDefinition.objects.get_or_create(
+                institution=institution,
+                state=state
+            )
+            if created:
+                self.stdout.write(self.style.SUCCESS(f'Successfully created state: {state}'))
+            else:
+                self.stdout.write(self.style.WARNING(f'State already exists: {state}'))