import os
import sqlite3
import sys

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

import models

CIUDADES = [
    'madrid', 'barcelona', 'valencia', 'sevilla', 'zaragoza',
    'malaga', 'murcia', 'palma', 'las-palmas', 'bilbao',
    'alicante', 'cordoba', 'valladolid', 'vigo'
]

if __name__ == '__main__':
    print("Inicializando base de datos...")
    models.init_db()

    print("Migrando ciudad en anuncios existentes...")
    models.migrar_ciudad_existentes(ciudad_default='madrid')

    conn = sqlite3.connect(os.path.join(os.environ.get('DATA_DIR', '/app/data'), 'anuncios.db'))
    cur = conn.execute('SELECT COUNT(*) FROM anuncios')
    count = cur.fetchone()[0]
    conn.close()

    if count == 0:
        print("Generando anuncios aleatorios...")
        models.seed_anuncios(ciudades=CIUDADES, por_ciudad=4)
    else:
        print(f"Ya existen {count} anuncios. No se generan seeds nuevos.")

    print("Listo.")