mirror of
https://salsa.debian.org/freeipa-team/freeipa.git
synced 2025-01-15 19:01:55 -06:00
237 lines
8.0 KiB
Python
237 lines
8.0 KiB
Python
#! /usr/bin/python -E
|
|
# Authors: Rob Crittenden <rcritten@redhat.com>
|
|
#
|
|
# Copyright (C) 2007 Red Hat
|
|
# see file 'COPYING' for use and warranty information
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License as
|
|
# published by the Free Software Foundation; version 2 only
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
#
|
|
import sys
|
|
try:
|
|
from optparse import OptionParser
|
|
import ipa
|
|
import ipa.group
|
|
import ipa.ipaclient as ipaclient
|
|
import ipa.config
|
|
import ipa.ipaerror
|
|
|
|
import xmlrpclib
|
|
import kerberos
|
|
import ldap
|
|
import errno
|
|
import socket
|
|
except ImportError:
|
|
print >> sys.stderr, """\
|
|
There was a problem importing one of the required Python modules. The
|
|
error was:
|
|
|
|
%s
|
|
""" % sys.exc_value
|
|
sys.exit(1)
|
|
|
|
def usage():
|
|
print "ipa-modgroup [-l|--list]"
|
|
print "ipa-modgroup [-a|--add] [-r|--remove] [-v|--verbose] user group"
|
|
print "ipa-modgroup [-g|--groupadd] [-e|--groupdel] [-v|--verbose] group group"
|
|
print "ipa-modgroup [-d|--desc description STRING] [--addattr attribute=value] [--delattr attribute] [--setattr attribute=value] [-v|--verbose] group"
|
|
sys.exit(1)
|
|
|
|
def set_add_usage(which):
|
|
print "%s option usage: --%s NAME=VALUE" % (which, which)
|
|
|
|
def parse_options():
|
|
parser = OptionParser()
|
|
parser.add_option("-a", "--add", dest="add", action="store_true",
|
|
help="Add a user to the group")
|
|
parser.add_option("-r", "--remove", dest="remove", action="store_true",
|
|
help="Remove a user from the group")
|
|
parser.add_option("-g", "--groupadd", dest="groupadd", action="store_true",
|
|
help="Add a group to the group")
|
|
parser.add_option("-e", "--groupdel", dest="groupdel", action="store_true",
|
|
help="Remove a group from the group")
|
|
parser.add_option("-d", "--description", dest="desc",
|
|
help="Modify the description of the group")
|
|
parser.add_option("--addattr", dest="addattr",
|
|
help="Adds an attribute or values to that attribute, attr=value",
|
|
action="append")
|
|
parser.add_option("--delattr", dest="delattr",
|
|
help="Remove an attribute", action="append")
|
|
parser.add_option("--setattr", dest="setattr",
|
|
help="Set an attribute, dropping any existing values that may exist",
|
|
action="append")
|
|
parser.add_option("-l", "--list", dest="list", action="store_true",
|
|
help="List common attributes (this is not an exhaustive list)")
|
|
parser.add_option("--usage", action="store_true",
|
|
help="Program usage")
|
|
parser.add_option("-v", "--verbose", action="store_true", dest="verbose",
|
|
help="Verbose output of the XML-RPC connection")
|
|
|
|
args = ipa.config.init_config(sys.argv)
|
|
options, args = parser.parse_args(args)
|
|
|
|
if ((not options.add and not options.remove) and
|
|
(not options.groupadd and not options.groupdel) and
|
|
(not options.desc and not options.addattr and
|
|
not options.delattr and not options.setattr and not options.list)):
|
|
usage()
|
|
|
|
return options, args
|
|
|
|
def get_group(client, options, group_cn):
|
|
try:
|
|
attrs = ['*']
|
|
|
|
# in case any attributes being modified are operational such as
|
|
# nsaccountlock. Any attribute to be deleted needs to be included
|
|
# in the original record so it can be seen as being removed.
|
|
if options.delattr:
|
|
for d in options.delattr:
|
|
attrs.append(d)
|
|
group = client.get_entry_by_cn(group_cn, sattrs=attrs)
|
|
|
|
except ipa.ipaerror.IPAError, e:
|
|
print "%s" % e.message
|
|
return None
|
|
|
|
return group
|
|
|
|
def main():
|
|
group=ipa.group.Group()
|
|
options, args = parse_options()
|
|
|
|
if options.list:
|
|
client = ipaclient.IPAClient()
|
|
list = client.get_all_attrs()
|
|
|
|
for x in list:
|
|
print x
|
|
return 0
|
|
|
|
if options.usage:
|
|
usage()
|
|
if (options.add or options.remove) and (len(args) != 3):
|
|
usage()
|
|
elif ((options.desc or options.addattr or options.delattr or options.setattr) and (len(args) != 2)):
|
|
usage()
|
|
|
|
client = ipaclient.IPAClient()
|
|
if options.add:
|
|
group = get_group(client, options, args[2])
|
|
if group is None:
|
|
return 1
|
|
users = args[1].split(',')
|
|
for user in users:
|
|
client.add_user_to_group(user, group.dn)
|
|
print user + " successfully added to " + args[2]
|
|
elif options.remove:
|
|
group = get_group(client, options, args[2])
|
|
if group is None:
|
|
return 1
|
|
users = args[1].split(',')
|
|
for user in users:
|
|
client.remove_user_from_group(user, group.dn)
|
|
print user + " successfully removed"
|
|
elif options.groupadd:
|
|
group = get_group(client, options, args[2])
|
|
if group is None:
|
|
return 1
|
|
groups = args[1].split(',')
|
|
for g in groups:
|
|
tgroup = get_group(client, options, g)
|
|
if tgroup is not None:
|
|
client.add_group_to_group(tgroup.dn, group.dn)
|
|
print g + " successfully added to " + args[2]
|
|
else:
|
|
print "Group %s not found" % g
|
|
elif options.groupdel:
|
|
group = get_group(client, options, args[2])
|
|
if group is None:
|
|
return 1
|
|
groups = args[1].split(',')
|
|
for g in groups:
|
|
tgroup = get_group(client, options, g)
|
|
if tgroup is not None:
|
|
client.remove_member_from_group(tgroup.dn, group.dn)
|
|
print g + " successfully removed " + args[2]
|
|
else:
|
|
print "Group %s not found" % g
|
|
else:
|
|
group = get_group(client, options, args[1])
|
|
if group is None:
|
|
return 1
|
|
|
|
if options.desc:
|
|
group.setValue('description', options.desc)
|
|
|
|
if options.delattr:
|
|
for d in options.delattr:
|
|
group.delValue(d)
|
|
|
|
if options.setattr:
|
|
for s in options.setattr:
|
|
s = s.split('=')
|
|
if len(s) != 2:
|
|
set_add_usage("set")
|
|
sys.exit(1)
|
|
(attr,value) = s
|
|
group.setValue(attr, value)
|
|
|
|
if options.addattr:
|
|
for a in options.addattr:
|
|
a = a.split('=')
|
|
if len(a) != 2:
|
|
set_add_usage("add")
|
|
sys.exit(1)
|
|
(attr,value) = a
|
|
cvalue = group.getValue(attr)
|
|
if cvalue:
|
|
if isinstance(cvalue,str):
|
|
cvalue = [cvalue]
|
|
value = cvalue + [value]
|
|
group.setValue(attr, value)
|
|
|
|
client.update_group(group)
|
|
print args[1] + " successfully updated"
|
|
|
|
return 0
|
|
|
|
try:
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|
|
except SystemExit, e:
|
|
sys.exit(e)
|
|
except KeyboardInterrupt, e:
|
|
sys.exit(1)
|
|
except xmlrpclib.Fault, fault:
|
|
if fault.faultCode == errno.ECONNREFUSED:
|
|
print "The IPA XML-RPC service is not responding."
|
|
else:
|
|
print fault.faultString
|
|
sys.exit(1)
|
|
except kerberos.GSSError, e:
|
|
print "Could not initialize GSSAPI: %s/%s" % (ipautil.get_gsserror(e))
|
|
sys.exit(1)
|
|
except xmlrpclib.ProtocolError, e:
|
|
print "Unable to connect to IPA server: %s" % (e.errmsg)
|
|
sys.exit(1)
|
|
except ipa.ipaerror.IPAError, e:
|
|
print "%s" % (e.message)
|
|
sys.exit(1)
|
|
except socket.error, e:
|
|
print e[1]
|
|
print "Re-run with -v flag for more details."
|
|
except Exception, e:
|
|
print "%s" % str(e)
|
|
sys.exit(1)
|