def add_upcoming_buttons(self, start):
btn_upcoming = InlineKeyboardButton(text=keyboard_text['upcoming'],
callback_data=make_callback_data(
level=Level.movies_list.value,
source=Source.upcoming.value,
start=start))
btn_upcoming_genre = InlineKeyboardButton(
text=keyboard_text['upcoming_genre'],
callback_data=make_callback_data(level=Level.genre_menu.value,
source=Source.upcoming.value,
start=start))
self._keyboard.add(btn_upcoming, btn_upcoming_genre)
return self
def add_top_rated_buttons(self, start):
btn_top_rated = InlineKeyboardButton(text=keyboard_text['top_rated'],
callback_data=make_callback_data(
level=Level.movies_list.value,
source=Source.top_rated.value,
start=start))
btn_top_rated_genre = InlineKeyboardButton(
text=keyboard_text['top_rated_genre'],
callback_data=make_callback_data(level=Level.genre_menu.value,
source=Source.popular.value,
start=start))
self._keyboard.add(btn_top_rated, btn_top_rated_genre)
return self
def add_genres_buttons(self, start, source):
for genre in all_genres:
callback_data = make_callback_data(level=Level.movies_list.value,
source=source,
genre_id=genre["id"],
start=start)
btn = InlineKeyboardButton(text=genre["name"].title(),
callback_data=callback_data)
self._keyboard.add(btn)
return self
def add_movies_buttons(self, source, **kwargs):
movies_list = get_movies_list(source, kwargs["genre_id"],
kwargs["movie_name"])
for movie_id, movie_name in movies_list.items():
kwargs["movie_id"] = movie_id
kwargs["level"] = Level.movie_info.value
btn = InlineKeyboardButton(text=movie_name,
callback_data=make_callback_data(
source=source, **kwargs))
self._keyboard.add(btn)
return self
def add_return_button(self, place_to_return, **kwargs):
if place_to_return == Level.movies_list.value:
text = keyboard_text['return_to_movie']
elif place_to_return == Level.genre_menu.value:
text = keyboard_text['return_to_genre']
else:
text = keyboard_text['return']
kwargs["level"] = place_to_return
btn_return = InlineKeyboardButton(
text=text, callback_data=make_callback_data(**kwargs))
self._keyboard.add(btn_return)
return self