From 3e69f8b34a48ffa4df4805c53aeaba144d91c6bc Mon Sep 17 00:00:00 2001
From: Lars-Dominik Braun <lars@6xq.net>
Date: Thu, 11 Oct 2018 17:03:31 +0200
Subject: recursive: Gracefully shut down on SIGINT/TERM

---
 crocoite/cli.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'crocoite/cli.py')

diff --git a/crocoite/cli.py b/crocoite/cli.py
index 0319dc9..55ff4a1 100644
--- a/crocoite/cli.py
+++ b/crocoite/cli.py
@@ -22,7 +22,7 @@
 Command line interface
 """
 
-import argparse, json, sys
+import argparse, json, sys, signal
 
 from . import behavior
 from .controller import SinglePageController, defaultSettings, \
@@ -104,6 +104,9 @@ def recursive ():
             concurrency=args.concurrency)
 
     loop = asyncio.get_event_loop()
+    stop = lambda signum: controller.cancel ()
+    loop.add_signal_handler (signal.SIGINT, stop, signal.SIGINT)
+    loop.add_signal_handler (signal.SIGTERM, stop, signal.SIGTERM)
     loop.run_until_complete(controller.run ())
     loop.close()
 
-- 
cgit v1.2.3