26 lines
894 B
Python
26 lines
894 B
Python
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 |