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