Ejemplo n.º 1
0
class BookAuthor(MenuScreen):
    """ Authors screen """
    def __init__(self, util, listeners, ch, f, go_author, parser, base_url,
                 voice_assistant, d):
        """ Initializer
        
        :param util: utility object
        :param listeners: screen listeners
        :param ch: selected character
        :param f: selected filter
        :param go_authors: callback
        :param parser: authors parser
        :param base_url: url
        :param d: dictionary with menu button flags 
        """
        self.util = util
        self.factory = Factory(util)
        self.base_url = base_url
        self.config = util.config

        self.parser = parser
        self.current_author_char = ch
        self.current_author_filter = f
        self.go_author = go_author
        self.author_cache = {}
        self.title = self.config[LABELS][KEY_AUTHORS]

        MenuScreen.__init__(self, util, listeners, MENU_ROWS, MENU_COLUMNS,
                            voice_assistant, d, self.turn_page)
        m = self.factory.create_book_author_menu_button

        self.authors_menu = MultiPageMenu(util, self.next_page,
                                          self.previous_page, self.set_title,
                                          self.reset_title, self.go_to_page, m,
                                          MENU_ROWS, MENU_COLUMNS, None,
                                          (0, 0, 0), self.menu_layout)
        self.set_menu(self.authors_menu)

    def set_current(self, ch=None, f=None):
        """ Apply selected character and filter
        
        :param ch: selected character
        :param f: selected filter
        """
        if not ch and not f:
            return

        self.set_loading(self.config[LABELS][KEY_AUTHORS])

        self.total_pages = 0
        self.current_page = 1
        self.current_author_char = ch
        self.current_author_filter = f

        try:
            self.author_cache[ch]
        except:
            self.get_authors()

        self.turn_page()
        self.reset_loading()

    def get_authors(self):
        """ Get authors from parser """

        self.parser.author_parser.current_author_char = self.current_author_char
        self.parser.author_parser.url = self.base_url
        authors = self.parser.get_authors()
        if authors:
            self.author_cache[self.current_author_char] = authors

    def turn_page(self):
        """ Turn authors page """

        self.authors_menu.set_items({}, 0, self.go_author)

        filtered_authors = []
        for a in self.author_cache[self.current_author_char]:
            if self.current_author_filter:
                if a[AUTHOR_NAME].startswith(self.current_author_filter):
                    filtered_authors.append(a)
            else:
                filtered_authors.append(a)

        start = (self.current_page - 1) * PAGE_SIZE
        end = self.current_page * PAGE_SIZE
        page = filtered_authors[start:end]
        self.author_dict = self.factory.create_book_author_items(page)
        self.authors_menu.set_items(self.author_dict, 0, self.go_author, False)
        self.authors_menu.align_content(ALIGN_LEFT)
        self.authors_menu.select_by_index(0)
        self.authors_menu.clean_draw_update()

        self.total_pages = math.ceil(len(filtered_authors) / PAGE_SIZE)

        left = str(self.current_page - 1)
        if self.total_pages == 0:
            right = "0"
        else:
            right = str(self.total_pages - self.current_page)
        self.navigator.left_button.change_label(left)
        self.navigator.right_button.change_label(right)
        self.set_title(self.current_page)

    def add_screen_observers(self, update_observer, redraw_observer):
        """ Add screen observers
        
        :param update_observer: observer for updating the screen
        :param redraw_observer: observer to redraw the whole screen
        """
        self.navigator.add_observers(update_observer, redraw_observer)
        self.authors_menu.add_menu_loaded_listener(redraw_observer)
        self.authors_menu.add_menu_observers(update_observer,
                                             redraw_observer,
                                             release=False)