"""
Скрипт для добавления роли создателя (owner)
"""
import sqlite3

def migrate():
    conn = sqlite3.connect('music_platform.db')
    cursor = conn.cursor()
    
    try:
        # Добавляем поле is_owner
        cursor.execute("ALTER TABLE users ADD COLUMN is_owner BOOLEAN DEFAULT 0")
        print("✓ Добавлено поле is_owner")
    except sqlite3.OperationalError as e:
        if "duplicate column name" in str(e):
            print("✓ Поле is_owner уже существует")
        else:
            print(f"✗ Ошибка при добавлении is_owner: {e}")
    
    conn.commit()
    
    # Показываем пользователя Leuteg
    cursor.execute("SELECT id, username, is_admin, is_owner FROM users WHERE username = 'Leuteg'")
    user = cursor.fetchone()
    
    if user:
        print(f"\nПользователь найден:")
        print(f"ID: {user[0]}, Username: {user[1]}, Admin: {user[2]}, Owner: {user[3]}")
        
        # Делаем Leuteg создателем
        cursor.execute("UPDATE users SET is_owner = 1, is_admin = 1 WHERE username = 'Leuteg'")
        conn.commit()
        print(f"\n✓ Пользователь 'Leuteg' теперь СОЗДАТЕЛЬ с полными правами!")
    else:
        print("\n✗ Пользователь 'Leuteg' не найден")
    
    conn.close()
    print("\n✓ Миграция завершена!")

if __name__ == "__main__":
    migrate()