Authenticating unifi wireless guest users.

unifi-auth.py

import json
import ssl
import requests
from requests_toolbelt import SSLAdapter

# URL or IP Address of the unifi controller:
PORTAL_URL = 'https://ubiquiti.example.com/'

# Username and password for the unifi controller:
PORTAL_LOGIN_PARAMS = {
    'login': 'login',
    'username': 'admin',
    'password': 'hunter9',
}

def authorize_guest(client_mac, session_length):
    login_url = '{}login'.format(PORTAL_URL)
    logout_url = '{}logout'.format(PORTAL_URL)
    api_url = '{}api/cmd/stamgr'.format(PORTAL_URL)

    auth = {
        'cmd': 'authorize-guest',
        'mac': client_mac,
        'minutes': session_length,
    }

    auth_params = {
        'json': json.dumps(auth)
    }

    # Use a Session object to handle cookies.
    # Use an SSLAdapter to work around SSL handshake issues.
    session = requests.Session()
    session.mount(PORTAL_URL, SSLAdapter(ssl.PROTOCOL_TLSv1))

    # Log in to the API
    login_response = session.post(login_url, data=PORTAL_LOGIN_PARAMS, timeout=5)
    if login_response.status_code != 200:
        return False

    # Authorize the guest
    auth_response = session.post(api_url, data=auth_params, timeout=5)
    if auth_response.status_code != 200:
        return False

    # Log out from the API
    logout_response = session.get(logout_url, timeout=5)
    return True