60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
|
from src import ModuleManager, utils
|
||
|
|
|
||
|
|
SETTING = "cron-reminders"
|
||
|
|
|
||
|
|
class Module(ModuleManager.BaseModule):
|
||
|
|
_name = "cron"
|
||
|
|
|
||
|
|
@utils.hook("cron")
|
||
|
|
def cron(self, event):
|
||
|
|
try:
|
||
|
|
self._catch_cron(event["schedule"])
|
||
|
|
except Exception as e:
|
||
|
|
self.log.error("Failed to call cron reminders: %s", [str(e)],
|
||
|
|
exc_info=True)
|
||
|
|
def _catch_cron(self, schedule_check):
|
||
|
|
all = []
|
||
|
|
for server in self.bot.servers.values():
|
||
|
|
channel_reminders = server.find_all_user_channel_settings(SETTING)
|
||
|
|
for channel_name, nickname, value in channel_reminders:
|
||
|
|
if channel_name in server.channels:
|
||
|
|
channel = server.channels.get(channel_name)
|
||
|
|
for schedule, text in value:
|
||
|
|
if schedule_check(schedule):
|
||
|
|
all.append([channel, nickname, text])
|
||
|
|
|
||
|
|
user_reminders = server.get_all_user_settings(SETTING)
|
||
|
|
for nickname, value in user_reminders:
|
||
|
|
if server.has_user(nickname):
|
||
|
|
user = server.get_user(nickname)
|
||
|
|
for schedule, text in value:
|
||
|
|
if schedule_check(schedule):
|
||
|
|
all.append([user, user.nickname, text])
|
||
|
|
|
||
|
|
for target, nickname, text in all:
|
||
|
|
target.send_message("%s, reminder: %s" % (nickname, text))
|
||
|
|
|
||
|
|
|
||
|
|
@utils.hook("received.command.cron")
|
||
|
|
@utils.kwarg("permission", "cron")
|
||
|
|
@utils.spec("!'add !<schedule>word !<reminder>string")
|
||
|
|
@utils.spec("!'remove !<index>int")
|
||
|
|
def command(self, event):
|
||
|
|
user_id = event["user"].get_id()
|
||
|
|
r = event["target"].get_user_setting(user_id, SETTING, [])
|
||
|
|
|
||
|
|
if event["spec"][0] == "add":
|
||
|
|
r.append([event["spec"][1].replace("+", " "), event["spec"][2]])
|
||
|
|
event["stdout"].write("%s: added reminder"
|
||
|
|
% event["user"].nickname)
|
||
|
|
|
||
|
|
elif event["spec"][0] == "remove":
|
||
|
|
if len(r) <= event["spec"][1]:
|
||
|
|
raise utils.EventError("%s: invalid index (max %d)"
|
||
|
|
% (event["user"].nickname, len(r)-1))
|
||
|
|
schedule, text = r.pop(event["spec"][1])
|
||
|
|
event["stdout"].write("%s: removed reminder %d: %s (%s)"
|
||
|
|
% (event["user"].nickname, event["spec"][1], schedule, text))
|
||
|
|
|
||
|
|
event["target"].set_user_setting(user_id, SETTING, r)
|