34 lines
947 B
Python
34 lines
947 B
Python
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.")
|