from django.core.management.base import BaseCommand
import yaml
import os
from django.utils.text import slugify
from tours.models import Tour


class Command(BaseCommand):
    help = "Import tours from knowledge base"

    def handle(self, *args, **kwargs):
        input_file = "haikal_kb.yaml"

        try:
            with open(input_file, "r", encoding="utf-8") as f:
                kb = yaml.safe_load(f)
        except Exception as e:
            self.stdout.write(
                self.style.ERROR(f"Error reading knowledge base file: {e}")
            )
            return

        workflows = kb.get("user_workflows", {})

        tours_created = 0

        for workflow_name, workflow in workflows.items():
            slug = slugify(workflow_name)
            tour_file = f"{slug}_tour.json"

            tour, created = Tour.objects.update_or_create(
                slug=slug,
                defaults={
                    "name": workflow_name,
                    "description": workflow.get("description", ""),
                    "tour_file": tour_file,
                    "is_active": True,
                },
            )

            if created:
                tours_created += 1
                self.stdout.write(self.style.SUCCESS(f"Created tour: {workflow_name}"))
            else:
                self.stdout.write(self.style.SUCCESS(f"Updated tour: {workflow_name}"))

        self.stdout.write(
            self.style.SUCCESS(f"✅ Imported {tours_created} tours from knowledge base")
        )