Switch -recursive to asyncio’s .cancel()

RecursiveController used a custom .cancel() method before. Instead we
can simply cancel .run() and handle the CancelledError inside run() and
fetch().

1 files changed, 3 insertions, 2 deletions

diff --git a/crocoite/cli.py b/crocoite/cli.py
index b0ad53a..f9ef52c 100644
--- a/crocoite/cli.py
+++ b/crocoite/cli.py
@@ -120,11 +120,12 @@ def recursive ():
             tempdir=args.tempdir, prefix=args.prefix,
             concurrency=args.concurrency)
 
+    run = asyncio.ensure_future (controller.run ())
     loop = asyncio.get_event_loop()
-    stop = lambda signum: controller.cancel ()
+    stop = lambda signum: run.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.run_until_complete(run)
     loop.close()
 
     return 0