Add database migration scripts and update models for JSON support

This commit is contained in:
2024-12-19 23:55:37 +01:00
parent abbc23ae46
commit ed7eb57f6c
11 changed files with 299 additions and 11 deletions
+26
View File
@@ -0,0 +1,26 @@
from sqlalchemy.orm import sessionmaker
from crawler_db_model import WebsPorVisitar, WebsVisitadas, engine
# Crear sesión de base de datos
Session = sessionmaker(bind=engine)
session = Session()
def limpiar_base_datos():
"""Elimina todos los registros de las tablas webs_por_visitar y webs_visitadas."""
try:
# Eliminar todos los registros de WebsPorVisitar
session.query(WebsPorVisitar).delete()
# Eliminar todos los registros de WebsVisitadas
session.query(WebsVisitadas).delete()
# Confirmar cambios
session.commit()
print("La base de datos ha sido limpiada completamente.")
except Exception as e:
print(f"Error al limpiar la base de datos: {e}")
session.rollback()
if __name__ == "__main__":
# Limpiar la base de datos cuando se ejecuta este script directamente
# limpiar_base_datos()
pass