allow individual channels to opt out of inactivity pruning

This commit is contained in:
jesopo 2020-02-27 10:33:15 +00:00
parent 5a78f03aa7
commit 4d38639742

View file

@ -2,11 +2,13 @@ import datetime
from src import ModuleManager, utils from src import ModuleManager, utils
PRUNE_TIMEDELTA = datetime.timedelta(weeks=2) PRUNE_TIMEDELTA = datetime.timedelta(weeks=2)
SETTING = utils.BoolSetting("inactive-channels", SETTING_NAME = "inactive-channels"
SETTING = utils.BoolSetting(SETTING_NAME,
"Whether or not to leave inactive channels after 2 weeks") "Whether or not to leave inactive channels after 2 weeks")
@utils.export("botset", SETTING) @utils.export("botset", SETTING)
@utils.export("serverset", SETTING) @utils.export("serverset", SETTING)
@utils.export("channelset", SETTING)
class Module(ModuleManager.BaseModule): class Module(ModuleManager.BaseModule):
def _get_timestamp(self, channel): def _get_timestamp(self, channel):
return channel.get_setting("last-message", None) return channel.get_setting("last-message", None)
@ -26,13 +28,18 @@ class Module(ModuleManager.BaseModule):
def hourly(self, event): def hourly(self, event):
parts = [] parts = []
now = utils.datetime.utcnow() now = utils.datetime.utcnow()
botwide_setting = self.bot.get_setting("inactive-channels", False) botwide_setting = self.bot.get_setting(SETTING_NAME, False)
for server in self.bot.servers.values(): for server in self.bot.servers.values():
if not server.get_setting("inactive-channels", botwide_setting): serverwide_setting = server.get_setting(
SETTING_NAME, botwide_setting)
if not serverwide_setting:
continue continue
for channel in server.channels: for channel in server.channels:
if not channel.get_setting(SETTING_NAME, serverwide_setting):
continue
timestamp = self._get_timestamp(channel) timestamp = self._get_timestamp(channel)
if timestamp: if timestamp:
dt = utils.datetime.parse.iso8601(timestamp) dt = utils.datetime.parse.iso8601(timestamp)