mint/ai-translate/ai_translate/__init__.py

import logging
import tomllib

import pika

from .worker import handle_message

logger = logging.getLogger(__name__)


def start_consumer(name, queue, config):
    connection = pika.BlockingConnection(
        pika.ConnectionParameters(
            host=config['host'], port=config['port'],
            credentials=pika.PlainCredentials(
                config['user'], config['password']),
            virtual_host=config['virtual-host']))
    channel = connection.channel()

    def callback(ch, method, properties, body):
        logger.info("received message(%s,%s)",
                    properties.message_id, properties.content_type)
        handle_message(properties.message_id, properties.content_type, body)

    channel.basic_consume(
        queue=queue, on_message_callback=callback, auto_ack=True)

    logger.info('start a consumer(%s) for queue(%s)', name, queue)
    channel.start_consuming()


def launch(name, queue, config_file):
    logger.debug('load configuration from %s', config_file)
    with open(config_file, "rb") as config_fd:
        config = tomllib.load(config_fd)
        start_consumer(name, queue, config['rabbitmq'])