public/form_user
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update: create unidade and tag default elements

Browse source
This commit is contained in:
Lucas F. 2024-02-07 10:08:33 -03:00
parent c602d0bf8d
commit 6b0bcdf723
1 changed files with 21 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

22
form_user/apps/contas/management/commands/elements.py
View file

@ -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!"))