Merge pull request #7 from dngfx/master

I've been busy!
This commit is contained in:
jesopo 2018-08-31 12:08:55 +01:00 committed by GitHub
commit 9cd4e86750
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 233 additions and 16 deletions

View file

@ -354,4 +354,4 @@ class Database(object):
DELETE CASCADE, PRIMARY KEY (user_id, channel_id,
setting))""")
self.execute("""CREATE INDEX user_channel_settings_index
ON user_channel_settings (user_id, channel_id, setting)""")
ON user_channel_settings (user_id, channel_id, setting)""")

211
modules/ducks.py Normal file
View file

@ -0,0 +1,211 @@
import datetime
import random
import IRCLogging
DUCK_LAST_SEEN = datetime.datetime.now()
class Module(object):
def __init__(self, bot):
self.log = IRCLogging.Log
self.bot = bot
self.active_duck = 0
bot.events.on("received.command.bef").hook(self.duck_bef,
help="Befriend a duck!")
bot.events.on("received.command.bang").hook(self.duck_bang,
help="Shoot a duck! Meanie.")
bot.events.on("received.command.decoy").hook(self.set_decoy,
help="Be a sneaky fellow and make a decoy duck.")
# bot.events.on("received.command.friends").hook(self.duck_friends,
# help="See who the friendliest people to ducks are!")
# bot.events.on("received.command.killers").hook(self.duck_killers,
# help="See who shoots the most amount of ducks.")
# bot.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 = random.randint(30, 40)
self.duck_times = {}
self.decoys = {}
tricky = next_duck_time - now.second
tricky += ((next_duck_time - (now.minute + 1)) * 2)
bot.events.on("postboot").on("configure").on(
"channelset").assure_call(setting="ducks-enabled",
help="Toggles ducks! (1 or 0)")
bot.events.on("postboot").on("configure").on(
"channelset").assure_call(setting="min-duck-time",
help="Minimum seconds before a duck is summoned")
bot.events.on("postboot").on("configure").on(
"channelset").assure_call(setting="max-duck-time",
help="Max seconds before a duck is summoned")
bot.events.on("timer").on("duck-appear").hook(self.show_duck)
bot.add_timer("duck-appear", next_duck_time, persist=False)
bot.events.on("received.numeric.366").hook(self.bootstrap)
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", 240)
max_duck_time = channel.get_setting("max-duck-time", 1200)
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)
print(self.duck_times)
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(300, 700)
def duck_bef(self, event):
target = event["user"].nickname
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:
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 befriending a non-existent duck. Creepy!")
else:
event["stderr"].write("Nope.")
else:
event["stderr"].write("That user is not in this channel")
else:
setting = "bef-ducks-%s-%s" % (target, event["target"].name)
befriended_ducks = event["server"].get_setting(setting, 0)
event["server"].set_setting(setting, 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);
next_duck_time = self.duck_time(event)
self.bot.add_timer("duck-appear", next_duck_time, persist=False)
def duck_bang(self, event):
target = event["user"].nickname
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:
setting = "shot-ducks-%s-%s" % (target, event["target"].name)
shot_ducks = event["server"].get_setting(setting, 0)
event["server"].set_setting(setting, 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);
next_duck_time = self.duck_time(event)
self.bot.add_timer("duck-appear", next_duck_time, persist=False)
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< quack!",
"・゜゜・。。・゜ \_o< QUACK!",
"・゜゜・。 。・゜゜\_ó< quack!",
"・゜゜・。 。・゜゜\_ó< quack 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)
next_duck_time = self.duck_time(channel.name)
self.bot.add_timer("duck-appear", next_duck_time, persist=False)
def duck_decoy(self, event):
ducks = [
"・゜゜・。。・゜ \_O< quack!",
"・゜゜・。。・゜ \_o< QUACK!",
"・゜゜・。 。・゜゜\_ó< quack!",
"・゜゜・。 。・゜゜\_ó< quack quack!",
"・゜゜ 。 。・゜ \_ó< bawk!",
"・゜゜ 。 。・゜゜\_ó< squawk!",
"・ ゜・。 。・゜゜ \_ó< beep beep!"
]
event["target"].send_message(random.choice(ducks))
def set_decoy(self, event):
next_decoy_time = self.decoy_time()
self.bot.events.on("timer").on("duck-decoy").hook(self.duck_decoy)
self.bot.add_timer("duck-decoy", next_decoy_time, persist=False)
# 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"))

