from operator import itemgetter from threading import Timer import Utils import random class Module(object): def __init__(self, bot, events): self.bot = bot self.events = events self.decoy_hooked = 0 self.duck_timer = None self.duck_list = [ "・゜゜・。。・゜ ​ ゜\_O​< q​uack!", "・゜゜・。。・゜ ​ ゜\_o< QUACK!", "・゜゜・。 ​ 。・゜゜\​_ó< qu​ack!", "・゜゜・。 ​ 。・゜゜\​_ó< qu​ack quack!", "・゜゜ 。 ​ 。・゜ \​_ó< bawk!", "・゜゜ 。 ​ 。・゜゜\​_ó< squawk!", "・ ゜・。 ​ 。・゜゜ \​_ó< beep beep!" ] 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.") self.duck_times = {} 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]) self.bot.log.info("Sending out a wave of ducks 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 " + Utils.bold(str( befriended_ducks + 1)) + " duck" + grammar + " in " + Utils.bold(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 " + Utils.bold(str(shot_ducks + 1)) + " duck" + grammar + " in " + Utils.bold(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: channel.send_message(random.choice(self.duck_list)) 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): event["stdout"].write(random.choice(self.duck_list)) 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"))