inventory_handler.py

from base_handler import *
import re
import datetime
from utils import fuzzy_match, PERM_ADMIN

PLACE_PATTERN = re.compile('^\s*the\s+(.+?)\s+(is|are)\s+(.+?)\s+the\s+(.+?)\.?\s*$',
                           flags=re.I)
SEARCH_PATTERN = re.compile('^\s*where\s+(is|are)\s+(?:all\s+)?the\s+(.+?)\??\s*$',
                            flags=re.I)
LIST_PATTERN = re.compile('^\s*what(?:\s+is|\'s)\s+(.+?)\s+the\s+(.+?)\??\s*$',
                          flags=re.I)


class InventoryHandler(BaseHandler):

    def __init__(self, config, messenger, service_hub):
        super().__init__(config, messenger, service_hub, key="inv", name="Inventory Keeper")

    def help(self, permission):
        if permission >= PERM_ADMIN:
            return {
                'summary': "Remembers where your things are",
                'examples': ["The <item> is in <location>", "Where is the <item>?"],
            }


    def matches_message(self, message):
        return PLACE_PATTERN.match(message) is not None \
               or SEARCH_PATTERN.match(message) is not None \
               or LIST_PATTERN.match(message) is not None


    def handle(self, message, **kwargs):
        if kwargs['permission'] < PERM_ADMIN:
            return "Sorry, I won't tell you where anything is."
        db = kwargs['db']
        match = PLACE_PATTERN.match(message.lower())
        if match:
            return self.handle_place(match, db)
        match = SEARCH_PATTERN.match(message.lower())
        if match:
            return self.handle_search(match, message, db)
        match = LIST_PATTERN.match(message.lower())
        if match:
            return self.handle_list(match, db)
        return "Oh... uh... something went terribly wrong. I'm sorry."


    def handle_button(self, data, **kwargs):
        db = kwargs['db']
        table = db['inventory']
        thing = table.find_one(id=data)

        if not thing:
            return "Uh oh, something went wrong"

        table.delete(id=data)
        msg = self.format_thing("he {} {} no longer {} the {}.", thing)
        answer = {
            'answer': 'T{}'.format(msg),
            'message': 'Okay, t{}'.format(msg),
        }
        return answer


    def handle_list(self, match, db):
        groups = match.groups()
        position = groups[0]
        location = groups[1]

        table = db['inventory']
        things = list(table.all())

        matches = fuzzy_match(
            location, things, lambda thing: thing['location'],
            min_words=1, min_chars_first_word=3,
            min_chars_total=3
            )

        msg = ""
        for match in matches:
            thing = match['item']
            msg += self.format_thing("The {} {} {} the {}.\n", thing)

        if not msg:
            return "There seems to be nothing {} the {}.".format(position, location)
        return msg


    def handle_search(self, match, message, db):
        groups = match.groups()
        plural = groups[0]
        name = groups[1]

        is_all = message.split()[2] == 'all'

        table = db['inventory']
        things = list(table.all())

        matches = fuzzy_match(name, things, lambda thing: thing['name'],
                              min_words=1, min_chars_first_word=3,
                              min_chars_total=3)

        if matches:
            if is_all and len(matches) > 1:
                msg = ""
                for match in matches:
                    thing = match['item']
                    msg += "The {} {} {} the {}.\n".format(thing['name'],
                                                           'are' if thing['plural'] else 'is',
                                                           thing['position'],
                                                           thing['location'])
                    ret = msg
            else:
                thing = matches[0]['item']
                button_data = {
                    'text': "No {} not".format("they're" if thing['plural'] else "it's"),
                    'data': thing['id'],
                }
                msg = "The {} {} {} the {}.".format(thing['name'],
                                                    'are' if thing['plural'] else 'is',
                                                    thing['position'],
                                                    thing['location'])
                ret = {
                    'message': msg,
                    'buttons': [[button_data]]
                }
            return ret

        else:
            return "Sorry, I don't know where the {} {}.".format(name, plural)


    def handle_place(self, match, db):
        groups = match.groups()

        thing = groups[0]
        plural = groups[1] == 'are'
        position = groups[2]
        location = groups[3]

        new_thing = {
            'name': thing,
            'plural': plural,
            'position': position,
            'location': location,
            'timestamp': datetime.datetime.now()
        }

        table = db['inventory']

        old_thing = table.find_one(name=thing)
        table.upsert(new_thing, ['name'])

        if old_thing:
            msg = "The {} {} moved from {} the {} to {} the {}.".format(thing,
                                                                        'have' if plural else 'has',
                                                                        old_thing['position'],
                                                                        old_thing['location'],
                                                                        new_thing['position'],
                                                                        new_thing['location'])
        else:
            msg = self.format_thing("The {} {} now {} the {}.", new_thing)

        return msg


    def format_thing(self, string, thing, singular='is', plural='are'):
        return string.format(
            thing['name'],
            plural if thing['plural'] else singular,
            thing['position'],
            thing['location']
        )