-
Notifications
You must be signed in to change notification settings - Fork 81
/
group.py
125 lines (100 loc) · 5 KB
/
group.py
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
# --- BEGIN COPYRIGHT BLOCK ---
# Copyright (C) 2016, William Brown <william at blackhats.net.au>
# All rights reserved.
#
# License: GPL (version 3 or any later version).
# See LICENSE for details.
# --- END COPYRIGHT BLOCK ---
from lib389.idm.group import Group, Groups, MUST_ATTRIBUTES
from lib389.cli_base import populate_attr_arguments, _generic_modify
from lib389.cli_idm import (
_generic_list,
_generic_get,
_generic_get_dn,
_generic_create,
_generic_delete,
_get_arg,
_get_attributes,
_warn,
)
SINGULAR = Group
MANY = Groups
RDN = 'cn'
# These are a generic specification, try not to tamper with them
def list(inst, basedn, log, args):
_generic_list(inst, basedn, log.getChild('_generic_list'), MANY, args)
def get(inst, basedn, log, args):
rdn = _get_arg( args.selector, msg="Enter %s to retrieve" % RDN)
_generic_get(inst, basedn, log.getChild('_generic_get'), MANY, rdn, args)
def get_dn(inst, basedn, log, args):
dn = _get_arg( args.dn, msg="Enter dn to retrieve")
_generic_get_dn(inst, basedn, log.getChild('_generic_get_dn'), MANY, dn, args)
def create(inst, basedn, log, args):
kwargs = _get_attributes(args, MUST_ATTRIBUTES)
_generic_create(inst, basedn, log.getChild('_generic_create'), MANY, kwargs, args)
def delete(inst, basedn, log, args, warn=True):
dn = _get_arg( args.dn , msg="Enter dn to delete")
if warn:
_warn(dn, msg="Deleting %s %s" % (SINGULAR.__name__, dn))
_generic_delete(inst, basedn, log.getChild('_generic_delete'), SINGULAR, dn, args)
def modify(inst, basedn, log, args, warn=True):
rdn = _get_arg( args.selector, msg="Enter %s to retrieve" % RDN)
_generic_modify(inst, basedn, log.getChild('_generic_modify'), MANY, rdn, args)
def members(inst, basedn, log, args):
cn = _get_arg( args.cn, msg="Enter %s of group" % RDN)
groups = MANY(inst, basedn)
group = groups.get(cn)
# Display members?
member_list = group.list_members()
if len(member_list) == 0:
log.info('No members to display')
else:
for m in member_list:
log.info('dn: %s' % m)
def add_member(inst, basedn, log, args):
cn = _get_arg( args.cn, msg="Enter %s of group to add member too" % RDN)
dn = _get_arg( args.dn, msg="Enter dn to add as member")
groups = MANY(inst, basedn)
group = groups.get(cn)
group.add_member(dn)
log.info('added member: %s' % dn)
def remove_member(inst, basedn, log, args):
cn = _get_arg( args.cn, msg="Enter %s of group to remove member from" % RDN)
dn = _get_arg( args.dn, msg="Enter dn to remove as member")
groups = MANY(inst, basedn)
group = groups.get(cn)
group.remove_member(dn)
log.info('removed member: %s' % dn)
def create_parser(subparsers):
group_parser = subparsers.add_parser('group', help='Manage groups')
subcommands = group_parser.add_subparsers(help='action')
list_parser = subcommands.add_parser('list', help='list')
list_parser.set_defaults(func=list)
get_parser = subcommands.add_parser('get', help='get')
get_parser.set_defaults(func=get)
get_parser.add_argument('selector', nargs='?', help='The term to search for')
get_dn_parser = subcommands.add_parser('get_dn', help='get_dn')
get_dn_parser.set_defaults(func=get_dn)
get_dn_parser.add_argument('dn', nargs='?', help='The dn to get')
create_parser = subcommands.add_parser('create', help='create')
create_parser.set_defaults(func=create)
populate_attr_arguments(create_parser, MUST_ATTRIBUTES)
delete_parser = subcommands.add_parser('delete', help='deletes the object')
delete_parser.set_defaults(func=delete)
delete_parser.add_argument('dn', nargs='?', help='The dn to delete')
modify_parser = subcommands.add_parser('modify', help='modify <add|delete|replace>:<attribute>:<value> ...')
modify_parser.set_defaults(func=modify)
modify_parser.add_argument('selector', nargs=1, help='The %s to modify' % RDN)
modify_parser.add_argument('changes', nargs='+', help="A list of changes to apply in format: <add|delete|replace>:<attribute>:<value>")
members_parser = subcommands.add_parser('members', help="List member dns of a group")
members_parser.set_defaults(func=members)
members_parser.add_argument('cn', nargs='?', help="cn of group to list members of")
add_member_parser = subcommands.add_parser('add_member', help="Add a member to a group")
add_member_parser.set_defaults(func=add_member)
add_member_parser.add_argument('cn', nargs='?', help="cn of group to add member to")
add_member_parser.add_argument('dn', nargs='?', help="dn of object to add to group as member")
remove_member_parser = subcommands.add_parser('remove_member', help="Remove a member from a group")
remove_member_parser.set_defaults(func=remove_member)
remove_member_parser.add_argument('cn', nargs='?', help="cn of group to remove member from")
remove_member_parser.add_argument('dn', nargs='?', help="dn of object to remove from group as member")
# vim: tabstop=4 expandtab shiftwidth=4 softtabstop=4