Browse Source

Tests passing

Ask Solem 12 years ago
parent
commit
aa260e75de
2 changed files with 3 additions and 2 deletions
  1. 1 0
      celery/worker/job.py
  2. 2 2
      celery/worker/strategy.py

+ 1 - 0
celery/worker/job.py

@@ -49,6 +49,7 @@ _does_debug = False
 
 
 
 
 def __optimize__():
 def __optimize__():
+    # this is also called by celery.task.trace.setup_worker_optimizations
     global _does_debug
     global _does_debug
     global _does_info
     global _does_info
     _does_debug = logger.isEnabledFor(logging.DEBUG)
     _does_debug = logger.isEnabledFor(logging.DEBUG)

+ 2 - 2
celery/worker/strategy.py

@@ -24,7 +24,7 @@ from .state import task_reserved
 
 
 def default(task, app, consumer,
 def default(task, app, consumer,
             info=logger.info, error=logger.error, task_reserved=task_reserved,
             info=logger.info, error=logger.error, task_reserved=task_reserved,
-            to_timestamp=to_timestamp, to_system_tz=timezone.to_system):
+            to_system_tz=timezone.to_system):
     hostname = consumer.hostname
     hostname = consumer.hostname
     eventer = consumer.event_dispatcher
     eventer = consumer.event_dispatcher
     Req = Request
     Req = Request
@@ -39,7 +39,7 @@ def default(task, app, consumer,
     handle = consumer.handle_task
     handle = consumer.handle_task
     limit_task = consumer._limit_task
     limit_task = consumer._limit_task
 
 
-    def task_message_handler(message, body, ack):
+    def task_message_handler(message, body, ack, to_timestamp=to_timestamp):
         req = Req(body, on_ack=ack, app=app, hostname=hostname,
         req = Req(body, on_ack=ack, app=app, hostname=hostname,
                   eventer=eventer, task=task,
                   eventer=eventer, task=task,
                   connection_errors=connection_errors,
                   connection_errors=connection_errors,