bitbot-3.11-fork/modules/ducks.py

from operator import itemgetter
from threading import Timer
import datetime
import random
import time

import IRCLogging

DUCK_LAST_SEEN = datetime.datetime.now()


class Module(object):
    def __init__(self, bot, events):
        self.log = IRCLogging.Log
        self.bot = bot
        self.events = events
        self.active_duck = 0
        self.decoy_hooked = 0
        self.duck_timer = None

        events.on("received.command.bef").hook(self.duck_bef,
                                               help="Befriend a duck!")
        events.on("received.command.bang").hook(self.duck_bang,
                                                help="Shoot a duck! Meanie.")
        events.on("received.command.decoy").hook(self.set_decoy,
                                                 help="Be a sneaky fellow and make a decoy duck.")

        events.on("received.command.friends").hook(self.duck_friends,
                                                   help="See who the friendliest people to ducks are!")
        events.on("received.command.killers").hook(self.duck_enemies,
                                                     help="See who shoots the most amount of ducks.")
        # events.on("received.command.ducks").hook(self.duck_list,
        #                                              help="Shows a list of the most popular duck superstars.")

        now = datetime.datetime.now()
        next_duck_time = self.get_random_duck_time()

        self.duck_timer = None
        self.duck_times = {}
        self.decoys = {}

        tricky = next_duck_time - now.second
        tricky += ((next_duck_time - (now.minute + 1)) * 2)

        events.on("postboot").on("configure").on(
            "channelset").assure_call(setting="ducks-enabled",
                                      help="Toggles ducks! (1 or 0)")

        events.on("postboot").on("configure").on(
            "channelset").assure_call(setting="min-duck-time",
                                      help="Minimum seconds before a duck is summoned")

        events.on("postboot").on("configure").on(
            "channelset").assure_call(setting="max-duck-time",
                                      help="Max seconds before a duck is summoned")

        events.on("received.numeric.366").hook(self.bootstrap)

        events.on("raw").on("376").hook(self.duck_loop_entry)

    def duck_loop_entry(self, event):
        wait = self.get_random_duck_time()
        self.timer = Timer(wait, self.show_duck, [event])
        print(" Sending out animal in %s seconds" % wait)
        self.timer.start()

    def bootstrap(self, event):
        for server in self.bot.servers.values():
            for channel in server.channels.values():
                ducks_enabled = channel.get_setting("ducks-enabled", 0)
                ducks_enabled = int(ducks_enabled) if isinstance(ducks_enabled,
                                                                 str) else ducks_enabled

                min_time = "min-duck-time-%s" % channel.name
                max_time = "max-duck-time-%s" % channel.name

                min_duck_time = channel.get_setting("min-duck-time", 20)
                max_duck_time = channel.get_setting("max-duck-time", 30)

                min_duck_time = int(min_duck_time) if isinstance(min_duck_time,
                                                                 str) else min_duck_time
                max_duck_time = int(max_duck_time) if isinstance(max_duck_time,
                                                                 str) else max_duck_time

                self.duck_times[min_time] = min_duck_time
                self.duck_times[max_time] = max_duck_time

                if ducks_enabled == 1:
                    channel.set_setting("active-duck", 0)

    def duck_time(self, channel):
        if isinstance(channel, str):
            channel_name = channel
        else:
            channel_name = channel["target"].name

        min = "min-duck-time-%s" % (channel_name)
        max = "max-duck-time-%s" % (channel_name)

        self.bot.log.debug("Attempting to set %s to %s",
                           [str(min), str(self.duck_times[min])]);
        self.bot.log.debug("Attempting to set %s to %s",
                           [str(max), str(self.duck_times[max])]);

        return random.randint(self.duck_times[min], self.duck_times[max])

    def decoy_time(self):
        return random.randint(60, 180)

    def duck_enemies(self, event):
        the_enemy = event["server"].find_all_user_channel_settings(
            "ducks-shot")

        notorious = {}
        enemy_nicks = []
        enemy_ducks = []

        for i in the_enemy:
            if i[1] in notorious.keys():
                notorious[i[1]] += i[2]
            else:
                notorious[i[1]] = i[2]

        for user, enemies in sorted(notorious.items(), key=itemgetter(1),
                                    reverse=True):
            enemy_nicks.append(user)
            enemy_ducks.append(enemies)

        sentence = "Most Notorious Users -- "

        length = len(enemy_nicks) if len(enemy_nicks) < 11 else 11

        for i in range(0, length):
            sentence += enemy_nicks[i] + " (" + str(enemy_ducks[i]) + ")"
            if i < 10:
                sentence += ", "

        sentence = sentence[0:-2]

        event["stdout"].write(sentence)

    def duck_friends(self, event):
        friends = event["server"].find_all_user_channel_settings(
            "ducks-befriended")

        friendliest = {}
        friend_nicks = []
        friend_ducks = []

        for i in friends:
            if i[1] in friendliest.keys():
                friendliest[i[1]] += i[2]
            else:
                friendliest[i[1]] = i[2]

        for user, friends in sorted(friendliest.items(), key = itemgetter(1),
                                    reverse = True):
            friend_nicks.append(user)
            friend_ducks.append(friends)

        sentence = "Friendliest Users -- "

        length = len(friend_nicks) if len(friend_nicks) < 11 else 11

        for i in range(0, length):
            sentence += friend_nicks[i] + " (" + str(friend_ducks[i]) + ")"
            if i < 10:
                sentence += ", "

        sentence = sentence[0:-2]

        event["stdout"].write(sentence)

    def duck_bef(self, event):
        user = event["user"]
        target = user.nickname
        id = user.id

        active_duck = event["target"].get_setting("active-duck", 0)
        active_duck = int(active_duck) if isinstance(active_duck,
                                                     str) else active_duck

        if active_duck == 0:
            if event["server"].has_user(target):
                if not event["server"].is_own_nickname(target):
                    event["target"].send_kick(target,
                                              "You tried befriending a non-existent duck. Creepy!")
                else:
                    event["stderr"].write("Nope.")
            else:
                event["stderr"].write("That user is not in this channel")
        else:
            befriended_ducks = event["target"].get_user_setting(id,
                                                                "ducks-befriended",
                                                                0)
            event["target"].set_user_setting(id,
                                             "ducks-befriended",
                                             befriended_ducks + 1)
            event["target"].set_setting("active-duck", 0)

            grammar = "" if befriended_ducks == 0 else "s"

            event["stdout"].write(
                target + ", you've befriended " + str(
                    befriended_ducks + 1) + " duck" + grammar + " in " + event[
                    "target"].name)

            self.duck_loop_entry(event)

    def duck_bang(self, event):
        user = event["user"]
        target = user.nickname
        id = user.id
        if event["target"].get_setting("active-duck", 0) == 0:
            event["stderr"].set_prefix("Kick")
            if event["server"].has_user(target):
                if not event["server"].is_own_nickname(target):
                    event["target"].send_kick(target,
                                              "You tried shooting a non-existent duck. Creepy!")
                else:
                    event["stderr"].write("Nope.")
            else:
                event["stderr"].write("That user is not in this channel")
        else:
            shot_ducks = event["target"].get_user_setting(id, "ducks-shot", 0)
            event["target"].set_user_setting(id, "ducks-shot", shot_ducks + 1)
            event["target"].set_setting("active-duck", 0)

            grammar = "" if shot_ducks == 0 else "s"

            event["stdout"].write(
                target + ", you've shot " + str(
                    shot_ducks + 1) + " duck" + grammar + " in " + event[
                    "target"].name)

            self.duck_loop_entry(event)

    def get_random_duck_time(self):
        return random.randint(120, 1200)

    def show_duck(self, event):
        for server in self.bot.servers.values():
            for channel in server.channels.values():
                ducks_enabled = channel.get_setting("ducks-enabled", 0)
                ducks_enabled = int(ducks_enabled) if isinstance(ducks_enabled,
                                                                 str) else ducks_enabled
                if ducks_enabled == 0:
                    continue

                self.bot.log.info("Ducks enabled for %s: %s",
                                  [str(channel.name), str(ducks_enabled)])
                active_duck = channel.get_setting("active-duck", 0)
                active_duck = int(active_duck) if isinstance(active_duck,
                                                             str) else active_duck

                if ducks_enabled == 1 and active_duck == 0:
                    ducks = [
                        "・゜゜・。。・゜ ​ ゜\_O​< q​uack!",
                        "・゜゜・。。・゜ ​ ゜\_o< QUACK!",
                        "・゜゜・。 ​ 。・゜゜\​_ó< qu​ack!",
                        "・゜゜・。 ​ 。・゜゜\​_ó< qu​ack quack!",
                        "・゜゜ 。 ​ 。・゜  \​_ó< bawk!",
                        "・゜゜ 。 ​ 。・゜゜\​_ó< squawk!",
                        "・ ゜・。 ​ 。・゜゜ \​_ó< beep beep!"
                    ]

                    channel.send_message(random.choice(ducks))

                    channel.set_setting("active-duck", 1)

                elif ducks_enabled == 1 and active_duck == 1:
                    pass

                else:
                    channel.set_setting("active-duck", 0)

    def duck_decoy(self, event):
        ducks = [
            "・゜゜・。。・゜ ​ ゜\_O​< q​uack!",
            "・゜゜・。。・゜ ​ ゜\_o< QUACK!",
            "・゜゜・。 ​ 。・゜゜\​_ó< qu​ack!",
            "・゜゜・。 ​ 。・゜゜\​_ó< qu​ack quack!",
            "・゜゜ 。 ​ 。・゜  \​_ó< bawk!",
            "・゜゜ 。 ​ 。・゜゜\​_ó< squawk!",
            "・ ゜・。 ​ 。・゜゜ \​_ó< beep beep!"
        ]

        event["stdout"].write(random.choice(ducks))

    def set_decoy(self, event):
        channel = event["target"]

        next_decoy_time = self.decoy_time()

        if self.decoy_hooked == 0:
            self.events.on("timer").on("duck-decoy").hook(self.duck_decoy)
            self.decoy_hooked = 1

        self.bot.add_timer("duck-decoy", next_decoy_time, None, None, False,
                           channel=channel)

# def coins(self, event):
#    if event["args_split"]:
#        target = event["server"].get_user(event["args_split"][0])
#    else:
#        target = event["user"]
#    coins = decimal.Decimal(target.get_setting("coins", "0.0"))
#    event["stdout"].write("%s has %s coin%s" % (target.nickname,
#        "{0:.2f}".format(coins), "" if coins == 1 else "s"))