def __call__(self):
query_categories = Category.get_categories()
category_schema = CategorySchema(many=True)
categories = category_schema.dump(query_categories)
return jsonify({"categories": categories})
class CategoryPage:
"""Class CategoryPage."""
def __init__(self):
"""Initialise."""
self.model = Category()
self.categories = self.model.get_categories()
self.view = CategoryView(self.categories)
def get_command(self) -> str:
"""Return user's category choice."""
choice = self.view.get_input()
choices = {num: f"goto_category_{num}" for num in range(1, len(self.categories) + 1)}
command = choices.get(command, "")
return command
class CategoryPage:
"""Class CategoryPage."""
def __init__(self):
"""Initialise."""
self.category_model = Category()
self.categories = self.category_model.get_categories()
def get_input(self) -> str:
"""Return user's category choice."""
user_choice = None
while user_choice not in range(len(self.categories)):
try:
user_choice = CategoryView.display_input()
category_choice = self.categories[user_choice][user_choice]
except ValueError:
pass
except IndexError:
pass
else:
return category_choice
