mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-11 06:00:59 -05:00
133 lines
4.8 KiB
Python
133 lines
4.8 KiB
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
from __future__ import unicode_literals, absolute_import, print_function, division
|
|
|
|
# sopel imports
|
|
import sopel.module
|
|
from sopel.config.types import StaticSection, ValidatedAttribute, ListAttribute
|
|
|
|
from sopel_modules.SpiceBot.Logs import botlogs
|
|
from sopel_modules.SpiceBot.Events import botevents
|
|
from sopel_modules.SpiceBot_SBTools import join_all_channels, chanadmin_all_channels, channel_list_current
|
|
|
|
import spicemanip
|
|
|
|
import time
|
|
|
|
|
|
class SpiceBot_Channels_MainSection(StaticSection):
|
|
announcenew = ValidatedAttribute('announcenew', default=False)
|
|
joinall = ValidatedAttribute('joinall', default=False)
|
|
operadmin = ValidatedAttribute('operadmin', default=False)
|
|
chanignore = ListAttribute('chanignore')
|
|
|
|
|
|
def configure(config):
|
|
config.define_section("SpiceBot_Channels", SpiceBot_Channels_MainSection, validate=False)
|
|
config.SpiceBot_Channels.configure_setting('announcenew', 'SpiceBot_Channels Announce New Channels')
|
|
config.SpiceBot_Channels.configure_setting('joinall', 'SpiceBot_Channels JOIN New Channels')
|
|
config.SpiceBot_Channels.configure_setting('operadmin', 'SpiceBot_Channels OPER ADMIN MODE')
|
|
config.SpiceBot_Channels.configure_setting('chanignore', 'SpiceBot_Channels Ignore JOIN for channels')
|
|
|
|
|
|
def setup(bot):
|
|
botlogs.log('SpiceBot_Channels', "Starting setup procedure")
|
|
botevents.startup_add([botevents.BOT_CHANNELS])
|
|
|
|
bot.config.define_section("SpiceBot_Channels", SpiceBot_Channels_MainSection, validate=False)
|
|
|
|
if "SpiceBot_Channels" not in bot.memory:
|
|
bot.memory["SpiceBot_Channels"] = {"channels": {}, "InitialProcess": False, "ProcessLock": False}
|
|
|
|
|
|
def shutdown(bot):
|
|
if "SpiceBot_Channels" in bot.memory:
|
|
del bot.memory["SpiceBot_Channels"]
|
|
|
|
|
|
@sopel.module.event(botevents.BOT_CONNECTED)
|
|
@sopel.module.rule('.*')
|
|
def unkickable_bot(bot, trigger):
|
|
if bot.config.SpiceBot_Channels.operadmin:
|
|
bot.write(('SAMODE', bot.nick, '+q'))
|
|
|
|
|
|
@sopel.module.event(botevents.BOT_CONNECTED)
|
|
@sopel.module.rule('.*')
|
|
def trigger_channel_list_initial(bot, trigger):
|
|
|
|
bot_part_empty(bot)
|
|
|
|
bot.write(['LIST'])
|
|
bot.memory['SpiceBot_Channels']['ProcessLock'] = True
|
|
|
|
botlogs.log('SpiceBot_Channels', "Initial Channel list populating")
|
|
starttime = time.time()
|
|
|
|
while not bot.memory['SpiceBot_Channels']['InitialProcess']:
|
|
timesince = time.time() - starttime
|
|
if timesince < 60:
|
|
pass
|
|
else:
|
|
botlogs.log('SpiceBot_Channels', "Initial Channel list populating Timed Out")
|
|
bot.memory['SpiceBot_Channels']['InitialProcess'] = True
|
|
|
|
channel_list_current(bot)
|
|
foundchannelcount = len(bot.memory['SpiceBot_Channels']['channels'].keys())
|
|
botlogs.log('SpiceBot_Channels', "Channel listing finished! " + str(foundchannelcount) + " channel(s) found.")
|
|
|
|
join_all_channels(bot)
|
|
chanadmin_all_channels(bot)
|
|
|
|
if "*" in bot.memory['SpiceBot_Channels']['channels']:
|
|
del bot.memory['SpiceBot_Channels']['channels']["*"]
|
|
|
|
bot_part_empty(bot)
|
|
botevents.trigger(bot, botevents.BOT_CHANNELS, "SpiceBot_Channels")
|
|
|
|
|
|
@sopel.module.event(botevents.BOT_CHANNELS)
|
|
@sopel.module.rule('.*')
|
|
def trigger_channel_list_recurring(bot, trigger):
|
|
while True:
|
|
try:
|
|
time.sleep(1800)
|
|
bot_part_empty(bot)
|
|
|
|
oldlist = list(bot.memory['SpiceBot_Channels']['channels'].keys())
|
|
bot.write(['LIST'])
|
|
bot.memory['SpiceBot_Channels']['ProcessLock'] = True
|
|
|
|
while bot.memory['SpiceBot_Channels']['ProcessLock']:
|
|
pass
|
|
|
|
newlist = [item.lower() for item in oldlist if item.lower() not in list(bot.memory['SpiceBot_Channels']['channels'].keys())]
|
|
if "*" in newlist:
|
|
newlist.remove("*")
|
|
if len(newlist) and bot.config.SpiceBot_Channels.announcenew:
|
|
bot.osd(["The Following channel(s) are new:", spicemanip.main(newlist, 'andlist')], bot.channels.keys())
|
|
|
|
join_all_channels(bot)
|
|
|
|
chanadmin_all_channels(bot)
|
|
|
|
if "*" in bot.memory['SpiceBot_Channels']['channels']:
|
|
del bot.memory['SpiceBot_Channels']['channels']["*"]
|
|
bot_part_empty(bot)
|
|
|
|
except KeyError:
|
|
return
|
|
|
|
|
|
def bot_part_empty(bot):
|
|
"""Don't stay in empty channels"""
|
|
ignorepartlist = []
|
|
if bot.config.core.logging_channel:
|
|
ignorepartlist.append(bot.config.core.logging_channel)
|
|
|
|
for channel in bot.channels.keys():
|
|
if len(bot.channels[channel].privileges.keys()) == 1 and channel not in ignorepartlist and channel.startswith("#"):
|
|
bot.part(channel, "Leaving Empty Channel")
|
|
if channel.lower() in bot.memory['SpiceBot_Channels']['channels']:
|
|
del bot.memory['SpiceBot_Channels']['channels'][channel.lower()]
|