import os
import wifi

import adafruit_connection_manager
import adafruit_minimqtt.adafruit_minimqtt as MQTT
from socket_logger import SocketPoolLogger 


radio = wifi.radio
pool = adafruit_connection_manager.get_radio_socketpool(radio)

# these lines are new:
pool_logger = SocketPoolLogger(pool)
pool_logger.enable_log_all(True)

# and here we pass in the poll_logger, not the regular pool:
client = MQTT.MQTT(
    broker="io.adafruit.com",
    username=os.getenv("AIO_USERNAME"),
    password=os.getenv("AIO_KEY"),
    socket_pool=pool_logger,
)
result = client.connect()
client.loop(5)
client.disconnect()