diet_installer.py

from sys import stderr

from sqlalchemy.exc import SQLAlchemyError

from datetime import datetime

from diet import DietDatabase, User, FoodType, Meal, MealFoodType


def add_starter_data(session):
    ada = User(name='Ada')
    pancakes = FoodType(name='Pancakes')
    meal = Meal(user=ada, meal_food_types=[MealFoodType(food_type=pancakes, amount=3)], timestamp=datetime.now())
    session.add(meal)


def main():
    try:
        url = DietDatabase.construct_mysql_url('localhost', 3306, 'diet', 'root', 'cse1208')
        diner_database = DietDatabase(url)

        diner_database.ensure_tables_exist()
        print('Tables created.')
        session = diner_database.create_session()
        add_starter_data(session)
        session.commit()
        print('Records created.')
    except SQLAlchemyError as exception:
        print('Database setup failed!', file=stderr)
        print(f'Cause: {exception}', file=stderr)
        exit(1)


if __name__ == '__main__':
    main()