286 lines
11 KiB
Python
286 lines
11 KiB
Python
from operator import itemgetter
|
||
from threading import Timer
|
||
import Utils
|
||
import random
|
||
|
||
DUCK_LIST = [
|
||
"・゜゜・。。・゜ ゜\_O< quack!",
|
||
"・゜゜・。。・゜ ゜\_o< QUACK!",
|
||
"・゜゜・。 。・゜゜\_ó< quack!",
|
||
"・゜゜・。 。・゜゜\_ó< quack quack!",
|
||
"・゜゜ 。 。・゜ \_ó< bawk!",
|
||
"・゜゜ 。 。・゜゜\_ó< squawk!",
|
||
"・ ゜・。 。・゜゜ \_ó< beep beep!"
|
||
]
|
||
|
||
class Module(object):
|
||
def __init__(self, bot, events):
|
||
self.bot = bot
|
||
self.events = events
|
||
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.")
|
||
|
||
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(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(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"))
|