def drop_all(): """Drops all database tables""" input_var = input('Are you sure you wish to drop all tables? y/n: ') if input_var == 'y' or input_var == 'Y': db.reflect() db.drop_all() print('Database tables dropped')
def db(app): _db.drop_all() _db.create_all() populate_db() yield _db _db.session.remove() _db.drop_all()
def db(app): print("Creating database") _db.drop_all() _db.create_all() populate_db() yield _db print("Dropping database") _db.session.remove() _db.drop_all()
def tearDown(self): db.session.remove() db.drop_all()