first pass at adding username + password for MQ

DARPA-ASKEM · Aug 15, 2023 · 5872526 · 5872526
1 parent 6cfe408
commit 5872526
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git 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
@@ -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):