from django.conf import settings
from django.core.management.base import BaseCommand
from src.dieta_especial.solicitacao_dieta_especial.models import (
SolicitacaoDietaEspecial,
)
[docs]
def deletar_dieta(dieta_uuid):
dieta = SolicitacaoDietaEspecial.objects.get(uuid__startswith=dieta_uuid)
dieta.delete()
[docs]
class Command(BaseCommand):
help = """
Deleta 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):
if settings.DEBUG:
uuid = options["uuid"]
deletar_dieta(uuid)
self.stdout.write("Dieta Deletada.")
else:
self.stdout.write("Usar somente em Dev.")