#!/usr/bin/env python3
"""
Миграция: Добавление поля image_path в таблицу services
"""
import sqlite3
from pathlib import Path

BASE_DIR = Path(__file__).resolve().parent.parent
DATABASE_PATH = BASE_DIR / 'database' / 'domovoy.db'

def migrate():
    conn = sqlite3.connect(DATABASE_PATH)
    cursor = conn.cursor()
    
    # Проверяем есть ли уже колонка
    cursor.execute("PRAGMA table_info(services)")
    columns = [col[1] for col in cursor.fetchall()]
    
    if 'image_path' in columns:
        print("✅ Колонка image_path уже существует")
        conn.close()
        return
    
    # Добавляем колонку
    cursor.execute('''
        ALTER TABLE services
        ADD COLUMN image_path TEXT NULL
    ''')
    
    conn.commit()
    conn.close()
    
    print("✅ Миграция выполнена: добавлена колонка image_path в services")

if __name__ == '__main__':
    migrate()
