phenny/modules/botsnack.py

#!/usr/bin/python3
"""
botsnack.py - .botsnack module (aka simulated hunger 1.0)
author: mutantmonkey <mutantmonkey@mutantmonkey.in>
author: Casey Link <unnamedrambler@gmail.com>

This module simulates bot hunger and provides a mechanism
for users to feed the bot.

To prevent abuse when the bot gets very, very full, it explodes
and enters a random cooldown period for 3-10 minutes. When in this
cooldown period all calls to botsnack are ignored.
"""

import random, math, time

# the rate that affects how much eating a snack nourishes the bot
# smaller number = less nourishment = more snacks can be eaten (before fullness)
# larger number = more nourishment = less snacks can be eaten
r_eat = 0.05

# the rate that affects how fast the bot becomes hungry over time
# smaller number = the bot gets hungry slower
# larger number = the bot gets hungry faster
r_hunger = 0.005

def increase_hunger(current_hunger, x):
    # more hungry === value closer to 0
    return current_hunger * math.exp(-r_hunger * x)

def decrease_hunger(current_hunger, food_value):
    # less hungry === closer to 100
    if current_hunger > 50: # exponential growth
        return min(100, current_hunger * math.exp(r_eat*food_value))
    else: # linear increase
        return current_hunger + food_value

def botsnack(phenny, input):
    """.botsnack - Feed me a bot snack."""

    now = time.time()

    # 0. Handle cooldown.
    #    Check if the cooldown period has elapsed, if not, then
    #    ignore this invocation. Else reset to the default state
    if botsnack.coolingdown:
        if now - botsnack.coolingstarted > botsnack.coolingperiod:
            print("cooling down over, reseting")
            botsnack.coolingdown = False
            botsnack.hunger = 50.0
            botsnack.last_tick = now
        else:
            print("cooling down! %s < %s" %(now - botsnack.coolingstarted, botsnack.coolingperiod))
            return # ignore!

    # 1. Time has has passed, so the bot has gotten
    #    hungry. Lets increase his/her hunger proportionate
    #    to the amount of time that has passeed.
    delta = now - botsnack.last_tick
    old_hunger = botsnack.hunger

    botsnack.hunger = increase_hunger(old_hunger, delta)

    print("hunger was %s, increased to %s" %(old_hunger, botsnack.hunger))

    botsnack.last_tick = now

    # 2. Eat some food. Send resposne

    old_hunger = botsnack.hunger
    botsnack.hunger = decrease_hunger(old_hunger, random.uniform(1,5))
    print("hunger was %s, decreased to %s" %(old_hunger, botsnack.hunger))

    if botsnack.hunger > 95: # special case to prevent abuse
        phenny.say("Too much food!")
        phenny.do("explodes")
        botsnack.coolingperiod = random.uniform(3,10)*60
        botsnack.coolingstarted = now
        botsnack.coolingdown = True
        return

    if botsnack.hunger > 90:
        messages = ["I don't think this will fit...", "Ugh, no more please", "Seriously, I can't eat anymore!", "/me shudders but downs the snack anyways"]
    elif botsnack.hunger > 70:
        messages = ["Thanks, but that's enough", "I suppose I could have one more", "If you insist"]
    elif botsnack.hunger > 50:
        messages = ["Om nom nom", "Delicious, thanks!", "Yummy!", "Wow! That's delicious"]
    elif botsnack.hunger > 30:
        messages = ["That really hit the spot!", "/me smacks lips", "Mmmmm!"]
    elif botsnack.hunger > 10:
        messages = ["Awww yea, that was tasty", "/me munches rudely", "Do you have any more?"]
    elif botsnack.hunger > 1:
        messages = ["/me noms furiously", "I really needed that!", "I'll take another!"]
    else:
        messages = ["I'M STARVING. GIVE ME MORE!", "/me gnaws ravenously on the snack with a starved look"]

    msg = random.choice(messages)
    if msg.startswith("/me "):
        phenny.do(msg.partition("/me ")[2])
    else:
        phenny.say(msg)

botsnack.commands = ['botsnack']
botsnack.priority = 'low'
botsnack.hunger = 50.0
botsnack.last_tick = time.time()
botsnack.coolingdown = False

if __name__ == '__main__':
    print(__doc__.strip())