Skip to content

Commit

Permalink
A task may or may not require any information as part of the message
Browse files Browse the repository at this point in the history
  • Loading branch information
dnaeon committed Jan 26, 2015
1 parent 0917309 commit 2fbe665
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions src/vpoller/decorators.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
from vpoller.registry import registry


def task(name, required):
def task(name, required=None):
"""
A decorator for creating new tasks
Expand All @@ -51,6 +51,6 @@ def wrapper(*args, **kwargs):
result = {'success': 1, 'msg': e.message}
finally:
return result
registry.register(name, required, fn)
registry.register(name=name, fn=fn, required=required)
return wrapper
return decorator
4 changes: 2 additions & 2 deletions src/vpoller/registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,14 +41,14 @@ def __init__(self):
def __contains__(self, item):
return item in self._registry

def register(self, name, required, fn):
def register(self, name, fn, required=None):
"""
Register a new task
Args:
name (str): Name of the task
required (list): List of required message keys
fn (function): A function executing the task
required (list): List of required message keys
"""
logging.info('Registering task %s', name)
Expand Down

0 comments on commit 2fbe665

Please sign in to comment.