from django.core.management.base import BaseCommand
from faker import Faker
from src.dieta_especial.solicitacao_dieta_especial.models import (
SolicitacaoDietaEspecial,
)
from src.perfil.models import Usuario
faker = Faker()
fake = Faker("pt-br")
[docs]
def cancelar_dieta(dieta_uuid):
usuario_nutricodae = Usuario.objects.get(email="nutricodae@admin.com")
dieta = SolicitacaoDietaEspecial.objects.get(uuid__startswith=dieta_uuid)
usuario = dieta.criado_por
justificativa = f"<p>{fake.paragraph(nb_sentences=10)}</p>"
dieta.inicia_fluxo_inativacao(user=usuario, justificativa=justificativa)
dieta.cancelar_pedido(user=usuario_nutricodae, justificativa=justificativa)
dieta.save()
return dieta
[docs]
class Command(BaseCommand):
help = """
Cancela uma Dieta Especial, informando o uuid da Dieta.
"""
[docs]
def add_arguments(self, parser):
parser.add_argument(
"--uuid",
"-u",
dest="uuid",
help="Informar uuid (5 primeiros dígitos) da dieta.",
)
[docs]
def handle(self, *args, **options):
uuid = options["uuid"]
dieta = cancelar_dieta(uuid)
self.stdout.write(f"Dieta Cancelada: {dieta}")