From 5872526ccf3a572818ac7e69b6764782c8189b39 Mon Sep 17 00:00:00 2001 From: Tom Szendrey Date: Tue, 15 Aug 2023 13:58:19 -0400 Subject: [PATCH] first pass at adding username + password for MQ --- service/settings.py | 2 ++ service/utils/rabbitmq.py | 8 +++++++- 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/service/settings.py b/service/settings.py index 5d994a6..252ce72 100644 --- a/service/settings.py +++ b/service/settings.py @@ -15,6 +15,8 @@ class Settings(BaseSettings): REDIS_PORT: int = 6379 RABBITMQ_HOST: str = "rabbitmq.pyciemss" RABBITMQ_PORT: int = 5672 + RABBITMQ_USERNAME: str = "terarium" + RABBITMQ_PASSWORD: str = "terarium123" settings = Settings() diff --git a/service/utils/rabbitmq.py b/service/utils/rabbitmq.py index 5251642..117de91 100644 --- a/service/utils/rabbitmq.py +++ b/service/utils/rabbitmq.py @@ -1,4 +1,5 @@ import pika +from pika.credentials import PlainCredentials import json import time import logging @@ -33,7 +34,12 @@ def callback(ch, method, properties, body): def gen_rabbitmq_hook(job_id): - connection = pika.BlockingConnection(conn_config) + connection = pika.BlockingConnection( + conn_config, + credentials=PlainCredentials( + username=settings.RABBITMQ_USERNAME, password=settings.RABBITMQ_PASSWORD + ), + ) channel = connection.channel() def hook(progress):