manage.py

'''Calling methods to create and drop tables and create admin'''

from app.api.v2.model import User, FoodMenu, FoodOrder
from run import app
from app.api.v2.database import DatabaseConnection


class Connection(DatabaseConnection):
    def __init__(self):
        super().__init__()

    def drop(self):
        '''drop all tables'''
        User().drop_tables()
        FoodOrder().drop_tables()
        FoodMenu().drop_tables()


    def create(self):
        '''craetes tables'''
        foodorder = FoodOrder()
        foodorder.create_table()

        foodmenu = FoodMenu()
        foodmenu.create_table()

        user = User()
        user.create_table()

    def create_admin(self):
        '''creates an admin'''
        admin = User(username='Admin', email='admin@gmail.com',
                    password='Admin123', is_admin=True)
        admin.add()

if __name__ == '__main__':
    with app.app_context():
        Connection().drop()
        Connection().create()
        Connection().create_admin()