import sqlite3
import sys

db_file = sys.argv[1] if len(sys.argv) > 1 else 'milan_baza.db'
conn = sqlite3.connect(db_file)
cursor = conn.cursor()

print(f"Povezan na {db_file}. Vpiši SQL ukaz (za izhod vpiši 'exit'):")

while True:
    try:
        query = input("SQL> ")
        if query.strip().lower() in ['exit', 'quit']:
            break
        if not query.strip():
            continue
        
        cursor.execute(query)
        if query.strip().upper().startswith("SELECT") or query.strip().upper().startswith("PRAGMA"):
            rows = cursor.fetchall()
            for row in rows:
                print(row)
        else:
            conn.commit()
            print("Ukaz uspešno izveden.")
    except Exception as e:
        print("Napaka:", e)

conn.close()