import sys
import os

sys.path.insert(0, '/home/vh13236/przydasie')
os.chdir('/home/vh13236/przydasie')
os.environ['DJANGO_SETTINGS_MODULE'] = 'finanse.settings'

import django
django.setup()

from django.db import connection

with connection.cursor() as cursor:
    # Dodaj kolumnę note do transakcji (jeśli nie istnieje)
    try:
        cursor.execute("ALTER TABLE app_transaction ADD COLUMN note TEXT NOT NULL DEFAULT ''")
        print("Kolumna 'note' dodana do app_transaction")
    except Exception as e:
        print(f"Kolumna 'note' już istnieje: {e}")

    # Utwórz tabelę kategorii
    cursor.execute("""CREATE TABLE IF NOT EXISTS app_category (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        emoji TEXT NOT NULL DEFAULT '💰',
        color TEXT NOT NULL DEFAULT '#BA7517',
        user_id INTEGER NOT NULL REFERENCES app_user(id) ON DELETE CASCADE,
        UNIQUE(user_id, name)
    )""")
    print("Tabela app_category gotowa")

    # Utwórz domyślne sesje i inne tabele jeśli brakuje
    cursor.execute("""CREATE TABLE IF NOT EXISTS django_session (
        session_key VARCHAR(40) NOT NULL PRIMARY KEY,
        session_data TEXT NOT NULL,
        expire_date DATETIME NOT NULL
    )""")

print("\nMigracja zakończona pomyślnie!")