Send SQS messages in batch

sqs_send_batch.py

class BatchSendToSQS:
    """
    >>> with BatchSendToSQS(someQueue, batch_size=10) as sqs_batcher:
    >>>     sqs_batcher.add_msg({'data': 'booba'})
    >>>     ...
    >>>     sqs_batcher.add_msg({'data': 'piu piu'})
    """

    def __init__(self, queue: "boto3.sqs.Queue", batch_size: int):
        self.queue = queue
        self.messages_to_send = []
        self.batch_size = batch_size

    def add_msg(self, msg):
        self.messages_to_send.append(msg)
        if len(self.messages_to_send) == self.batch_size:
            self.queue.send_messages(Entries=self.messages_to_send)
            self.messages_to_send.clear()

    def __enter__(self):
        return self

    def __exit__(self, *exc):
        # send last batch of messages
        if self.messages_to_send:
            self.queue.send_messages(Entries=self.messages_to_send)