Proper kill after processing.

basilfx · Jul 25, 2017 · 34d9a2e · 34d9a2e
1 parent eae2a41
commit 34d9a2e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/photod-backend/photod/cli/management/commands/process.py b/photod-backend/photod/cli/management/commands/process.py
@@ -111,9 +111,12 @@ def handle(self, *args, **options):
                 job.save()
 
             # Wait for the processes to finish.
-            self.stdout.write(
-                self.style.SUCCESS('Waiting for workers to finish.'))
+            self.stdout.write(self.style.SUCCESS(
+                'Waiting for workers to finish.'))
+            pool.close_and_join()
         except KeyboardInterrupt:
+            self.stdout.write(self.style.WARNING(
+                'Interrupt received, killing workers softly.'))
             pool.close_and_join()
         finally:
             # Update job.