Skip to content

Commit e8c3951

Browse files
committed
chore(tasks): change apply_async parameters
1 parent fca588c commit e8c3951

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

rootfs/tasks/task.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,14 +44,14 @@ def register_task(*args, **kwargs):
4444
return register_task
4545

4646

47-
def apply_async(target, delay=0, callback=None, *args, **kwargs):
47+
def apply_async(target, delay=0, callback=None, args=(), kwargs=None):
4848
target_id = "%s.%s" % (target.__module__, target.__name__)
4949
if target_id not in TASKS:
5050
raise NotImplemented("This task is not registered.")
5151
message = json.dumps({
5252
"target_id": target_id,
5353
"args": args,
54-
"kwargs": kwargs
54+
"kwargs": {} if kwargs == None else kwargs
5555
}).encode("utf-8")
5656
if delay <= 0:
5757
NSQD_WRITER.pub(NSQ_TOPIC, message, callback=callback)

rootfs/tasks/tests/test_task.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,12 @@ def t2(t):
2727
self.assertEqual(time.time() - t > 3, True)
2828

2929
def callback(addr, msg):
30-
self.assertEqual(msg==b'OK', True)
30+
self.assertEqual(msg == b'OK', True)
3131

3232
threading.Thread(
3333
target=tornado.ioloop.IOLoop.current().start).start()
3434
time.sleep(3)
35-
apply_async(t1, callback=callback, name="hello", value="word")
36-
apply_async(t2, callback=callback, delay=3000, t=time.time())
35+
apply_async(t1, callback=callback, args=("hello", "word"))
36+
apply_async(t2, callback=callback, delay=3000, args=(time.time(), ))
3737
time.sleep(12)
3838
tornado.ioloop.IOLoop.current().stop()

0 commit comments

Comments
 (0)