from django.core.management.base import BaseCommand
from django.db.models import Count
from inventory.models import CarMake, CarModel, CarSerie, CarTrim
import csv
import os
from django.conf import settings


class Command(BaseCommand):
    help = "Analyzes the car hierarchy to identify makes without models, models without series, and series without trims"

    def add_arguments(self, parser):
        parser.add_argument(
            "--export",
            action="store_true",
            help="Export results to CSV files",
        )
        parser.add_argument(
            "--export-path",
            type=str,
            default="exports",
            help='Directory to export CSV files (default: "exports")',
        )

    def handle(self, *args, **options):
        export = options["export"]
        export_path = options["export_path"]

        # Create export directory if needed
        if export:
            export_dir = os.path.join(settings.BASE_DIR, export_path)
            if not os.path.exists(export_dir):
                os.makedirs(export_dir)

        # Analyze makes without models
        all_makes = CarMake.objects.all()
        total_makes = all_makes.count()
        makes_without_models = CarMake.objects.annotate(
            model_count=Count("carmodel")
        ).filter(model_count=0)
        makes_without_models_count = makes_without_models.count()

        self.stdout.write(self.style.SUCCESS(f"Total car makes: {total_makes}"))
        self.stdout.write(
            self.style.SUCCESS(
                f"Car makes without models: {makes_without_models_count} "
                f"({makes_without_models_count / total_makes * 100:.2f}% of all makes)"
            )
        )

        if makes_without_models_count > 0:
            self.stdout.write("\nSample of car makes without models:")
            for make in makes_without_models[:5]:
                self.stdout.write(f"- {make.name}")

        # Analyze models without series
        all_models = CarModel.objects.all()
        total_models = all_models.count()
        models_without_series = CarModel.objects.annotate(
            serie_count=Count("carserie")
        ).filter(serie_count=0)
        models_without_series_count = models_without_series.count()

        self.stdout.write(self.style.SUCCESS(f"\nTotal car models: {total_models}"))
        self.stdout.write(
            self.style.SUCCESS(
                f"Car models without series: {models_without_series_count} "
                f"({models_without_series_count / total_models * 100:.2f}% of all models)"
            )
        )

        if models_without_series_count > 0:
            self.stdout.write("\nSample of car models without series:")
            for model in models_without_series[:5]:
                self.stdout.write(f"- {model.name} (Make: {model.id_car_make.name})")

        # Analyze series without trims
        all_series = CarSerie.objects.all()
        total_series = all_series.count()
        series_without_trims = CarSerie.objects.annotate(
            trim_count=Count("cartrim")
        ).filter(trim_count=0)
        series_without_trims_count = series_without_trims.count()

        self.stdout.write(self.style.SUCCESS(f"\nTotal car series: {total_series}"))
        self.stdout.write(
            self.style.SUCCESS(
                f"Car series without trims: {series_without_trims_count} "
                f"({series_without_trims_count / total_series * 100:.2f}% of all series)"
            )
        )

        if series_without_trims_count > 0:
            self.stdout.write("\nSample of car series without trims:")
            for serie in series_without_trims[:5]:
                self.stdout.write(
                    f"- {serie.name} (Model: {serie.id_car_model.name}, "
                    f"Make: {serie.id_car_model.id_car_make.name})"
                )

        # Export data if requested
        if export:
            # Export makes without models
            if makes_without_models_count > 0:
                filepath = os.path.join(export_dir, "makes_without_models.csv")
                with open(filepath, "w", newline="") as csvfile:
                    writer = csv.writer(csvfile)
                    writer.writerow(["make_id", "make_name", "is_sa_import"])
                    for make in makes_without_models:
                        writer.writerow(
                            [make.id_car_make, make.name, make.is_sa_import]
                        )
                self.stdout.write(
                    self.style.SUCCESS(f"Exported makes without models to {filepath}")
                )

            # Export models without series
            if models_without_series_count > 0:
                filepath = os.path.join(export_dir, "models_without_series.csv")
                with open(filepath, "w", newline="") as csvfile:
                    writer = csv.writer(csvfile)
                    writer.writerow(["model_id", "model_name", "make_id", "make_name"])
                    for model in models_without_series:
                        writer.writerow(
                            [
                                model.id_car_model,
                                model.name,
                                model.id_car_make.id_car_make,
                                model.id_car_make.name,
                            ]
                        )
                self.stdout.write(
                    self.style.SUCCESS(f"Exported models without series to {filepath}")
                )

            # Export series without trims
            if series_without_trims_count > 0:
                filepath = os.path.join(export_dir, "series_without_trims.csv")
                with open(filepath, "w", newline="") as csvfile:
                    writer = csv.writer(csvfile)
                    writer.writerow(
                        [
                            "serie_id",
                            "serie_name",
                            "model_id",
                            "model_name",
                            "make_id",
                            "make_name",
                        ]
                    )
                    for serie in series_without_trims:
                        writer.writerow(
                            [
                                serie.id_car_serie,
                                serie.name,
                                serie.id_car_model.id_car_model,
                                serie.id_car_model.name,
                                serie.id_car_model.id_car_make.id_car_make,
                                serie.id_car_model.id_car_make.name,
                            ]
                        )
                self.stdout.write(
                    self.style.SUCCESS(f"Exported series without trims to {filepath}")
                )

        # Summary
        self.stdout.write("\n" + "=" * 50)
        self.stdout.write(self.style.SUCCESS("SUMMARY"))
        self.stdout.write("=" * 50)
        self.stdout.write(
            f"Total makes: {total_makes} | Without models: {makes_without_models_count} ({makes_without_models_count / total_makes * 100:.2f}%)"
        )
        self.stdout.write(
            f"Total models: {total_models} | Without series: {models_without_series_count} ({models_without_series_count / total_models * 100:.2f}%)"
        )
        self.stdout.write(
            f"Total series: {total_series} | Without trims: {series_without_trims_count} ({series_without_trims_count / total_series * 100:.2f}%)"
        )