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

Support kick and ban command in private message (channel_op.py)

Browse source
This commit is contained in:
jesopo 2019-01-30 21:02:12 +00:00
parent cd93118571
commit 8345ea36d4
1 changed files with 31 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
modules/channel_op.py
View file

@ -32,6 +32,26 @@ class Module(ModuleManager.BaseModule):
channel.send_kick(nickname, reason) channel.send_kick(nickname, reason)
else: else:
raise UserNotFoundException("That user is not in this channel") raise UserNotFoundException("That user is not in this channel")
def _kick_command(self, event, channel, args_split):
target = args_split[0]
reason = " ".join(args_split[1:]) or None
try:
self._kick(event["server"], event["target"], target, reason)
except UserNotFoundException:
event["stderr"].write(str(e))
@utils.hook("received.command.kick", private_only=True, min_args=2)
def private_kick(self, event):
"""
:help: Kick a user from the current channel
:usage: <nickname> [reason]
:require_access: kick
:channel_arg: 0
:prefix: Kick
"""
channel = event["server"].channels.get(event["args_split"][0])
self._kick_command(event, channel, event["args_split"][1:])
@utils.hook("received.command.k", alias_of="kick") @utils.hook("received.command.k", alias_of="kick")
@utils.hook("received.command.kick", channel_only=True, min_args=1) @utils.hook("received.command.kick", channel_only=True, min_args=1)
@ -43,13 +63,7 @@ class Module(ModuleManager.BaseModule):
:require_access: kick :require_access: kick
:prefix: Kick :prefix: Kick
""" """
target = event["args_split"][0] self._kick_command(event, event["target"], event["args_split"])
reason = " ".join(event["args_split"][1:]) or None
try:
self._kick(event["server"], event["target"], target, reason)
except UserNotFoundException:
event["stderr"].write(str(e))
def _ban_format(self, user, s): def _ban_format(self, user, s):
return s.replace("$n", user.nickname).replace("$u", user.username return s.replace("$n", user.nickname).replace("$u", user.username
@ -79,6 +93,16 @@ class Module(ModuleManager.BaseModule):
event["target"].send_unban(target) event["target"].send_unban(target)
return target return target
@utils.hook("received.command.ban", private_only=True, min_args=2)
def private_ban(self, event):
"""
:help: Ban a user/hostmask from the current channel
:usage: <channel> <nickname/hostmask>
:require_access: ban
:channel_arg: 0
"""
channel = event["server"].channels.get(event["args_split"][0])
self._ban(event["server"], channel, True, event["args_split"][1])
@utils.hook("received.command.ban", channel_only=True, min_args=1) @utils.hook("received.command.ban", channel_only=True, min_args=1)
def ban(self, event): def ban(self, event):
""" """