mirror of
https://github.com/django/daphne.git
synced 2025-04-27 12:13:47 +03:00
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
|
|
class BaseConsumer(object):
|
|
"""
|
|
Base class-based consumer class. Provides the mechanisms to be a direct
|
|
routing object and a few other things.
|
|
|
|
Class-based consumers should be used directly in routing with their
|
|
filters, like so::
|
|
|
|
routing = [
|
|
JsonWebsocketConsumer(path=r"^/liveblog/(?P<slug>[^/]+)/"),
|
|
]
|
|
"""
|
|
|
|
method_mapping = {}
|
|
|
|
def __init__(self, message, **kwargs):
|
|
"""
|
|
Constructor, called when a new message comes in (the consumer is
|
|
the uninstantiated class, so calling it creates it)
|
|
"""
|
|
self.message = message
|
|
self.dispatch(message, **kwargs)
|
|
|
|
@classmethod
|
|
def channel_names(cls):
|
|
"""
|
|
Returns a list of channels this consumer will respond to, in our case
|
|
derived from the method_mapping class attribute.
|
|
"""
|
|
return set(cls.method_mapping.keys())
|
|
|
|
def dispatch(self, message, **kwargs):
|
|
"""
|
|
Called with the message and all keyword arguments; uses method_mapping
|
|
to choose the right method to call.
|
|
"""
|
|
return getattr(self, self.method_mapping[message.channel.name])(message, **kwargs)
|