Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save moki9/a7a209e1e428679bd4fcc3a71c8ae1dd to your computer and use it in GitHub Desktop.
Save moki9/a7a209e1e428679bd4fcc3a71c8ae1dd to your computer and use it in GitHub Desktop.
Sqlalchemy: Truncate all tables
def truncate_db(engine):
# delete all table data (but keep tables)
# we do cleanup before test 'cause if previous test errored,
# DB can contain dust
meta = MetaData(bind=engine, reflect=True)
con = engine.connect()
trans = con.begin()
con.execute('SET FOREIGN_KEY_CHECKS = 0;')
for table in meta.sorted_tables:
con.execute(table.delete())
con.execute('SET FOREIGN_KEY_CHECKS = 1;')
trans.commit()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment