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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
import signal, sys
from src import Config, IRCLine, ModuleManager, utils
class Module(ModuleManager.BaseModule):
def on_load(self):
self._exited = False
signal.signal(signal.SIGINT, self.SIGINT)
signal.signal(signal.SIGUSR1, self.SIGUSR1)
signal.signal(signal.SIGHUP, self.SIGHUP)
def SIGINT(self, signum, frame):
print()
self.bot.trigger(lambda: self._kill(signum))
def _kill(self, signum):
if self._exited:
return
self._exited = True
self.events.on("signal.interrupt").call(signum=signum)
written = False
for server in list(self.bot.servers.values()):
if server.connected:
server.socket.clear_send_buffer()
line = IRCLine.ParsedLine("QUIT", ["Shutting down"])
sent_line = server.send(line, immediate=True)
sent_line.events.on("send").hook(self._make_hook(server))
server.send_enabled = False
written = True
if not written:
sys.exit()
def _make_hook(self, server):
return lambda e: self._disconnect_hook(server)
def _disconnect_hook(self, server):
self.bot.disconnect(server)
if not self.bot.servers:
sys.exit()
def SIGUSR1(self, signum, frame):
self.bot.trigger(self._reload_config)
def SIGHUP(self, signum, frame):
self.bot.trigger(self._SIGHUP)
def _SIGHUP(self):
self._reload_config()
self._reload_modules()
def _reload_config(self):
self.bot.log.info("Reloading config file")
self.bot.config.load()
self.bot.log.info("Reloaded config file")
def _reload_modules(self):
self.bot.log.info("Reloading modules")
result = self.bot.try_reload_modules()
if result.success:
self.bot.log.info(result.message)
else:
self.bot.log.warn(result.message)
|
