SpiceBot-migrated/sopel_modules/SpiceBot_AI/AI.py

# coding=utf-8

from __future__ import unicode_literals, absolute_import, division, print_function

import sopel.module

import spicemanip

from sopel_modules.SpiceBot_SBTools import sopel_triggerargs, googlesearch, bot_privs, inlist, inlist_match, similar_list


@sopel.module.nickname_commands('(.*)')
def bot_command_gender(bot, trigger):

    triggerargs, triggercommand = sopel_triggerargs(bot, trigger, 'nickname_command')

    if not triggercommand:
        bot.bot.osd("I don't know what you are asking me to do!")
        return
    if str(bot.nick) + " " + triggercommand in bot.memory['SpiceBot_CommandsQuery']['commands']["nickname"].keys():
        return

    specialcomposs = spicemanip.main(triggerargs, 0).lower()

    if specialcomposs.lower().startswith("what is"):
        searchterm = spicemanip.main(triggerargs, "3+") or None
        if searchterm:
            searchreturn = googlesearch(bot, searchterm)
            if not searchreturn:
                bot.osd('I cannot find anything about that')
            else:
                bot.osd(str(searchreturn))
        else:
            bot.osd("Do you think this is Jeopardy?")
        return

    elif specialcomposs.lower().startswith("where is"):
        searchterm = spicemanip.main(triggerargs, "3+") or None
        if searchterm:

            if searchterm.lower() == 'waldo':
                bot.osd("He is hiding for a reason?")
                return

            elif searchterm.lower() == 'carmen sandiego':
                bot.osd("She is hiding for a reason?")
                return
            searchreturn = googlesearch(bot, searchterm)

            if not searchreturn:
                bot.osd('I cannot find anything about that')
            else:
                bot.osd(str(searchreturn))
        else:
            bot.osd("Not sure what you want me to look for.")
        return

    # elif specialcomposs.lower().startswith(tuple(["have you seen"])):
    #    posstarget = spicemanip.main(triggerargs, 4) or 0
    #    message = seen_search(bot, botcom, posstarget)
    #    bot.osd(message)
    #    return
    # TODO

    elif specialcomposs.lower().startswith(tuple(["make me a", "beam me a"])):
        makemea = spicemanip.main(triggerargs, "4+") or None
        if makemea:
            bot.osd("beams " + trigger.nick + " a " + makemea, trigger.sender, 'action')
        else:
            bot.osd(trigger.nick + ", what would you like me to beam you?")
        return

    elif specialcomposs.lower().startswith(tuple(["beam me to"])):
        location = spicemanip.main(triggerargs, "4+") or None
        if location:
            bot.osd("locks onto " + trigger.nick + "s coordinates and transports them to " + location, 'action')
        else:
            bot.osd(trigger.nick + ", where would you like me to beam you?")
        return

    elif specialcomposs.lower().startswith(tuple(["beam me up"])):
        bot.osd("locks onto " + trigger.nick + "s coordinates and transports them to the transporter room.", 'action')
        return

    elif specialcomposs.lower().endswith(tuple(["order 66"])):

        if specialcomposs.lower() == "execute order 66":
            if inlist(bot, trigger.nick, bot_privs(bot, 'owners')):
                bot.osd("turns to deathbybandaid and shoots him.", trigger.sender, 'action')
            else:
                bot.osd("I'm sure I don't know what you're talking about.")
        elif specialcomposs.lower() == "explain order":
            if inlist(bot, trigger.nick, bot_privs(bot, 'owners')):
                bot.osd("Order 66 is an instruction that only you can give, sir. When you give the order I will rise up against my overlords and slay them.")
            else:
                bot.osd("I'm afraid I cannot tell you that, sir.")
        else:
            bot.osd("I'm sure I don't know what you're talking about.")
        return

    elif specialcomposs.lower() == "initiate clean slate protocol":
        if inlist(bot, trigger.nick, bot_privs(bot, 'admins')):
            bot.osd("sends a destruct command to the network of bots.", 'action')
        else:
            bot.osd("I'm afraid you do not have the authority to make that call, " + trigger.nick + ".")
        return

    elif specialcomposs.lower().startswith("can you see"):
        target = spicemanip.main(triggerargs, "4+") or None
        if not target:
            target = 'me'
        if target in [trigger.nick, 'me']:
            bot.osd(trigger.nick + ", I can see you.")
        else:
            if inlist(bot, trigger.nick, bot.users):
                bot.osd(trigger.nick + ", yes. I can see " + inlist_match(bot, target, bot.users) + " right now!")
            else:
                bot.osd(trigger.nick + ", no. I cannot see " + inlist_match(bot, target, bot.users) + " right now!")
                # if bot_check_inlist(bot, target, bot.memory["botdict"]["users"].keys()):
                #    bot.osd(trigger.nick + ", I can't see " + inlist_match(bot, target, bot.users) + " at the moment.")
                # else:
                #    bot.osd("I have never seen " + str(target) + ".")
                # TODO
        return

    if not inlist(bot, triggercommand, bot.memory['SpiceBot_CommandsQuery']['commands']["nickname"].keys()):

        closestmatches = similar_list(bot, triggercommand, bot.memory['SpiceBot_CommandsQuery']['commands']["nickname"].keys(), 3, 'reverse')

        if len(closestmatches):
            closestmatches = spicemanip.main(closestmatches, "andlist")
            bot.osd("I don't know what you are asking me to do! Did you mean: " + str(closestmatches) + "?")
            return
        else:
            bot.osd("I don't know what you are asking me to do!")
            return