from django.core.management.base import BaseCommand

from technical.demo_data import clear_wordpress_demo_data, seed_wordpress_demo_data


class Command(BaseCommand):
    help = "Cria dados fictícios de WordPress, plugins, temas, SSL e monitorização sem limpar a base completa."

    def add_arguments(self, parser):
        parser.add_argument("--reset", action="store_true", help="Elimina apenas os dados técnicos DEMO V9.4 antes de os recriar.")
        parser.add_argument("--clear", action="store_true", help="Elimina apenas os dados técnicos DEMO V9.4 e termina.")

    def handle(self, *args, **options):
        if options["clear"]:
            counts = clear_wordpress_demo_data()
            self.stdout.write(self.style.WARNING(f"Dados DEMO V9.4 eliminados: {counts}"))
            return
        totals = seed_wordpress_demo_data(reset=options["reset"])
        self.stdout.write(self.style.SUCCESS("Dados técnicos fictícios V9.4 preparados com sucesso."))
        self.stdout.write(
            "Sites: {sites_total} | Plugins: {plugins_total} | Temas: {themes_total} | SSL: {ssl_total} | Monitores: {monitors_total}".format(**totals)
        )
