add dvm and hc as variable

queue_message/CookiesPublisher.py

@@ -2,25 +2,27 @@ import pika
 
 QUEUE_HOST = "appointment.lpaconsulting.fr"
 REQUEST_DATA_QUEUE = 'REQUEST_DATA'
+REQUEST_DATA_QUEUE_TEST = 'REQUEST_DATA_TEST'
 credentials = pika.PlainCredentials('appointment', 'ZyuhJZ2xEYWhElhpJjy7YEpZGZwNYJz2fHIu')
 
 
 class CookiesPublisher:
 
-    def __init__(self):
+    def __init__(self, queue_name=REQUEST_DATA_QUEUE):
         self.connection = None
         self.channel = None
         self.queue_method = None
+        self.to_queue = queue_name
 
     def set_up_connection(self):
         self.connection = pika.BlockingConnection(
             pika.ConnectionParameters(host=QUEUE_HOST, port=5672, credentials=credentials))
         self.channel = self.connection.channel()
-        self.queue_method = self.channel.queue_declare(queue=REQUEST_DATA_QUEUE, durable=True)
+        self.queue_method = self.channel.queue_declare(queue=self.to_queue, durable=True)
 
     def publish_body(self, body: str):
-        self.channel.basic_publish(exchange='', routing_key=REQUEST_DATA_QUEUE, body=body)
+        self.channel.basic_publish(exchange='', routing_key=self.to_queue, body=body)
 
     def message_count(self):
-        return self.channel.queue_declare(queue=REQUEST_DATA_QUEUE, durable=True).method.message_count
+        return self.channel.queue_declare(queue=self.to_queue, durable=True).method.message_count
         # return self.queue_method.method.message_count