From 0976e3bd3970f6dd5f0790aeda65ffb68a30e1de Mon Sep 17 00:00:00 2001 From: Tim Abbott Date: Fri, 2 Nov 2018 16:05:13 -0700 Subject: [PATCH] purge_queue: Use more robust Tornado patterns. We're about to make the list of possible Tornado queue names broader. --- zerver/management/commands/purge_queue.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/zerver/management/commands/purge_queue.py b/zerver/management/commands/purge_queue.py index bf8b97c9e0..582953e640 100644 --- a/zerver/management/commands/purge_queue.py +++ b/zerver/management/commands/purge_queue.py @@ -31,8 +31,9 @@ class Command(BaseCommand): raise CommandError("Missing queue_name argument!") else: queue_name = options['queue_name'] - if queue_name not in ['notify_tornado', 'tornado_return', - ] + get_active_worker_queues(): + if not (queue_name in get_active_worker_queues() or + queue_name.startswith("notify_tornado") or + queue_name.startswith("tornado_return")): raise CommandError("Unknown queue %s" % (queue_name,)) print("Purging queue %s" % (queue_name,))