SQLAlchemy, borre el contenido de la base de datos, pero no descarte el esquema

Question

Estoy desarrollando una aplicación Pylons que se basa en la existencia de una base de datos, así que estoy usando el reflejo. Tengo un archivo SQL con el esquema que utilicé para crear mi base de datos de prueba. Es por eso que no puedo simplemente usar drop_all y create_all.

Me gustaría escribir algunas pruebas de unidad y me enfrenté al problema de borrar el contenido de la base de datos después de cada prueba. Solo quiero borrar todos los datos pero dejar las tablas intactas. es posible?

La aplicación utiliza Postgres y esto es lo que se debe usar también para las pruebas.

Answer 1

Pregunté por lo mismo en el grupo SQLAlchemy de Google, y obtuve una receta que parece funcionar bien (todas mis tablas están vacías). Ver the thread para referencia.

Mi código (extracto) se ve así:

import contextlib 
from sqlalchemy import MetaData 

meta = MetaData() 

with contextlib.closing(engine.connect()) as con: 
    trans = con.begin() 
    for table in reversed(meta.sorted_tables): 
     con.execute(table.delete()) 
    trans.commit() 

Edición: He modificado el código para eliminar tablas en el orden inverso; supuestamente esto debería asegurar que los niños sean eliminados antes que los padres.

Answer 2

Cómo sobre el uso truncado:

TRUNCATE [TABLA] nombre [...]

(http://www.postgresql.org/docs/8.4/static/sql-truncate.html)

Esto eliminará todos los registros de la tabla, pero deje el esquema en tacto.

Answer 3

Para PostgreSQL usando TRUNCATE:

with contextlib.closing(engine.connect()) as con: 
    trans = con.begin() 
    con.execute('TRUNCATE {} RESTART IDENTITY;'.format(
     ','.join(table.name 
       for table in reversed(Base.metadata.sorted_tables)))) 
    trans.commit() 

Nota: RESTART IDENTITY; asegura que todas las secuencias se restablecen también. Sin embargo, esto es más lento que la receta DELETE de @ aknuds1 en un 50%.

Otra receta es soltar todas las tablas primero y luego volver a crearlas. Esto es más lento en otro 50%:

Base.metadata.drop_all(bind=engine) 
Base.metadata.create_all(bind=engine)