Add periodic_task() wrapper for MiniHuey.

coleifer · Jun 28, 2022 · 390925d · 390925d
1 parent 66d1f1a
commit 390925d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/huey/contrib/mini.py b/huey/contrib/mini.py
@@ -66,6 +66,11 @@ def _schedule(args=None, kwargs=None, delay=None, eta=None):
 
         return decorator
 
+    def periodic_task(self, validate_func):
+        def decorator(fn):
+            return self.task(validate_func)(fn)
+        return decorator
+
     def start(self):
         if self._run_t is not None:
             raise Exception('Task runner is already running.')