def drop_tables():
database.connect()
# Initialize db by deleting all tables
Item.drop_table(fail_silently=True)
User.drop_table(fail_silently=True)
Address.drop_table(fail_silently=True)
Order.drop_table(fail_silently=True)
OrderItem.drop_table(fail_silently=True)
Picture.drop_table(fail_silently=True)
database.close()
def drops_all_tables(database):
"""Doesn't drop unknown tables."""
tables = database.get_tables()
for table in tables:
if table == 'item':
Item.drop_table()
if table == 'order':
Order.drop_table()
if table == 'user':
User.drop_table()
if table == 'orderitem':
OrderItem.drop_table()
if table == 'address':
Address.drop_table()
if table == 'picture':
Picture.drop_table()
if table == 'favorite':
Favorite.drop_table()
