import network
import time
from umqtt.simple import MQTTClient

WiFi_SSID = 'Wokwi-GUEST'
WiFi_PASS = ''

MQTT_CLIENT_ID = 'kiee-007'
MQTT_BROKER    = 'broker.hivemq.com'
MQTT_USER      = ''
MQTT_PASS      = ''

def on_message(topic, msg):
    incoming_message = msg.decode('utf8')
    print('{}: {}'.format(topic, incoming_message))

wlan = network.WLAN(network.STA_IF)
wlan.active(True)

wlan.connect(WiFi_SSID, WiFi_PASS)
print('WiFi ', end="")
while not wlan.isconnected():
    print('.', end='')
    time.sleep(0.5)
print(' ', wlan.ifconfig()[0], '[Connected]')

client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER,
                    user=MQTT_USER, password=MQTT_PASS)
print('MQTT ... ', MQTT_BROKER, end='')
try:
    client.connect()
    print(' [Connected]')
except:
    print(' [Error]')

client.set_callback(on_message)
client.subscribe('ae_iot/#')

while True:
    client.check_msg()