1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
import datetime
from src import ModuleManager, utils
PRUNE_TIMEDELTA = datetime.timedelta(weeks=2)
SETTING = utils.BoolSetting("inactive-channels",
"Whether or not to leave inactive channels after 2 weeks")
@utils.export("botset", SETTING)
@utils.export("serverset", SETTING)
class Module(ModuleManager.BaseModule):
def _get_timestamp(self, channel):
return channel.get_setting("last-message", None)
def _set_timestamp(self, channel):
channel.set_setting("last-message",
utils.datetime.format.iso8601(utils.datetime.utcnow()))
def _del_timestamp(self, channel):
channel.del_setting("last-message")
@utils.hook("new.channel")
def new_channel(self, event):
if self._get_timestamp(event["channel"]) == None:
self._set_timestamp(event["channel"])
@utils.hook("cron")
@utils.kwarg("schedule", "0")
def hourly(self, event):
parts = []
now = utils.datetime.utcnow()
botwide_setting = self.bot.get_setting("inactive-channels", False)
for server in self.bot.servers.values():
if not server.get_setting("inactive-channels", botwide_setting):
continue
for channel in server.channels:
timestamp = self._get_timestamp(channel)
if timestamp:
dt = utils.datetime.parse.iso8601(timestamp)
if (now-dt) >= PRUNE_TIMEDELTA:
parts.append([server, channel])
for server, channel in parts:
self.log.warn("Leaving %s:%s due to channel inactivity",
[str(server), str(channel)])
channel.send_part("Channel inactive")
self._del_timestamp(channel)
@utils.hook("received.message.channel")
def channel_message(self, event):
self._set_timestamp(event["channel"])
|
