notnull/banned
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add signal handling

Browse Source 
- SIGINT (Ctrl-C) and SIGTERM will do a clean quit now.
- SIGUSR1 will restart the bot.
This commit is contained in:
nyuszika7h 2014-12-22 23:38:06 +01:00
parent c611db5109
commit 036ac29f4b
1 changed files with 30 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
modules/wolfgame.py
View File

@ -38,6 +38,7 @@ import math
import fnmatch
import random
import subprocess
import signal
from imp import reload
BOLD = "\u0002"
@ -107,6 +108,16 @@ if botconfig.DEBUG_MODE:
def connect_callback(cli):
def handler(signum, frame):
if signum in (signal.SIGINT, signal.SIGTERM):
forced_exit(cli, "<console>", botconfig.CHANNEL, "")
elif signum == signal.SIGUSR1:
restart_program(cli, "<console>", botconfig.CHANNEL, "")
signal.signal(signal.SIGINT, handler)
signal.signal(signal.SIGTERM, handler)
signal.signal(signal.SIGUSR1, handler)
to_be_devoiced = []
cmodes = []
@ -362,11 +373,17 @@ def make_stasis(nick, penalty):
var.STASISED[cloak] += penalty
var.set_stasis(cloak, var.STASISED[cloak])
@pmcmd("fdie", "fbye", admin_only=True)
@cmd("fdie", "fbye", admin_only=True)
def forced_exit(cli, nick, chan, *rest): # Admin Only
@pmcmd("fdie", "fbye", raw_nick=True)
@cmd("fdie", "fbye", raw_nick=True)
def forced_exit(cli, raw_nick, chan, *rest): # Admin Only
"""Forces the bot to close."""
(nick, _, _, cloak) = parse_nick(raw_nick)
if nick != "<console>" and not var.is_admin(cloak):
cli.notice(nick, "You are not an admin.")
return
if var.PHASE in ("day", "night"):
#ignore all errors that prevent the bot from stopping
try:
@ -384,9 +401,16 @@ def forced_exit(cli, nick, chan, *rest): # Admin Only
@cmd("frestart", admin_only=True)
def restart_program(cli, nick, chan, rest):
@cmd("frestart", raw_nick=True)
def restart_program(cli, raw_nick, chan, rest):
"""Restarts the bot."""
(nick, _, _, cloak) = parse_nick(raw_nick)
if nick != "<console>" and not var.is_admin(cloak):
cli.notice(nick, "You are not an admin.")
return
try:
if var.PHASE in ("day", "night"):
try:
@ -414,7 +438,7 @@ def restart_program(cli, nick, chan, rest):
else:
os.execl(python, python, *sys.argv)
@pmcmd("frestart", admin_only=True)
@pmcmd("frestart")
def pm_restart_program(cli, nick, rest):
restart_program(cli, nick, nick, rest)