def check_database_for(self, meal_name=None, beer_name=None,
category_name=None, user_chat_id=None):
"""Checks the database for various things. Returns truth value."""
if meal_name:
query = Meal.select().where(Meal.name == meal_name)
elif beer_name:
query = Beer.select().where(Beer.name == beer_name)
elif category_name:
query = MealCategory.select().where(MealCategory.name == category_name)
elif user_chat_id:
query = User.select().where(User.chat_id == user_chat_id)
else:
print 'No query specified!'
return False
if len(query) > 0:
return True # return true when results are found
return False
def get_categories(self, desired_categories='meal'):
"""Gets categories from the database and returns them as a list."""
if desired_categories == 'meal':
query = MealCategory.select()
categories = [category.name for category in query]
return categories
