Fix atexit not working on SIGTERM (#1882)

2019-01-30 21:47:16 +08:00 · 2019-01-30 21:47:16 +08:00 · dd46a64e79
commit dd46a64e79
parent 31a7effa5f
1 changed files with 4 additions and 0 deletions
--- a/zeronet.py
+++ b/zeronet.py
@ -11,6 +11,10 @@ def main():

    main = None
    try:
+        import signal
+
+        signal.signal(signal.SIGTERM, lambda signum, stack_frame: sys.exit(0))
+        
        app_dir = os.path.dirname(os.path.abspath(__file__))
        os.chdir(app_dir)  # Change working dir to zeronet.py dir
        sys.path.insert(0, os.path.join(app_dir, "src/lib"))  # External liblary directory