update: create unidade and tag default elements

form_user/apps/contas/management/commands/elements.py

@@ -2,7 +2,7 @@ from decouple import config
 from django.core.management.base import BaseCommand
 from django.utils.translation import gettext_lazy as _
 
-from form_user.apps.contas.models import CustomUser
+from form_user.apps.contas.models import CustomUser, TagUnidade, Unidade
 
 
 class AppException(Exception, BaseCommand):
@@ -31,3 +31,23 @@ class Command(BaseCommand):
                 self.stdout.write(self.style.SUCCESS(_("Superuser created!")))
         else:
             self.stdout.write(self.style.NOTICE(_("Superuser already exists!")))
+
+        if not TagUnidade.objects.count():
+            self.create_tag_unidade(model=TagUnidade, el_list=["primeira", "segunda"])
+        else:
+            self.stdout.write(self.style.NOTICE(_("TagUnidade already exists!")))
+
+        if not Unidade.objects.count():
+            self.create_tag_unidade(model=Unidade, el_list=["qualquer", "coisa"])
+        else:
+            self.stdout.write(self.style.NOTICE(_("Unidade already exists!")))
+
+    def create_tag_unidade(self, model, el_list):
+        tags_uni = []
+
+        for el in el_list:
+            tags_uni.append(model(nome=el))
+
+        created_element = model.objects.bulk_create(tags_uni)
+        if created_element:
+            self.stdout.write(self.style.SUCCESS(f"{model.__name__} created!"))