aboutsummaryrefslogtreecommitdiff
path: root/modules/signals.py
blob: fa26902ee9c48ac2fd180b714be19c161b616215 (about) (plain) (blame)
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
import signal
from src import Config, ModuleManager, utils

@utils.export("serverset", {"setting": "quit-quote",
    "help": "Set whether I pick a random quote to /quit with",
    "validate": utils.bool_or_none})
class Module(ModuleManager.BaseModule):
    def on_load(self):
        signal.signal(signal.SIGINT, self.SIGINT)
        signal.signal(signal.SIGUSR1, self.SIGUSR1)

    def SIGINT(self, signum, frame):
        print()
        self.bot.trigger(lambda: self._kill(signum))

    def _kill(self, signum):
        self.events.on("signal.interrupt").call(signum=signum)

        for server in self.bot.servers.values():
            reason = "Leaving"
            if server.get_setting("quit-quote", True):
                reason = self.events.on("get.quit-quote"
                    ).call_for_result(default=reason)
            server.send_quit(reason)

        self.events.on("writebuffer.empty").hook(
            lambda event: self.bot.disconnect(event["server"]))

    def SIGUSR1(self, signum, frame):
        self.bot.trigger(self._reload_config)

    def _reload_config(self):
        self.bot.log.info("Reloading config file", [])
        self.bot.config.load()
        self.bot.log.info("Reloaded config file", [])