Add auto restart for celery

mantiumai · Aug 21, 2023 · d555842 · d555842
1 parent 9a0e695
commit d555842
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/chirps/base_app/management/commands/celery.py b/chirps/base_app/management/commands/celery.py
@@ -2,6 +2,7 @@
 import os
 
 from django.core.management.base import BaseCommand
+from django.utils import autoreload
 
 
 class Command(BaseCommand):
@@ -18,12 +19,13 @@ def add_arguments(self, parser):
     def handle(self, *args, **options):
         """Handle the command."""
         if options['start']:
-            self.start()
+            self.stop()
+            autoreload.run_with_reloader(self.start)
         elif options['stop']:
             self.stop()
         elif options['restart']:
             self.stop()
-            self.start()
+            autoreload.run_with_reloader(self.start)
 
     def start(self):
         """Start the celery server."""