from datetime import datetime
from asyncio import get_event_loop, sleep
from azure.identity.aio import DefaultAzureCredential
from azure.servicebus.aio import ServiceBusClient
from azure.servicebus import ServiceBusReceiveMode

import logging

logging.basicConfig(level=logging.DEBUG, format="%(asctime)-15s %(name)s [%(levelname)s] %(message)s")

SB_NAMESPACE_FQDN = "..."
QUEUE_NAME = "..."


async def receive_forever():
    attempt = 0
    while True:
        start = datetime.now()
        try:
            attempt += 1
            logging.info(f"== Starting attempt={attempt}")

            async with DefaultAzureCredential(exclude_environment_credential=True) as credential:
                service_bus_client = ServiceBusClient(SB_NAMESPACE_FQDN, credential, logging_enable=True)
                async with service_bus_client:
                    queue_receiver = service_bus_client.get_queue_receiver(
                        QUEUE_NAME,
                        receive_mode=ServiceBusReceiveMode.RECEIVE_AND_DELETE,
                    )
                    async with queue_receiver:
                        while True:
                            messages = await queue_receiver.receive_messages(max_message_count=1, max_wait_time=30)
                            if messages:
                                logging.debug(f"Received: {messages[0]}")
        except Exception:
            logging.exception(f"== Running for {datetime.now()-start}")
            await sleep(10)


if __name__ == "__main__":
    loop = get_event_loop()
    loop.run_until_complete(receive_forever())