#!/usr/bin/env python3
from __future__ import annotations

import socket
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent))
from mk_apply_hotspot import HOST, PORT, login, talk, ok, trap_msg  # noqa: E402

name = "699000001"
pin = "0001"
sock = socket.create_connection((HOST, PORT), timeout=8)
login(sock)
r = talk(sock, ["/ip/hotspot/user/print", "?name=" + name])
ids = [w[5:] for s in r for w in s if w.startswith("=.id=")]
for i in ids:
    talk(sock, ["/ip/hotspot/user/remove", "=numbers=" + i])
r = talk(
    sock,
    [
        "/ip/hotspot/user/add",
        "=name=" + name,
        "=password=" + pin,
        "=profile=TEST-1H",
        "=limit-uptime=1h",
        "=comment=Woyla smoke",
    ],
)
print("ADD", "OK" if ok(r) else trap_msg(r))
r = talk(sock, ["/ip/hotspot/user/print", "?name=" + name])
print("USER", r)
sock.close()