import json

import switch_client
import todo
from extractors.xe_page import fetch_tables

PATH = "/LLDPStats.html"


def get_lldp_statistics(session) -> dict:
    """Returns {"global": record, "interfaces": list[dict]}."""
    tables = fetch_tables(session, PATH)
    return {
        "global": tables["LLDP Global Statistics"]["record"],
        "interfaces": tables["LLDP Interface Statistics"]["rows"],
    }


if __name__ == "__main__":
    session = switch_client.get_session()
    print(json.dumps(get_lldp_statistics(session), indent=2))
    todo.mark_implemented(switch_client.BASE_URL + PATH)