with

prefect.context.get("logger")

__init__

@resource_manager
class AzureEventHubProducerClient:
    """Build an Eventhub producer client"""

    def __init__(self, conn_str: str, eventhub_name: str) -> None:
        self.conn_str = conn_str
        self.eventhub_name = eventhub_name

    def setup(self) -> EventHubProducerClient:
        logger = prefect.context.get("logger")
        <http://logger.info|logger.info>(f"Connecting to {self.eventhub_name}")
        return EventHubProducerClient.from_connection_string(
            self.conn_str, eventhub_name=self.eventhub_name, retry_total=1
        )

    def cleanup(self, client: EventHubProducerClient) -> None:
        logger = prefect.context.get("logger")
        <http://logger.info|logger.info>("Closing Azure Eventhub Producer client")
        client.close()

# Set up clients
    with Oauth2_HTTPClient(
        base_url=coldstream_url,
        token_url=token_url,
        client_id=client_id,
        client_secret=client_secret,
        scope=scope,
    ) as http_client, AzureEventHubProducerClient(
        conn_str=eventhub_conn_str, eventhub_name=eventhub_name
    ) as eventhub_client, PostgresEngine(
        timescale_conn_str
    ) as timescale:
        all_gaps = fetch_gaps(
            engine=timescale,
            start_time=start,
            end_time=end,
            min_gap_duration=min_gap,
        )

resource_managers