"""
migrate_category.py — přidá sloupec category do photos (idempotentní)
a provede konkrétní UPDATE podle pravidel.
"""
import psycopg2

DB = dict(host="192.168.1.76", port=5432, user="vladimir.buzalka",
          password="Vlado7309208104++", database="fotky_buzalkovi")

conn = psycopg2.connect(**DB)
conn.autocommit = True
cur = conn.cursor()

# 1. Přidat sloupec category pokud neexistuje
cur.execute("""
    ALTER TABLE photos
        ADD COLUMN IF NOT EXISTS category VARCHAR(100);
""")
print("Sloupec category OK.")

# 2. BolyMedia SG520 → wanted=TRUE, category='Fotopast'
cur.execute("""
    UPDATE photos
    SET wanted   = TRUE,
        category = 'Fotopast'
    WHERE camera_make = 'BolyMedia'
      AND camera_model = 'SG520';
""")
print(f"SG520 aktualizováno: {cur.rowcount:,} řádků → wanted=TRUE, category='Fotopast'")

conn.close()