bothub-tutorial-wordbook-3

bot.py

from bothub_client.bot import BaseBot
from bothub_client.messages import Message


class Bot(BaseBot):
    def handle_message(self, event, context):
        content = event['content']
        if content.startswith('/'):
            if content.startswith('/list'):
                self.send_word_list(event)

    def send_word_list(self, event):
        content = event['content']
        splitted = content.split()

        words_to_count = self.get_user_data()
        sorted_list = sorted(words_to_count.items(), key=lambda d: d[1], reverse=True)

        start_pos = int(splitted[1]) if len(splitted) > 1 else 0
        paged_list = sorted_list[start_pos:start_pos+10]
        word_list = '\n'.join([
            '{}, {}'.format(word, count)
            for word, count
            in paged_list
        ])

        has_next = len(paged_list) == 10
        message = Message(event).set_text(word_list)

        if has_next:
            message.add_postback_button('Next list', '/list {}'.format(start_pos+10))
        self.send_message(message)