import pika

QUEUE_HOST = "appointment.lpaconsulting.fr"
REQUEST_DATA_QUEUE = 'REQUEST_DATA'
# REQUEST_DATA_QUEUE_TEST = 'REQUEST_DATA_TEST'
REQUEST_DATA_QUEUE_TEST = 'REQUEST_DATA_TEST'
TEST_QUEUE = 'TEST_QUEUE'
SHARED_OBJECT = 'SHARED_OBJECT'
REQUEST_DATA_QUEUE_DE = 'REQUEST_DATA_DE'
REQUEST_DATA_OBJECT = 'REQUEST_DATA_OBJECT'
REGISTER_QUEUE = 'REGISTER_QUEUE'
credentials = pika.PlainCredentials('appointment', 'ZyuhJZ2xEYWhElhpJjy7YEpZGZwNYJz2fHIu')


class CookiesPublisher:

    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=self.to_queue, durable=True)

    def publish_body(self, body: str):
        print("will push to queue {}".format(self.to_queue))
        self.channel.basic_publish(exchange='', routing_key=self.to_queue, body=body)

    def message_count(self):
        return self.channel.queue_declare(queue=self.to_queue, durable=True).method.message_count
        # return self.queue_method.method.message_count