diff --git a/Database.py b/Database.py index 1c85f8ef..c315d801 100644 --- a/Database.py +++ b/Database.py @@ -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)""") \ No newline at end of file diff --git a/modules/ducks.py b/modules/ducks.py new file mode 100644 index 00000000..685a49f9 --- /dev/null +++ b/modules/ducks.py @@ -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​< 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) + + 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​< q​uack!", + "・゜゜・。。・゜ ​ ゜\_o< QUACK!", + "・゜゜・。 ​ 。・゜゜\​_ó< qu​ack!", + "・゜゜・。 ​ 。・゜゜\​_ó< qu​ack 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")) diff --git a/modules/google.py b/modules/google.py index b961186e..b8d5c653 100644 --- a/modules/google.py +++ b/modules/google.py @@ -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][ diff --git a/modules/strax.py b/modules/strax.py index b2496839..144d729a 100644 --- a/modules/strax.py +++ b/modules/strax.py @@ -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)