sql_beans_tutorial.py

import database

MENU_PROMPT = """
Choose option:

1. Add a new bean
2. See all beans
3. Find a bean by name
4. See best preparation
5. Exit
"""

def menu():
  connection = database.connect()
  database.create_tables(connection)

  while (user_input := input(MENU_PROMPT)) != "5":
      if user_input == "1":
          prompt_add_new_bean(connection)
      elif user_input == "2":
          prompt_see_all_beans(connection)
      elif user_input == "3":
          prompt_find_bean(connection)
      elif user_input == "4":
          prompt_find_best_method(connection)
      else:
          print("Invalid input")

def prompt_add_new_bean(connection):
    name = input("Enter bean name: ")
    method = input("Enter preparation method: ")
    rating = int(input("Enter rating: "))

    database.add_bean(connection, name, method, rating)

def prompt_see_all_beans(connection):
   beans = database.get_all_beans(connection)
   for bean in beans:
     print(f"{bean[1]} {bean[2]} - {bean[3]}/ 100 ")

def prompt_find_bean(connection):
    name = input("Enter bean name to find: ")
    beans = database.get_beans_by_name(connection, name)

    for bean in beans:
        print(f"{bean[1]} {bean[2]} - {bean[3]}/ 100")

def prompt_find_best_method(connection):
    name = input("Enter bean name to find: ")
    best_method = database.get_best_preparation_for_bean(connection, name)

    print(f"The best bean preparation for {name} is: {best_method[2]}.")


menu()