你正在阅读 Celery 3.1 的文档。开发版本文档见: 此处.

celery.events.dumper 源代码

# -*- coding: utf-8 -*-
"""
    celery.events.dumper
    ~~~~~~~~~~~~~~~~~~~~

    This is a simple program that dumps events to the console
    as they happen. Think of it like a `tcpdump` for Celery events.

"""
from __future__ import absolute_import

import sys

from datetime import datetime

from celery.app import app_or_default
from celery.datastructures import LRUCache
from celery.utils.timeutils import humanize_seconds

TASK_NAMES = LRUCache(limit=0xFFF)

HUMAN_TYPES = {'worker-offline': 'shutdown',
               'worker-online': 'started',
               'worker-heartbeat': 'heartbeat'}

CONNECTION_ERROR = """\
-> Cannot connect to %s: %s.
Trying again %s
"""


[文档]def humanize_type(type): try: return HUMAN_TYPES[type.lower()] except KeyError: return type.lower().replace('-', ' ')
[文档]def say(msg, out=sys.stdout): out.write(msg + '\n')
[文档]class Dumper(object): def __init__(self, out=sys.stdout): self.out = out
[文档] def say(self, msg): say(msg, out=self.out)
[文档] def on_event(self, ev): timestamp = datetime.utcfromtimestamp(ev.pop('timestamp')) type = ev.pop('type').lower() hostname = ev.pop('hostname') if type.startswith('task-'): uuid = ev.pop('uuid') if type in ('task-received', 'task-sent'): task = TASK_NAMES[uuid] = '%s(%s) args=%s kwargs=%s' % ( ev.pop('name'), uuid, ev.pop('args'), ev.pop('kwargs')) else: task = TASK_NAMES.get(uuid, '') return self.format_task_event(hostname, timestamp, type, task, ev) fields = ', '.join('%s=%s' % (key, ev[key]) for key in sorted(ev)) sep = fields and ':' or '' self.say('%s [%s] %s%s %s' % (hostname, timestamp, humanize_type(type), sep, fields))
[文档] def format_task_event(self, hostname, timestamp, type, task, ev): fields = ', '.join('%s=%s' % (key, ev[key]) for key in sorted(ev)) sep = fields and ':' or '' self.say('%s [%s] %s%s %s %s' % ( hostname, timestamp, humanize_type(type), sep, task, fields, ))
[文档]def evdump(app=None, out=sys.stdout): app = app_or_default(app) dumper = Dumper(out=out) dumper.say('-> evdump: starting capture...') conn = app.connection() def _error_handler(exc, interval): dumper.say(CONNECTION_ERROR % ( conn.as_uri(), exc, humanize_seconds(interval, 'in', ' ') )) while 1: try: conn = conn.clone() conn.ensure_connection(_error_handler) recv = app.events.Receiver(conn, handlers={'*': dumper.on_event}) recv.capture() except (KeyboardInterrupt, SystemExit): return conn and conn.close() except conn.connection_errors + conn.channel_errors: dumper.say('-> Connection lost, attempting reconnect')
if __name__ == '__main__': # pragma: no cover evdump()