django/daphne
1
1
Fork 0
mirror of https://github.com/django/daphne.git synced 2025-07-12 17:02:18 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add --alias option to runworker

Browse Source
This commit is contained in:
Andrew Godwin 2016-03-10 17:55:28 -08:00
parent 053850fdd6
commit 1ee7b2861c
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
channels/management/commands/runworker.py
View File

@ -8,11 +8,16 @@ from channels.worker import Worker
class Command(BaseCommand): class Command(BaseCommand):
def add_arguments(self, parser):
super(Command, self).add_arguments(parser)
parser.add_argument('--alias', action='store', dest='alias', default=DEFAULT_CHANNEL_LAYER,
help='Channel layer alias to use, if not the default.')
def handle(self, *args, **options): def handle(self, *args, **options):
# Get the backend to use # Get the backend to use
self.verbosity = options.get("verbosity", 1) self.verbosity = options.get("verbosity", 1)
self.logger = setup_logger('django.channels', self.verbosity) self.logger = setup_logger('django.channels', self.verbosity)
self.channel_layer = channel_layers[DEFAULT_CHANNEL_LAYER] self.channel_layer = channel_layers[options.get("alias", DEFAULT_CHANNEL_LAYER)]
# Check that handler isn't inmemory # Check that handler isn't inmemory
if self.channel_layer.local_only(): if self.channel_layer.local_only():
raise CommandError( raise CommandError(