bitbot-3.11-fork/modules/channel_save.py

35 lines
1.3 KiB
Python

from src import ModuleManager, utils
BATCH_MAX = 10
class Module(ModuleManager.BaseModule):
@utils.hook("received.001")
def on_connect(self, event):
channels = event["server"].get_setting("autojoin", [])
if channels:
channel_batches = [channels[i:i+BATCH_MAX] for i in range(
0, len(channels), BATCH_MAX)]
for channel_batch in channel_batches:
event["server"].send_joins(channel_batch)
@utils.hook("self.join")
def on_join(self, event):
channels = event["server"].get_setting("autojoin", [])
channel_name = event["server"].irc_lower(event["channel"].name)
if not channel_name in channels:
channels.append(channel_name)
event["server"].set_setting("autojoin", channels)
def _remove_channel(self, server, channel_name):
channels = server.get_setting("autojoin", [])
if channel_name in channels:
channels.remove(channel_name)
server.set_setting("autojoin", channels)
@utils.hook("self.part")
def on_part(self, event):
self._remove_channel(event["server"], event["channel"].name)
@utils.hook("self.kick")
def on_kick(self, event):
self._remove_channel(event["server"], event["channel"].name)