Add extensibility to registry.Message in ancitipation of new features

coleifer · Jun 17, 2021 · 6b1a9f9 · 6b1a9f9
1 parent 345857d
commit 6b1a9f9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/huey/registry.py b/huey/registry.py
@@ -7,6 +7,11 @@
                                  'priority', 'args', 'kwargs', 'on_complete',
                                  'on_error'))
 
+# Automatically set missing parameters to None. This is kind-of a hack, but it
+# allows us to add new parameters while continuing to be able to handle
+# messages enqueued with a smaller-set of arguments.
+Message.__new__.__defaults__ = (None,) * len(Message._fields)
+
 
 class Registry(object):
     def __init__(self):