"""
Script for ping of SQS message queue.

For load testing.
"""

import time

from ytownership import config
from ytownership.connectors import sqs

YOUTUBE_SQS_QUEUE_TEST = '{}-yt-ownership-2'.format(config.ENVIRONMENT)
sqs_queue = sqs.get_queue(YOUTUBE_SQS_QUEUE_TEST)


def ping_queue():
    """Ping queue state.

    Ping queue state every second. Return amount of messages
    in queue

    Returns: True if work done.
    """

    amount = sqs_queue.count()

    while amount > 0:
        print('SQS Queue contains {0} messages'.format(amount))
        time.sleep(1)
        amount = sqs_queue.count()
    return True


def get_message():
    """Pull single message from queue.
    """

    return sqs_queue.get_messages()

if __name__ == '__main__':
    print('Start working')
    while True:
        ping_queue()
    print('Finish working')