View file

@ -19,7 +19,7 @@ class Module(object):
"q": phrase, "key": self.bot.config[
"google-api-key"], "cx": self.bot.config[
"google-search-id"], "prettyPrint": "true",
"num": 1, "gl": "us"}, json=True)
"num": 1, "gl": "gb"}, json=True)
if page:
if "items" in page and len(page["items"]):
event["stdout"].write(page["items"][0][

View file

@ -4,35 +4,41 @@ import random
class Module(object):
def __init__(self, bot):
bot.events.on("received.command.strax").hook(
self.strax, help="Glory to the sontaran empire, through IRC!")
self.strax, help="Suggests a glorious method of battle for the glory of the Sontaran Empire, through IRC!")
def strax(self, event):
suggestion_greeting = ["Might I suggest", "I'd suggest"]
command_greeting = ["We should attack now with", "We must attack now with"]
suggestion_greeting = ["Might I suggest", "Can I suggest", "Should we attack immediately with"]
command_greeting = ["We should attack now with", "We must attack now with", "I suggest attacking with",
"We should coordinate an attack with"]
method_of_attack_a = ["full-frontal", "pincer", "surprise", "brutally excessive", "multi-pronged", "glorious",
"violent", "devestating", "superior"]
method_of_attack_an = ["acid-heavy", "immediate", "overwhelming", "unstoppable"]
"violent", "devastating", "superior", "fast-paced", "fleet-wide", "stealth",
"diversionary", "exceptional", "point-blank", "night time"]
method_of_attack_an = ["acid-heavy", "immediate", "overwhelming", "unstoppable", "underground", "arial",
"naval", "amphibious", "full-scale"]
type_of_attack = ["assault", "attack", "bombardment", "offensive", "barrage", "charge", "strike", "operation",
"manoeuvre", "blitzkrieg"]
"manoeuvre", "blitzkrieg", "ambush", "massacre"]
attack_adjective = ["laser", "berserker", "acid", "armoured attack", "proton",
"three kinds of", "atomic", "toxic", "explosive",
"red-hot", "thermal", "automated fire", "cluster",
"enhanced germ", "energy-drink-fueled"]
"enhanced germ", "energy-drink-fueled", "battle ready", "Sontaran", "military"]
attack_object = ["bees", "chainsaws", "marmots", "acid", "monkeys", "mines", "bombs", "snakes", "spiders",
"knives", "rockets", "sharks", "owls", "repurposed cybermats", "cannons", "alligators"]
attack_object_two = ["robots", "ninjas", "grenades", "a dolphin full of napalm", "acid", "dynamite",
"xenomorphs", "lots and lots of C4", "tactical nukes", "MacGyver", "bio-weapons",
"knives", "rockets", "sharks", "owls", "repurposed cybermats", "cannons", "alligators", "ants",
"gorillas", "genetically enhanced cyber-elephants", "mechanoids", "KGB agents",
"MI5 operatives", "thermonuclear missiles"]
attack_object_two = ["robots", "ninjas", "grenades", "a dolphin full of napalm", "dynamite",
"xenomorphs", "lots and lots of C4", "tactical nukes", "bio-weapons",
"rocket launchers", "an elephant", "a memory worm for afterwards", "this pencil"]
method_of_attack = " an " + random.choice(method_of_attack_an) if random.choice([1,
2]) == 1 else " a " + random.choice(
2]) == 1 else " a " + random.choice(
method_of_attack_a)
greeting_choice = random.choice([1,2])
greeting_choice = random.choice([1, 2])
greeting = random.choice(suggestion_greeting) if greeting_choice == 1 else random.choice(command_greeting)
exclamation = "?" if greeting_choice == 1 else "!"
suggestion = greeting + method_of_attack + " " + random.choice(type_of_attack) + " with " + random.choice(attack_adjective) + " " + random.choice(attack_object) + " and " + random.choice(attack_object_two) + exclamation
suggestion = greeting + method_of_attack + " " + random.choice(type_of_attack) + " with " + random.choice(
attack_adjective) + " " + random.choice(attack_object) + " and " + random.choice(
attack_object_two) + exclamation
event["stdout"].write(suggestion)