# Python imports from __future__ import print_function, unicode_literals # from pprint import pprint import json # Lib imports from PyInquirer import style_from_dict, Token, prompt, Separator # Application imports from .utils import Logger from .mixins import StylesMixin GROUPS = [ "Search...", "Favorites", "Accessories", "Multimedia", "Graphics", "Office", "Development", "Internet", "Settings", "System", "Game", "Wine", "Other", "[ Exit ]" ] class Context(StylesMixin): """ The menu class has sub methods that are called per run. """ def __init__(self, args): """ Construct a new 'Menu' object which pulls in mixins. :param args: The terminal passed arguments :return: returns nothing """ self.logger = Logger().get_logger("MAIN") # Set the theme self.theme = self.call_method(args.theme) self.menuData = None def mainMenu(self, _grouplist = None): """ Displays the main menu using the defined GROUPS list... """ grouplist = GROUPS if not _grouplist else _grouplist menu = { 'type': 'list', 'name': 'group', 'message': '[ MAIN MENU ]', 'choices': grouplist } return prompt(menu, style=self.theme) def favoritesMenu(self, _grouplist = None): GROUPS = ["[ TO MAIN MENU ]", "This is a stub method for Favorites..."] grouplist = GROUPS if not _grouplist else _grouplist menu = { 'type': 'list', 'name': 'faves', 'message': '[ Favorites ]', 'choices': grouplist } return prompt(menu, style=self.theme) def subMenu(self, data = ["NO GROUP NAME", "NO PROGRAMS PASSED IN"]): group = data[0] progList = data[1] menu = { 'type': 'list', 'name': 'prog', 'message': '[ ' + group + ' ]', 'choices': progList } return prompt(menu, style=self.theme) def searchMenu(self): menu = { 'type': 'input', 'name': 'query', 'message': 'Program you\'re looking for: ', } return prompt(menu, style=self.theme)