jesopo
GitHub
commit
5453b33d6e
2 changed files with 293 additions and 189 deletions
|
|
@ -1,4 +1,5 @@
|
|||
import random
|
||||
import Utils
|
||||
|
||||
CHOICES = [
|
||||
"Definitely",
|
||||
|
|
@ -13,7 +14,18 @@ CHOICES = [
|
|||
"The answer is unclear",
|
||||
"Absolutely",
|
||||
"Dubious at best",
|
||||
"I'm on a break, ask again later"
|
||||
"I'm on a break, ask again later",
|
||||
"As I see it, yes",
|
||||
"It is certain",
|
||||
"Naturally",
|
||||
"Reply hazy, try again later",
|
||||
Utils.color(4) + Utils.underline("DO NOT WASTE MY TIME"),
|
||||
"Hmm... Could be!",
|
||||
"I'm leaning towards no",
|
||||
"Without a doubt",
|
||||
"Sources say no",
|
||||
"Sources say yes",
|
||||
"Sources say maybe"
|
||||
]
|
||||
|
||||
class Module(object):
|
||||
|
|
@ -26,4 +38,5 @@ class Module(object):
|
|||
)
|
||||
|
||||
def decide(selfs, event):
|
||||
event["stdout"].write(random.choice(CHOICES))
|
||||
event["stdout"].write("You shake the magic ball... it "
|
||||
"says " + Utils.bold(random.choice(CHOICES)))
|
||||
|
|
|
|||
459
modules/ducks.py
459
modules/ducks.py
|
|
@ -1,65 +1,297 @@
|
|||
from operator import itemgetter
|
||||
from threading import Timer
|
||||
import Utils
|
||||
import random
|
||||
from operator import itemgetter
|
||||
from time import time
|
||||
|
||||
DUCK_LIST = [
|
||||
"・゜゜・。。・゜ ゜\_O< quack!",
|
||||
"・゜゜・。。・゜ ゜\_o< QUACK!",
|
||||
"・゜゜・。 。・゜゜\_ó< quack!",
|
||||
"・゜゜・。 。・゜゜\_ó< quack quack!",
|
||||
"・゜゜ 。 。・゜ \_ó< bawk!",
|
||||
"・゜゜ 。 。・゜゜\_ó< squawk!",
|
||||
"・ ゜・。 。・゜゜ \_ó< beep beep!"
|
||||
]
|
||||
import Utils
|
||||
|
||||
DUCK_TAIL = "・゜゜・。。・゜゜"
|
||||
DUCK_HEAD = ["\_o< ", "\_O< ", "\_0< ", "\_\u00f6< ", "\_\u00f8< ",
|
||||
"\_\u00f3< "]
|
||||
DUCK_MESSAGE = ["QUACK!", "FLAP FLAP!", "quack!", "squawk!"]
|
||||
DUCK_MESSAGE_RARE = ["beep boop!", "QUACK QUACK QUACK QUACK QUACK!!", "HONK!",
|
||||
Utils.underline("I AM THE METAL DUCK")]
|
||||
|
||||
DUCK_MINIMUM_MESSAGES = 10
|
||||
DUCK_MINIMUM_UNIQUE = 3
|
||||
|
||||
class Module(object):
|
||||
|
||||
def __init__(self, bot, events, exports):
|
||||
self.bot = bot
|
||||
self.events = events
|
||||
|
||||
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,
|
||||
events.on("received.command.bef").hook(self.befriend,
|
||||
priority=1,
|
||||
help="Befriend a "
|
||||
"duck!")
|
||||
events.on("received.command.bang").hook(self.shoot,
|
||||
priority=1,
|
||||
help="Shoot a "
|
||||
"duck! "
|
||||
"Meanie.", )
|
||||
# events.on("received").on("command").on("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.duckstats").hook(self.duck_stats,
|
||||
help="Shows your duck "
|
||||
"stats!")
|
||||
events.on("received.command.killers").hook(
|
||||
self.duck_enemies, help="See who shoots the most smount of ducks!")
|
||||
events.on("received.command.duckstats").hook(
|
||||
self.duck_stats, help="Shows your duck stats!")
|
||||
|
||||
exports.add("channelset",
|
||||
{"setting": "ducks-enabled", "help": "Toggle ducks!",
|
||||
"validate": Utils.bool_or_none})
|
||||
|
||||
exports.add("channelset", {"setting": "ducks-kick",
|
||||
"help": "Should the bot kick if there's no "
|
||||
"duck?",
|
||||
"validate": Utils.bool_or_none})
|
||||
|
||||
events.on("new.channel").hook(self.bootstrap)
|
||||
|
||||
|
||||
exports.add("channelset", {"setting": "ducks-enabled",
|
||||
"help": "Toggle ducks!", "validate": Utils.bool_or_none})
|
||||
events.on("received").on("message").on("channel").hook(
|
||||
self.channel_message, priority=2)
|
||||
|
||||
events.on("received.numeric.366").hook(self.bootstrap)
|
||||
self.bootstrap_lazy(events)
|
||||
|
||||
events.on("raw").on("376").hook(self.duck_loop_entry)
|
||||
def bootstrap_lazy(self, event):
|
||||
servers = self.bot.servers
|
||||
|
||||
events.on("timer").on("duck-decoy").hook(self.duck_decoy)
|
||||
events.on("timer").on("show-duck").hook(self.show_duck)
|
||||
|
||||
def duck_loop_entry(self, event):
|
||||
wait = self.get_random_duck_time()
|
||||
self.bot.log.info("Sending out a wave of ducks in %s seconds",
|
||||
[wait])
|
||||
self.bot.add_timer("show-duck", wait, persist=False)
|
||||
for server in servers.values():
|
||||
for channel in server.channels.values():
|
||||
self.bootstrap(channel)
|
||||
|
||||
def bootstrap(self, event):
|
||||
for server in self.bot.servers.values():
|
||||
for channel in server.channels.values():
|
||||
channel = event["channel"] if hasattr(event, "channel") else event
|
||||
|
||||
self.init_game_var(channel)
|
||||
# getset
|
||||
ducks_enabled = channel.get_setting("ducks-enabled", False)
|
||||
|
||||
if ducks_enabled == True:
|
||||
channel.set_setting("active-duck", False)
|
||||
self.start_game(channel)
|
||||
|
||||
def is_duck_channel(self, channel):
|
||||
if channel.get_setting("ducks-enabled", False) == False:
|
||||
return False
|
||||
|
||||
if hasattr(channel, 'games') == False:
|
||||
return False
|
||||
|
||||
if "ducks" not in channel.games.keys():
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
def init_game_var(self, channel):
|
||||
if hasattr(channel, 'games') == False:
|
||||
channel.games = {}
|
||||
|
||||
def clear_ducks(self, channel):
|
||||
rand_time = self.generate_next_duck_time()
|
||||
|
||||
channel.games["ducks"] = {'messages': 0, 'duck_spawned': 0,
|
||||
'unique_users': [],
|
||||
'next_duck_time': rand_time}
|
||||
|
||||
def start_game(self, channel):
|
||||
# event is immediately the IRCChannel.Channel() event for the current
|
||||
# channel
|
||||
self.clear_ducks(channel)
|
||||
|
||||
def generate_next_duck_time(self):
|
||||
rand_time = random.randint(int(time()) + 360, int(time()) + 1200)
|
||||
return rand_time
|
||||
|
||||
def is_duck_visible(self, event):
|
||||
channel = event["target"]
|
||||
|
||||
visible = bool(channel.games["ducks"]["duck_spawned"])
|
||||
return visible
|
||||
|
||||
def should_kick(self, event):
|
||||
channel = event["target"]
|
||||
return channel.get_setting("ducks-kick", False)
|
||||
|
||||
def kick_bef(self, event):
|
||||
channel = event["target"]
|
||||
target = event["user"].nickname
|
||||
|
||||
channel.send_kick(target,
|
||||
"You tried befriending a non-existent duck. Creepy!")
|
||||
|
||||
def kick_bang(self, event):
|
||||
channel = event["target"]
|
||||
target = event["user"].nickname
|
||||
|
||||
channel.send_kick(target,
|
||||
"You tried shooting a non-existent duck. Creepy!")
|
||||
|
||||
def should_generate_duck(self, event):
|
||||
channel = event["channel"]
|
||||
game = channel.games["ducks"]
|
||||
|
||||
spawned = game["duck_spawned"]
|
||||
unique = len(game["unique_users"])
|
||||
messages = game["messages"]
|
||||
next_duck = game["next_duck_time"]
|
||||
|
||||
# DUCK_MINIMUM_MESSAGES = 10
|
||||
# DUCK_MINIMUM_UNIQUE = 3
|
||||
|
||||
if spawned == 0 and next_duck < time() and unique >= \
|
||||
DUCK_MINIMUM_UNIQUE and messages >= DUCK_MINIMUM_MESSAGES:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
def show_duck(self, event):
|
||||
channel = event["channel"]
|
||||
duck = ""
|
||||
|
||||
if channel.games["ducks"]["duck_spawned"] == 1:
|
||||
return
|
||||
|
||||
duck += DUCK_TAIL
|
||||
duck += random.choice(DUCK_HEAD)
|
||||
|
||||
duck = str(Utils.color(4) + Utils.bold(
|
||||
duck + random.choice(DUCK_MESSAGE_RARE)) + Utils.color(
|
||||
4)) if 1 == random.randint(1, 20) else duck + random.choice(
|
||||
DUCK_MESSAGE)
|
||||
|
||||
channel.send_message(duck)
|
||||
channel.games["ducks"]["duck_spawned"] = 1
|
||||
|
||||
def channel_message(self, event):
|
||||
channel = event["channel"]
|
||||
|
||||
if "ducks" not in channel.games.keys():
|
||||
return
|
||||
|
||||
user = event["user"]
|
||||
game = channel.games["ducks"]
|
||||
|
||||
if game["duck_spawned"] == 1 or channel.has_user(
|
||||
event["user"]) == False:
|
||||
return
|
||||
|
||||
unique = game["unique_users"]
|
||||
messages = game["messages"]
|
||||
|
||||
if user not in unique:
|
||||
game["unique_users"].append(user)
|
||||
messages_increment = 1
|
||||
else:
|
||||
messages_increment = 0.5
|
||||
|
||||
game["messages"] = messages + messages_increment
|
||||
|
||||
if self.should_generate_duck(event) == True:
|
||||
self.show_duck(event)
|
||||
|
||||
def befriend(self, event):
|
||||
channel = event["target"]
|
||||
user = event["user"]
|
||||
nick = user.nickname
|
||||
uid = user.get_id()
|
||||
if self.is_duck_channel(channel) == False:
|
||||
return
|
||||
|
||||
if self.is_duck_visible(event) == False:
|
||||
if self.should_kick(event):
|
||||
self.kick_bef(event)
|
||||
return
|
||||
|
||||
channel.games["ducks"][
|
||||
"next_duck_time"] = self.generate_next_duck_time()
|
||||
channel.games["ducks"]["duck_spawned"] = 0
|
||||
|
||||
total_befriended = channel.get_user_setting(uid, "ducks-befriended", 0)
|
||||
total_befriended = total_befriended + 1
|
||||
|
||||
channel.set_user_setting(uid, "ducks-befriended", total_befriended)
|
||||
|
||||
event["stdout"].write(
|
||||
"Aww! " + nick + " befriended a duck! You've befriended " +
|
||||
Utils.bold(
|
||||
str(total_befriended)) + " ducks in " + Utils.bold(
|
||||
channel.name) + "!")
|
||||
|
||||
self.clear_ducks(channel)
|
||||
|
||||
def shoot(self, event):
|
||||
channel = event["target"]
|
||||
user = event["user"]
|
||||
nick = user.nickname
|
||||
uid = user.get_id()
|
||||
|
||||
if self.is_duck_channel(channel) == False:
|
||||
return
|
||||
|
||||
if self.is_duck_visible(event) == False:
|
||||
if self.should_kick(event):
|
||||
self.kick_bang(event)
|
||||
return
|
||||
|
||||
channel.games["ducks"][
|
||||
"next_duck_time"] = self.generate_next_duck_time()
|
||||
channel.games["ducks"]["duck_spawned"] = 0
|
||||
|
||||
total_shot = channel.get_user_setting(uid, "ducks-shot", 0)
|
||||
total_shot = total_shot + 1
|
||||
|
||||
channel.set_user_setting(uid, "ducks-shot", total_shot)
|
||||
|
||||
event["stdout"].write(
|
||||
"Pow! " + nick + " shot a duck! You've shot " + Utils.bold(
|
||||
str(total_shot)) + " ducks in " + Utils.bold(
|
||||
channel.name) + "!")
|
||||
|
||||
self.clear_ducks(channel)
|
||||
|
||||
def duck_stats(self, event):
|
||||
user = event["user"]
|
||||
channel = event["target"].name
|
||||
nick = user.nickname
|
||||
id = user.get_id()
|
||||
|
||||
poached = user.get_channel_settings_per_setting("ducks-shot", [])
|
||||
friends = user.get_channel_settings_per_setting("ducks-befriended", [])
|
||||
|
||||
channel_friends = 0
|
||||
channel_poached = 0
|
||||
|
||||
total_friends = 0
|
||||
total_poached = 0
|
||||
|
||||
for room, number in friends:
|
||||
if room == channel:
|
||||
channel_friends = number
|
||||
total_friends += number
|
||||
else:
|
||||
total_friends += number
|
||||
|
||||
for room, number in poached:
|
||||
if room == channel:
|
||||
channel_poached = number
|
||||
total_poached += number
|
||||
else:
|
||||
total_poached += number
|
||||
|
||||
event["stdout"].write(
|
||||
nick + ": " + str(total_poached) + " ducks killed (" + str(
|
||||
channel_poached) + " in " + channel + "), and " + str(
|
||||
total_friends) + " ducks befriended (" + str(
|
||||
channel_friends) + " in " + channel + ")")
|
||||
|
||||
def duck_enemies(self, event):
|
||||
the_enemy = event["server"].find_all_user_channel_settings(
|
||||
"ducks-shot")
|
||||
the_enemy = event["server"].find_all_user_channel_settings("ducks-shot")
|
||||
|
||||
notorious = {}
|
||||
enemy_nicks = []
|
||||
|
|
@ -103,8 +335,8 @@ class Module(object):
|
|||
else:
|
||||
friendliest[i[1]] = i[2]
|
||||
|
||||
for user, friends in sorted(friendliest.items(), key = itemgetter(1),
|
||||
reverse = True):
|
||||
for user, friends in sorted(friendliest.items(), key=itemgetter(1),
|
||||
reverse=True):
|
||||
friend_nicks.append(user)
|
||||
friend_ducks.append(friends)
|
||||
|
||||
|
|
@ -120,144 +352,3 @@ class Module(object):
|
|||
sentence = sentence[0:-2]
|
||||
|
||||
event["stdout"].write(sentence)
|
||||
|
||||
def duck_bef(self, event):
|
||||
user = event["user"]
|
||||
target = user.nickname
|
||||
id = user.get_id()
|
||||
|
||||
active_duck = event["target"].get_setting("active-duck", False)
|
||||
|
||||
if active_duck == False:
|
||||
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", False)
|
||||
|
||||
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.get_id()
|
||||
if not event["target"].get_setting("active-duck", False):
|
||||
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", False)
|
||||
|
||||
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(720, 3600)
|
||||
|
||||
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", False)
|
||||
|
||||
if ducks_enabled == False:
|
||||
continue
|
||||
|
||||
active_duck = channel.get_setting("active-duck", False)
|
||||
|
||||
if ducks_enabled == True and active_duck == False:
|
||||
channel.send_message(random.choice(DUCK_LIST))
|
||||
|
||||
channel.set_setting("active-duck", True)
|
||||
|
||||
elif ducks_enabled == True and active_duck == True:
|
||||
pass
|
||||
|
||||
else:
|
||||
channel.set_setting("active-duck", False)
|
||||
|
||||
def duck_decoy(self, event):
|
||||
self.events.on("send").on("stdout").call(target=event["channel"],
|
||||
module_name="Ducks", server=event["server"],
|
||||
message=random.choice(DUCK_LIST))
|
||||
|
||||
def set_decoy(self, event):
|
||||
next_decoy_time = self.get_random_duck_time()
|
||||
self.bot.add_timer("duck-decoy", next_decoy_time, persist=False,
|
||||
server=event["server"], channel=event["target"])
|
||||
|
||||
|
||||
def duck_stats(self, event):
|
||||
user = event["user"]
|
||||
channel = event["target"].name
|
||||
nick = user.nickname
|
||||
id = user.get_id()
|
||||
|
||||
poached = user.get_channel_settings_per_setting("ducks-shot", []
|
||||
)
|
||||
friends = user.get_channel_settings_per_setting(
|
||||
"ducks-befriended", []
|
||||
)
|
||||
|
||||
channel_friends = 0
|
||||
channel_poached = 0
|
||||
|
||||
total_friends = 0
|
||||
total_poached = 0
|
||||
|
||||
for room, number in friends:
|
||||
if room == channel:
|
||||
channel_friends = number
|
||||
total_friends += number
|
||||
else:
|
||||
total_friends += number
|
||||
|
||||
for room, number in poached:
|
||||
if room == channel:
|
||||
channel_poached = number
|
||||
total_poached += number
|
||||
else:
|
||||
total_poached += number
|
||||
|
||||
event["stdout"].write(
|
||||
nick + ": " + str(total_poached) + " ducks killed (" + str(
|
||||
channel_poached) + " in " + channel + "), and " + str(
|
||||
total_friends) + " ducks befriended (" + str(channel_friends)
|
||||
+ " in " + channel + ")")
|
||||
|
||||
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue