purge_queue: Use more robust Tornado patterns.

We're about to make the list of possible Tornado queue names broader.
This commit is contained in:
Tim Abbott 2018-11-02 16:05:13 -07:00
parent 9339c191da
commit 0976e3bd39

View File

@ -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,))