AirWave：无线控制器和AP相关数据获取

import requests
import os
import urllib.parse
import xmltodict


class AirWaveAPI(object):
    """Aruba networks AirWave API client.
    Attributes:
        :username (str): AirWave Login username.
        :password (str): AirWave Login password.
        :url (str): AirWave URL.
        :session (requests.sessions.Session): Session for connection pooling.
    """

    def __init__(self, username, password, url, **kwargs):
        """
        :param username:
        :param password:
        :param url:
        :param kwargs:
        """
        self.username = username
        self.password = password
        self.url = url
        self.session = None

    def login(self):
        """Login to AirWave"""
        self.session = requests.Session()
        url = os.path.join(self.url, 'LOGIN').replace('\\', '/')
        destination = '/'
        next_action = ''
        params = {
            'credential_0': self.username,
            'credential_1': self.password,
            'login': 'Log In',
            'destination': destination,
            'next_action': next_action
        }
        return self.session.post(url, params=params, verify=False)

    def logout(self):
        self.session.close()

    def api_path(self, path):
        """
        :param path: Path for API URL
        :return: URL string 'https://xxx.xxx.xxx.xxx/xxxxxx'
        """
        return os.path.join(self.url, path)

    def ap_list(self, ap_ids=None):
        """
        :param ap_ids: (optional[list]): You may specify multiple
                        Access Point IDs. Default is None
        :return:
        """
        url = self.api_path('ap_list.xml')
        if ap_ids:
            params = self.id_params(ap_ids)
            return self.session.get(url, verify=False, params=params)
        return self.session.get(url, verify=False)

    def ap_detail(self, ap_id):
        """Get Access Point detail information.
        Args:
            :ap_id (int): Access Point ID.
        Returns:
            :Response: requests.models.Response.
        Usage: ::
            # >>> res = airwave.ap_detail(123)
            # >>> res.status_code
            # 200
            # >>> res.url
            # 'https://192.168.1.1/ap_detail.xml?id=123'
            # >>> res.text  # xml output.
            # '<?xml version="1.0" encoding="utf-8" ...'
        """
        url = self.api_path('ap_detail.xml')
        params = {'id': ap_id}
        params = self.urlencode(params)
        return self.session.get(url, verify=False, params=params)

    def client_detail(self, mac):
        """Client detail information.
        Args:
            :mac (str): Client device's MAC address.
        Returns:
            :Response: requests.models.Response.
        Usage: ::
            # >>> res = airwave.client_detail('12:34:56:78:90:AB')
            # >>> res.status_code
            # 200
            # >>> res.url
            # 'https://192.168.1.1/client_detail.xml?mac=12%3A34%3A56%3A78%3A90%3AAB'
            # >>> res.text  # xml output.
            # '<?xml version="1.0" encoding="utf-8" ...'
        """
        url = self.api_path('client_detail.xml')
        params = {'mac': mac}
        params = self.urlencode(params)
        return self.session.get(url, verify=False, params=params)

    def rogue_detail(self, ap_id):
        """Rogue detail information.
        Args:
            :ap_id (int): Access Point ID.
        Returns:
            :Response: requests.models.Response.
        Usage: ::
            # >>> res = airwave.rogue_detail(123)
            # >>> res.status_code
            # 200
            # >>> res.text  # xml output.
            # '<?xml version="1.0" encoding="utf-8" ...'
        """
        url = self.api_path('rogue_detail.xml')
        params = {'id': ap_id}
        params = self.urlencode(params)
        return self.session.get(url, verify=False, params=params)

    def latest_report(self, report_definition_id):
        """Latest report information.
        Args:
            :report_definition_id (int): Report definition ID.
                Please get it from "https://x.x.x.x/reports_definition".
        Returns:
            :Response: requests.models.Response.
        Usage: ::
            # >>> res = airwave.latest_report(123)
            # >>> res.status_code
            # 200
            # >>> res.url
            # 'https://192.1681.1/latest_report.xml?id=123'
            # >>> res.text
            # '<?xml version="1.0" encoding="utf-8" ...'
        """
        url = self.api_path('latest_report.xml')
        params = {'id': report_definition_id}
        params = self.urlencode(params)
        return self.session.get(url, verify=False, params=params)

    def client_search(self, username):
        url = self.api_path('client_search.xml')
        params = {'query': username}
        params = self.urlencode(params)
        return self.session.get(url, verify=False, params=params)

    @staticmethod
    def id_params(ap_ids):
        """Make access point id string."""
        return '&'.join(["id=%s" % ap_id for ap_id in ap_ids])

    @staticmethod
    def urlencode(params):
        """URL Encode."""
        params = sorted(params.items())
        return urllib.parse.urlencode(params)


if __name__ == '__main__':
    url = 'https://10.10.10.10/'
    aw = AirWaveAPI(username='username', password='password', url=url)
    aw.login()
    res = aw.ap_list()
    aw.logout()
    if res.status_code != 200:
        exit()
    ap_list = xmltodict.parse(res.text).get('amp:amp_ap_list').get('ap')
    for ap in ap_list:
        print(ap)
        break