mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
120 lines
4.4 KiB
Python
120 lines
4.4 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
|
|
|
|
import sopel_modules.SpiceBot as SpiceBot
|
|
|
|
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):
|
|
SpiceBot.botlogs.log('SpiceBot_Channels', "Starting setup procedure")
|
|
SpiceBot.botevents.startup_add([SpiceBot.botevents.BOT_CHANNELS])
|
|
bot.config.define_section("SpiceBot_Channels", SpiceBot_Channels_MainSection, validate=False)
|
|
|
|
|
|
@sopel.module.event(SpiceBot.botevents.RPL_WELCOME)
|
|
@sopel.module.rule('.*')
|
|
def unkickable_bot(bot, trigger):
|
|
if bot.config.SpiceBot_Channels.operadmin:
|
|
bot.write(('SAMODE', bot.nick, '+q'))
|
|
|
|
|
|
@sopel.module.event(SpiceBot.botevents.RPL_WELCOME)
|
|
@sopel.module.rule('.*')
|
|
def request_channels_list_initial(bot, trigger):
|
|
|
|
SpiceBot.botchannels.bot_part_empty(bot)
|
|
|
|
SpiceBot.botlogs.log('SpiceBot_Channels', "Sending Request for all server channels")
|
|
SpiceBot.botchannels.channel_list_request(bot)
|
|
|
|
starttime = time.time()
|
|
|
|
# wait 60 seconds for initial population of information
|
|
while not SpiceBot.botchannels.SpiceBot_Channels['InitialProcess']:
|
|
if time.time() - starttime >= 60:
|
|
SpiceBot.botlogs.log('SpiceBot_Channels', "Initial Channel list populating Timed Out")
|
|
SpiceBot.botchannels.SpiceBot_Channels['InitialProcess'] = True
|
|
else:
|
|
pass
|
|
|
|
foundchannelcount = len(SpiceBot.botchannels.SpiceBot_Channels['list'].keys())
|
|
SpiceBot.botlogs.log('SpiceBot_Channels', "Channel listing finished! " + str(foundchannelcount) + " channel(s) found.")
|
|
|
|
SpiceBot.botchannels.join_all_channels(bot)
|
|
SpiceBot.botchannels.chanadmin_all_channels(bot)
|
|
|
|
if "*" in SpiceBot.botchannels.SpiceBot_Channels['list'].keys():
|
|
del SpiceBot.botchannels.SpiceBot_Channels['list']["*"]
|
|
|
|
SpiceBot.botchannels.bot_part_empty(bot)
|
|
SpiceBot.botevents.trigger(bot, SpiceBot.botevents.BOT_CHANNELS, "SpiceBot_Channels")
|
|
|
|
|
|
@sopel.module.event('321')
|
|
@sopel.module.rule('.*')
|
|
def watch_chanlist_start(bot, trigger):
|
|
SpiceBot.botchannels.channel_list_recieve_start()
|
|
|
|
|
|
@sopel.module.event('322')
|
|
@sopel.module.rule('.*')
|
|
def watch_chanlist_populate(bot, trigger):
|
|
SpiceBot.botchannels.channel_list_recieve_input(trigger)
|
|
|
|
|
|
@sopel.module.event('323')
|
|
@sopel.module.rule('.*')
|
|
def watch_chanlist_complete(bot, trigger):
|
|
SpiceBot.botchannels.channel_list_recieve_finish()
|
|
|
|
|
|
@sopel.module.event(SpiceBot.botevents.BOT_CHANNELS)
|
|
@sopel.module.rule('.*')
|
|
def trigger_channel_list_recurring(bot, trigger):
|
|
while True:
|
|
time.sleep(1800)
|
|
SpiceBot.botchannels.bot_part_empty(bot)
|
|
|
|
oldlist = list(SpiceBot.botchannels.SpiceBot_Channels['list'].keys())
|
|
SpiceBot.botchannels.channel_list_request(bot)
|
|
|
|
while SpiceBot.botchannels.channel_lock:
|
|
pass
|
|
|
|
newlist = [item.lower() for item in oldlist if item.lower() not in list(SpiceBot.botchannels.SpiceBot_Channels['list'].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())
|
|
|
|
SpiceBot.botchannels.join_all_channels(bot)
|
|
|
|
SpiceBot.botchannels.chanadmin_all_channels(bot)
|
|
|
|
if "*" in SpiceBot.botchannels.SpiceBot_Channels['list'].keys():
|
|
del SpiceBot.botchannels.SpiceBot_Channels['list']["*"]
|
|
SpiceBot.botchannels.bot_part_empty(bot)
|