forked from aewens/babili-bot
115 lines
3.1 KiB
Python
Executable File
115 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from argparse import ArgumentParser
|
|
from os.path import dirname, realpath
|
|
|
|
from bot import Bot, Tasks, Responses
|
|
from actions import actions
|
|
from coroutines import coroutines
|
|
|
|
parser = ArgumentParser(description="A meta bot for ~team")
|
|
parser.add_argument(
|
|
"-c",
|
|
"--config",
|
|
dest="config",
|
|
default="settings.debug.json",
|
|
help="Load config file"
|
|
)
|
|
arguments = parser.parse_args()
|
|
|
|
bot = Bot("127.0.0.1", 6667)
|
|
responses = Responses(bot)
|
|
tasks = Tasks(bot)
|
|
|
|
# for coro in coroutines:
|
|
# worker = coro["worker"]
|
|
# interval = coro["interval"]
|
|
# state = coro.get("state", None)
|
|
# coro_state = state if state is not None else (bot,)
|
|
# tasks.add_coroutine(worker, interval, coro_state)
|
|
tasks.coroutines = coroutines
|
|
|
|
for action in actions:
|
|
if "type" in action and "pattern" in action and "callback" in action:
|
|
responses.add_trigger(
|
|
action["type"],
|
|
action["pattern"],
|
|
action["callback"]
|
|
)
|
|
|
|
def try_to_king_me(channel):
|
|
bot.send_message("ChanServ", "REGISTER {}", channel)
|
|
bot.send_message("ChanServ", "SET Successor {} {}", channel, bot.botnick)
|
|
bot.send_message("ChanServ", "SET Founder {} {}", channel, bot.author)
|
|
|
|
def handle_pm(name, response):
|
|
print("PM: {} - {}".format(name, response))
|
|
|
|
def handle_mode(channel, mode):
|
|
if mode == "-r":
|
|
try_to_king_me(channel)
|
|
|
|
def handle_invite(channel, name):
|
|
changed = False
|
|
kingme = bot.settings.get("extras", dict()).get("kingme", [])
|
|
|
|
if channel in kingme:
|
|
try_to_king_me(channel)
|
|
|
|
users = bot.memories["users"]
|
|
if name not in users:
|
|
bot.memories["users"][name] = dict()
|
|
changed = True
|
|
|
|
if "invites" not in users[name]:
|
|
bot.memories["users"][name]["invites"] = list()
|
|
changed = True
|
|
|
|
if channel not in bot.memories["users"][name]["invites"]:
|
|
bot.memories["users"][name]["invites"].append(channel)
|
|
changed = True
|
|
|
|
if changed:
|
|
bot.thread(bot.save_memories)
|
|
|
|
def handle_kick(name):
|
|
users = bot.memories["users"]
|
|
if name not in users:
|
|
bot.memories["users"][name] = dict()
|
|
|
|
bot.memories["users"][name]["kicker"] = True
|
|
bot.thread(bot.save_memories)
|
|
|
|
def handle_message(name, source, response):
|
|
responses.parse(name, source, response)
|
|
if response == "!debug":
|
|
bot.logger.debug(":: {}".format(bot.memories))
|
|
|
|
def handle_crashed():
|
|
bot.logger.debug("Rebooting")
|
|
bot.crashed = True
|
|
bot.tasks.stop()
|
|
|
|
tasks = Tasks(bot)
|
|
tasks.coroutines = coroutines
|
|
|
|
bot.tasks = tasks
|
|
bot.start(arguments.config, dirname(realpath(__file__)), {
|
|
"pm": handle_pm,
|
|
"mode": handle_mode,
|
|
"invite": handle_invite,
|
|
"kick": handle_kick,
|
|
"crashed": handle_crashed,
|
|
"message": handle_message
|
|
})
|
|
|
|
if __name__ == "__main__":
|
|
bot.tasks = tasks
|
|
bot.start(arguments.config, dirname(realpath(__file__)), {
|
|
"pm": handle_pm,
|
|
"mode": handle_mode,
|
|
"invite": handle_invite,
|
|
"kick": handle_kick,
|
|
"crashed": handle_crashed,
|
|
"message": handle_message
|
|
}) |