2020-09-30 18:42:28 +02:00
|
|
|
from phi.ldap.entry import get_entry_by_uid, get_entries_by_ou
|
2017-12-21 13:44:54 +01:00
|
|
|
|
2020-09-30 18:42:28 +02:00
|
|
|
shown_keys = ['uid', 'mail', 'createTimestamp', 'modifyTimestamp']
|
2017-12-21 13:44:54 +01:00
|
|
|
|
2020-09-30 16:38:31 +02:00
|
|
|
|
2020-09-30 18:42:28 +02:00
|
|
|
def flatten_attribute(attr):
|
|
|
|
if isinstance(attr, list) and len(attr)==1:
|
|
|
|
return attr[0]
|
|
|
|
else:
|
|
|
|
return attr
|
|
|
|
|
|
|
|
|
|
|
|
def flatten_user(entry):
|
|
|
|
return {k: flatten_attribute(attr)
|
|
|
|
for k, attr in entry['attributes'].items()
|
|
|
|
if k in shown_keys}
|
|
|
|
|
|
|
|
|
|
|
|
def get_user_by_uid(client, uid):
|
2017-12-21 13:44:54 +01:00
|
|
|
entry = get_entry_by_uid(client, uid)
|
|
|
|
|
|
|
|
if not entry:
|
|
|
|
return None
|
|
|
|
|
2020-09-30 18:42:28 +02:00
|
|
|
return flatten_user(entry)
|
2017-12-21 13:44:54 +01:00
|
|
|
|
|
|
|
|
2020-09-30 18:42:28 +02:00
|
|
|
def get_users_by_ou(client, ou):
|
|
|
|
entries = get_entries_by_ou(client, ou)
|
|
|
|
users = [flatten_user(entry) for entry in entries]
|
|
|
|
|
|
|
|
return users
|