python: Try to shut ashd-wsgi down more nicely upon signal receipt.

[ashd.git] / python / ashd-wsgi

diff --git a/python/ashd-wsgi b/python/ashd-wsgi
index b671059..d5438fa 100755 (executable)
--- a/python/ashd-wsgi
+++ b/python/ashd-wsgi
@@ -1,6 +1,6 @@
 #!/usr/bin/python
 
-import sys, os, getopt, socket, logging, time
+import sys, os, getopt, socket, logging, time, signal
 import ashd.util, ashd.serve
 try:
     import pdm.srv
@@ -191,6 +191,11 @@ except ValueError as exc:
     sys.stderr.write("ashd-wsgi: %s\n" % exc)
     sys.exit(1)
 
+def sigterm(sig, frame):
+    socket.fromfd(0, socket.AF_UNIX, socket.SOCK_SEQPACKET).shutdown(socket.SHUT_RDWR) # :P
+for signum in [signal.SIGINT, signal.SIGTERM]:
+    signal.signal(signum, sigterm)
+
 reqhandler = hclass(**hargs)
 try:
     ashd.util.serveloop(handle)