phi/src/phi/ldap/user.py

from phi.ldap.entry import get_entry_by_uid
from phi.ldap.utils import flatten_attributes


def user_attributes_mapping(client):
    return {
        client.attribute_id: "uid",
        client.attribute_mail: "mail",
        "createTimestamp": "created_at",
        "modifyTimestamp": "modified_at",
    }


def get_user_by_uid(client, uid):
    entry = get_entry_by_uid(client, uid)

    if not entry:
        return None

    mapping = user_attributes_mapping(client)

    user = {
        mapping[k]: v for k, v in entry["attributes"].items() if k in mapping.keys()
    }

    return flatten_attributes(user)