From 482368a23934f40a2270da504b5cd024717d8e19 Mon Sep 17 00:00:00 2001 From: Mark Murnane Date: Tue, 1 Aug 2023 23:11:57 -0400 Subject: [PATCH] Munging amqp ssl support --- uber/tasks/__init__.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/uber/tasks/__init__.py b/uber/tasks/__init__.py index 6a4065bef..880754e8c 100644 --- a/uber/tasks/__init__.py +++ b/uber/tasks/__init__.py @@ -12,8 +12,14 @@ celery.conf.beat_schedule = {} celery.conf.beat_startup_tasks = [] celery.conf.update(config_dict['celery']) -celery.conf.update(broker_url=config_dict['secret']['broker_url']) -celery.conf.update(result_backend=config_dict['secret']['broker_url'].replace("amqp://", "rpc://")) + +broker_url = config_dict['secret']['broker_url'] +if broker_url.startswith("amqps://"): + celery.conf.update(broker_use_ssl=True) + broker_url = broker_url.replace("amqps://", "amqp://") + +celery.conf.update(broker_url=broker_url) +celery.conf.update(result_backend=broker_url.replace("amqp://", "rpc://")) celery.conf.update(task_ignore_result=True) def celery_on_startup(fn, *args, **kwargs):