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()