#!/usr/bin/env python
import sys
import os
from pathlib import Path

# Добавляем родительскую директорию в sys.path, чтобы импортировать модули приложения
sys.path.append(str(Path(__file__).parent))

from app.core import SessionLocal
from app.models.catalog_models import Product, Category, Collection, Size

def main():
    """
    Скрипт для проверки данных в базе данных.
    """
    print("Проверка данных в базе данных...")
    
    # Создаем сессию базы данных
    db = SessionLocal()
    
    try:
        # Проверяем наличие категорий
        categories = db.query(Category).all()
        print(f"Категории: {len(categories)}")
        for category in categories:
            print(f"  - {category.id}: {category.name} (slug: {category.slug})")
        
        # Проверяем наличие коллекций
        collections = db.query(Collection).all()
        print(f"Коллекции: {len(collections)}")
        for collection in collections:
            print(f"  - {collection.id}: {collection.name} (slug: {collection.slug})")
        
        # Проверяем наличие размеров
        sizes = db.query(Size).all()
        print(f"Размеры: {len(sizes)}")
        for size in sizes:
            print(f"  - {size.id}: {size.name} (code: {size.code})")
        
        # Проверяем наличие продуктов
        products = db.query(Product).all()
        print(f"Продукты: {len(products)}")
        for product in products[:5]:  # Выводим только первые 5 продуктов
            print(f"  - {product.id}: {product.name} (slug: {product.slug})")
        
    except Exception as e:
        print(f"Ошибка при проверке данных: {str(e)}")
    finally:
        db.close()

if __name__ == "__main__":
    main()