Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Finally implement auto_mode.py!

Browse source
This commit is contained in:
jesopo 2018-09-06 17:58:53 +01:00
parent e76c3b3230
commit 25ef1965fd
1 changed files with 61 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

72
modules/auto_mode.py
View file

@ -1,23 +1,73 @@
import Utils
class Module(object):
_name = "AutoMode"
def __init__(self, bot, events, exports):
self.bot = bot
events.on("received.join").hook(self.on_join)
events.on("received.command.addop").hook(self.add_op,
require_mode="o", min_args=1, channel_only=True,
usage="<nickname>", help="Add a user to the automode op list")
events.on("received.command.removeop").hook(self.remove_op,
require_mode="o", min_args=1, channel_only=True,
usage="<nickname>", help="Remove a user from the automode "
"op list")
events.on("received.command.addvoice").hook(self.add_voice,
require_mode="o", min_args=1, channel_only=True,
usage="<nickname>", help="Add a user to the automode voice list")
events.on("received.command.removevoice").hook(self.remove_voice,
require_mode="o", min_args=1, channel_only=True,
usage="<nickname>", help="Remove a user from the automode "
"voice list")
exports.add("channelset", {"setting": "automode",
"help": "Disable/Enable automode",
"validate": Utils.bool_or_none})
events.on("channel").on("mode").hook(self.on_mode)
events.on("received").on("join").hook(self.on_join)
def on_mode(self, event):
if event["channel"].get_setting("automode", False):
remove = event["remove"]
channel = event["channel"]
mode = event["mode"]
args = event["mode_args"]
def on_join(self, event):
if event["channel"].get_setting("automode", False):
pass
modes = event["channel"].get_user_setting(event["user"].get_id(),
"automodes", [])
if modes:
event["channel"].send_mode("+%s" % "".join(modes),
" ".join([event["user"].nickname for mode in modes]))
def _add_mode(self, event, mode, mode_name):
target_user = event["server"].get_user(event["args_split"][0])
automodes = event["target"].get_user_setting(target_user.get_id(),
"automodes", [])
if mode in automodes:
event["stderr"].write("'%s' already has automode %s" % (
target_user.nickname, mode_name))
else:
automodes.append("o")
event["target"].set_user_setting(target_user.get_id(), "automodes",
automodes)
event["stdout"].write("Added automode %s for '%s'" % (
mode_name, target_user.nickname))
def _remove_mode(self, event, mode, mode_name):
target_user = event["server"].get_user(event["args_split"][0])
automodes = event["target"].get_user_setting(target_user.get_id(),
"automodes", [])
if not mode in automodes:
event["stderr"].write("'%s' doesn't have automode %s" % (
target_user.nickname, mode_name))
else:
automodes.remove("o")
event["target"].set_user_setting(target_user.get_id(), "automodes",
automodes)
event["stdout"].write("Removed automode %s from '%s'" % (
mode_name, target_user.nickname))
def add_op(self, event):
self._add_mode(event, "o", "op")
def remove_op(self, event):
self._remove_mode(event, "o", "op")
def add_voice(self, event):
self._add_mode(event, "v", "voice")
def remove_voice(self, event):
self._remove_mode(event, "v", "voice")