1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
#!/usr/bin/env python3
"""
Query LDAP directory for users and print it in a mutt query format.
"""
import argparse
import sys
import ldap3
def first(values):
if values is None or len(values) == 0:
return ""
return values[0]
def split_to_list(string):
return string.split(',')
def ldap_search(args):
search_filter = [f'({f}=*{args.term}*)' for f in args.search_filters]
search_filter = '(|' + ''.join(search_filter) + ')'
with ldap3.Connection(ldap3.Server(args.server, port=args.port),
auto_bind=ldap3.AUTO_BIND_NO_TLS,
read_only=True,
check_names=True,
user=args.username, password=args.password) as c:
results = c.extend.standard.paged_search(search_base=args.search_base,
search_filter=search_filter,
search_scope=ldap3.SUBTREE,
attributes=list(args.search_attributes),
get_operational_attributes=True,
generator=False)
count = len(results)
if count:
print(f'Found {count} entries:')
for r in results:
attrs = [first(r['attributes'][attr]) for attr in args.search_attributes[:2]]
attrs.append(' / '.join([first(r['attributes'][attr]) for attr in args.search_attributes[2:]]))
last = '\t'.join(attrs)
print(last)
else:
print('Not found.')
def main():
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('term', help="search term")
parser.add_argument('-s', '--server', required=True, help='server hostname')
parser.add_argument('-p', '--port', type=int, default=389, help='server port (default: %(default)s)')
parser.add_argument('-u', '--username', help='username')
parser.add_argument('-P', '--password', help='password')
parser.add_argument('-b', '--search-base', metavar="SB", default='ou=people', type=split_to_list,
help="search base string (default: %(default)s)")
parser.add_argument('-f', '--search-filters', metavar="FILTERS",
default='mail,cn', type=split_to_list,
help="list of fields to filter (default: %(default)s)")
parser.add_argument('-a', '--search-attributes', metavar="ATTRIBUTES",
default='mail,cn', type=split_to_list,
help="list of resulting attributes (default: %(default)s)")
args = parser.parse_args()
ldap_search(args)
if __name__ == '__main__':
try:
main()
except Exception as e:
print(e)
sys.exit(1)
|
