daphne/channels/management/commands/runwsserver.py
2015-11-09 13:01:02 +01:00

39 lines
1.8 KiB
Python

from django.core.management import BaseCommand, CommandError
from channels import DEFAULT_CHANNEL_BACKEND, channel_backends
from channels.log import setup_logger
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument('port', nargs='?',
help='Optional port number')
def handle(self, *args, **options):
self.verbosity = options.get("verbosity", 1)
self.logger = setup_logger('django.channels', self.verbosity)
# Get the backend to use
channel_backend = channel_backends[DEFAULT_CHANNEL_BACKEND]
if channel_backend.local_only:
raise CommandError(
"You have a process-local channel backend configured, and so cannot run separate interface servers.\n"
"Configure a network-based backend in CHANNEL_BACKENDS to use this command."
)
# Run the interface
port = int(options.get("port", None) or 9000)
try:
import asyncio # NOQA
except ImportError:
from channels.interfaces.websocket_twisted import WebsocketTwistedInterface
self.logger.info("Running Twisted/Autobahn WebSocket interface server")
self.logger.info(" Channel backend: %s", channel_backend)
self.logger.info(" Listening on: ws://0.0.0.0:%i" % port)
WebsocketTwistedInterface(channel_backend=channel_backend, port=port).run()
else:
from channels.interfaces.websocket_asyncio import WebsocketAsyncioInterface
self.logger.info("Running asyncio/Autobahn WebSocket interface server")
self.logger.info(" Channel backend: %s", channel_backend)
self.logger.info(" Listening on: ws://0.0.0.0:%i", port)
WebsocketAsyncioInterface(channel_backend=channel_backend, port=port).run()