src/liquidcode/commands/make/controller.py - develop-0.2.0 - liquid-g/liquid-code

src/liquidcode/commands/make/controller.py
155 строк6 KB
User
0.3.0-RC1 - исправление: при создании контроллера не находился его вьюв.
04 июл 2026, 17:30
04 июл 2026, 17:30e4af674
"""
Генератор контроллеров для LiquidCode.

Команда: lcc make:controller <name> [--route ROUTE] [--directory DIR] [--force] [--no-view] [--no-interactive]
"""

import argparse
import os

from ..base import (
    generate_from_template,
    TemplateError,
    normalize_class_name,
    normalize_route,
    extract_routes_from_file,
)


def configure_parser(subparser: argparse.ArgumentParser) -> None:
    """
    Добавляет аргументы для команды make:controller.

    Args:
        subparser: Объект ArgumentParser для команды.
    """
    subparser.add_argument("name", nargs="?", help="Имя контроллера (например, User)")
    subparser.add_argument(
        "--route",
        help="Префикс маршрута (по умолчанию: имя в нижнем регистре)",
    )
    subparser.add_argument(
        "--directory",
        default="src/Controller",
        help="Целевая директория (по умолчанию: src/Controller)",
    )
    subparser.add_argument(
        "--force",
        action="store_true",
        help="Перезаписать файл, если он существует",
    )
    subparser.add_argument(
        "--no-view",
        action="store_true",
        help="Не создавать HTML-шаблон (по умолчанию создаётся)",
    )
    subparser.add_argument(
        "--no-interactive",
        action="store_true",
        help="Отключить интерактивный режим (имя должно быть передано через аргумент)",
    )


def _check_route_conflict(normalized_route: str, force: bool) -> bool:
    """
    Проверяет, не занят ли маршрут другим контроллером.
    Возвращает True, если можно создавать (нет конфликта или force=True).
    """
    controller_dir = "src/Controller"
    if not os.path.exists(controller_dir):
        return True  # папки нет – конфликта быть не может

    existing_routes = []
    for root, _, files in os.walk(controller_dir):
        for file in files:
            if file.endswith(".py") and not file.startswith("__"):
                filepath = os.path.join(root, file)
                routes = extract_routes_from_file(filepath)
                existing_routes.extend(routes)

    if normalized_route in existing_routes:
        if force:
            print(
                f"⚠️  Маршрут '{normalized_route}' уже используется, но --force задан, перезаписываем."
            )
            return True
        else:
            print(f"❌ Маршрут '{normalized_route}' уже занят другим контроллером.")
            print(
                "   Используйте другой маршрут или добавьте --force для принудительной перезаписи."
            )
            return False
    return True


def run(args: argparse.Namespace) -> None:
    """
    Выполняет команду make:controller.

    Args:
        args: Объект аргументов командной строки.
    """
    try:
        # 1. Если имя не передано и не отключен интерактив — запрашиваем
        name = args.name
        if not name and not args.no_interactive:
            name = input("Введите имя контроллера (например, User): ").strip()
        if not name:
            print("❌ Имя не может быть пустым")
            return

        # Устанавливаем имя в args для generate_from_template
        args.name = name

        # 2. Нормализуем имя (убираем дублирующий суффикс Controller)
        normalized_name = normalize_class_name(name, suffix="Controller")

        # 3. Нормализуем маршрут
        route = args.route or normalized_name.lower()
        normalized_route = normalize_route(route)

        # 4. Проверяем конфликт маршрутов
        if not _check_route_conflict(normalized_route, args.force):
            return

        # 5. Генерируем контроллер
        extra = {"route": normalized_route, "with_view": not args.no_view}
        created = generate_from_template(
            template_name="controller",
            args=args,
            extra_kwargs=extra,
            default_filename="{name}Controller.py",
            default_dir="src/Controller",
        )
        if created:
            print(f"✅ Создан контроллер: {normalized_name}Controller")
            print(f"   Маршрут: {normalized_route}")
        else:
            print(
                "⚠️  Файл контроллера уже существует. Используйте --force для перезаписи."
            )
            # Если контроллер не создан, шаблон тоже не создаём (если не force)
            if not args.force:
                return

        # 6. Создаём шаблон, если не указан --no-view
        if not args.no_view:
            view_args = argparse.Namespace(
                name=normalized_name,
                directory='templates',
                force=args.force,
            )
            view_created = generate_from_template(
                template_name='view',
                args=view_args,
                extra_kwargs={'name': normalized_name},
                default_filename='{name}.html.j2',
                default_dir='templates',
                template_extension='.html.j2',  # <-- добавлено
            )
            if view_created:
                print(f"✅ Создан шаблон: templates/{normalized_name}.html.j2")
            else:
                print("⚠️  Шаблон уже существует. Используйте --force для перезаписи.")
    except (TypeError, ValueError, TemplateError, IOError) as e:
        print(f"❌ Ошибка: {e}")