bitbot-3.11-fork/modules/kick-rejoin.py

40 lines
1.4 KiB
Python
Raw Normal View History

2019-07-11 20:28:05 +00:00
#--depends-on config
from src import ModuleManager, utils
DELAY = 5
rejoin_setting = utils.BoolSetting("kick-rejoin",
"Whether or not I should rejoin channels I get kicked from")
delay_setting = utils.IntSetting("kick-rejoin-delay",
"Amount of seconds to wait before rejoining a channel")
@utils.export("serverset", rejoin_setting)
@utils.export("serverset", delay_setting)
@utils.export("channelset", rejoin_setting)
@utils.export("channelset", delay_setting)
2019-07-11 20:28:05 +00:00
class Module(ModuleManager.BaseModule):
def _should_rejoin(self, server, channel):
return channel.get_setting("kick-rejoin",
server.get_setting("kick-rejoin", False))
def _get_delay(self, server, channel):
return channel.get_setting("kick-rejoin-delay",
server.get_setting("kick-rejoin-delay", DELAY))
2019-07-11 20:28:05 +00:00
@utils.hook("self.kick")
def on_kick(self, event):
if self._should_rejoin(event["server"], event["channel"]):
delay = self._get_delay(event["server"], event["channel"])
if delay == 0:
self._rejoin(event["server"], event["channel"].name)
else:
self.timers.add("kick-rejoin", delay, server=event["server"],
channel_name=event["channel"].name)
@utils.hook("timer.kick-rejoin")
def timer(self, event):
self._rejoin(event["server"], event["channel_name"])
def _rejoin(self, server, channel_name):
server.send_join(channel_name)