mirror of
https://salsa.debian.org/freeipa-team/freeipa.git
synced 2024-12-28 18:01:23 -06:00
256 lines
8.5 KiB
Python
256 lines
8.5 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
|
|
from optparse import OptionParser
|
|
import ipa
|
|
import ipa.user
|
|
import ipa.ipaclient as ipaclient
|
|
import ipa.ipavalidate as ipavalidate
|
|
import ipa.config
|
|
|
|
import xmlrpclib
|
|
import kerberos
|
|
import ldap
|
|
import errno
|
|
|
|
def usage():
|
|
print "ipa-usermod [-c|--gecos STRING] [-d|--directory STRING] [-f|--firstname STRING] [-l|--lastname STRING] [-s|--shell STRING] [--add attribute=value] [--del attribute] [--set attribute=value] user"
|
|
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("-c", "--gecos", dest="gecos",
|
|
help="Set the GECOS field")
|
|
parser.add_option("-d", "--directory", dest="directory",
|
|
help="Set the User's home directory")
|
|
parser.add_option("-f", "--firstname", dest="gn",
|
|
help="User's first name")
|
|
parser.add_option("-l", "--lastname", dest="sn",
|
|
help="User's last name")
|
|
parser.add_option("-s", "--shell", dest="shell",
|
|
help="Set user's login shell to shell")
|
|
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("-M", "--mailAddress", dest="mail",
|
|
help="Set user's e-mail address")
|
|
parser.add_option("--usage", action="store_true",
|
|
help="Program usage")
|
|
|
|
args = ipa.config.init_config(sys.argv)
|
|
options, args = parser.parse_args(args)
|
|
|
|
return options, args
|
|
|
|
def main():
|
|
# The following fields are required
|
|
givenname = ""
|
|
lastname = ""
|
|
username = ""
|
|
mail = ""
|
|
gecos = ""
|
|
directory = ""
|
|
groups = ""
|
|
shell = ""
|
|
|
|
match = False
|
|
cont = False
|
|
|
|
options, args = parse_options()
|
|
|
|
if len(args) != 2:
|
|
usage()
|
|
|
|
username = args[1]
|
|
|
|
client = ipaclient.IPAClient()
|
|
try:
|
|
user = client.get_user_by_uid(username)
|
|
except ipa.ipaerror.exception_for(ipa.ipaerror.LDAP_NOT_FOUND):
|
|
print "User %s not found" % username
|
|
return 1
|
|
except ipa.ipaerror.IPAError, e:
|
|
print "%s" % e.message
|
|
return 1
|
|
except kerberos.GSSError, e:
|
|
print "Could not initialize GSSAPI: %s/%s" % (e[0][0][0], e[0][1][0])
|
|
return 1
|
|
|
|
# If any options are set we use just those. Otherwise ask for all of them.
|
|
if options.gn or options.sn or options.directory or options.gecos or options.mail or options.shell or options.addattr or options.delattr or options.setattr:
|
|
givenname = options.gn
|
|
lastname = options.sn
|
|
gecos = options.gecos
|
|
directory = options.directory
|
|
mail = options.mail
|
|
shell = options.shell
|
|
else:
|
|
if not options.gn:
|
|
while (cont != True):
|
|
givenname = raw_input("First name: [%s] " % user.getValue('givenname'))
|
|
if (ipavalidate.plain(givenname, notEmpty=False)):
|
|
print "Must be letters or '"
|
|
else:
|
|
cont = True
|
|
if len(givenname) < 1:
|
|
givenname = None
|
|
cont = True
|
|
else:
|
|
givenname = options.gn
|
|
if (ipavalidate.plain(givenname, notEmpty=True)):
|
|
print "First name must be letters or '"
|
|
return 1
|
|
|
|
cont = False
|
|
if not options.sn:
|
|
while (cont != True):
|
|
lastname = raw_input(" Last name: [%s] " % user.getValue('sn'))
|
|
if (ipavalidate.plain(lastname, notEmpty=False)):
|
|
print "Must be letters or '"
|
|
else:
|
|
cont = True
|
|
if len(lastname) < 1:
|
|
lastname = None
|
|
cont = True
|
|
else:
|
|
lastname = options.sn
|
|
if (ipavalidate.plain(lastname, notEmpty=True)):
|
|
print "Last name must be letters or '"
|
|
return 1
|
|
|
|
cont = False
|
|
if not options.mail:
|
|
while (cont != True):
|
|
mail = raw_input("E-mail addr: [%s]" % user.getValue('mail'))
|
|
if (ipavalidate.email(mail, notEmpty=False)):
|
|
print "Must include a user and domain name"
|
|
else:
|
|
cont = True
|
|
else:
|
|
mail = options.mail
|
|
if (ipavalidate.email(mail)):
|
|
print "E-mail must include a user and domain name"
|
|
return 1
|
|
|
|
# Ask the questions we don't normally force. We don't require answers
|
|
# for these.
|
|
cont = False
|
|
if not options.gecos:
|
|
while (cont != True):
|
|
gecos = raw_input("gecos: [%s] " % user.getValue('gecos'))
|
|
if (ipavalidate.plain(gecos, notEmpty=False)):
|
|
print "Must be letters, numbers, spaces or '"
|
|
else:
|
|
cont = True
|
|
|
|
cont = False
|
|
if not options.directory:
|
|
while (cont != True):
|
|
directory = raw_input("home directory: [%s] " % user.getValue('homeDirectory'))
|
|
if (ipavalidate.path(gecos, notEmpty=False)):
|
|
print "Must be letters, numbers, spaces or '"
|
|
else:
|
|
cont = True
|
|
cont = False
|
|
if not options.shell:
|
|
while (cont != True):
|
|
shell = raw_input("shell: [%s] " % user.getValue('loginshell'))
|
|
|
|
if len(shell) < 1:
|
|
shell = None
|
|
cont = True
|
|
cont = False
|
|
|
|
if givenname:
|
|
user.setValue('givenname', givenname)
|
|
if lastname:
|
|
user.setValue('sn', lastname)
|
|
if mail:
|
|
user.setValue('mail', mail)
|
|
|
|
if gecos:
|
|
user.setValue('gecos', gecos)
|
|
if directory:
|
|
user.setValue('homedirectory', directory)
|
|
if shell:
|
|
user.setValue('loginshell', shell)
|
|
|
|
if options.delattr:
|
|
for d in options.delattr:
|
|
# doesn't truly delete the attribute but does null out the value
|
|
user.setValue(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
|
|
user.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 = user.getValue(attr)
|
|
if cvalue:
|
|
if isinstance(cvalue,str):
|
|
cvalue = [cvalue]
|
|
value = cvalue + [value]
|
|
user.setValue(attr, value)
|
|
|
|
|
|
try:
|
|
client.update_user(user)
|
|
except xmlrpclib.Fault, fault:
|
|
if fault.faultCode == errno.ECONNREFUSED:
|
|
print "The IPA XML-RPC service is not responding."
|
|
else:
|
|
print fault.faultString
|
|
return 1
|
|
except kerberos.GSSError, e:
|
|
print "Could not initialize GSSAPI: %s/%s" % (e[0][0][0], e[0][1][0])
|
|
return 1
|
|
except xmlrpclib.ProtocolError, e:
|
|
print "Unable to connect to IPA server: %s" % (e.errmsg)
|
|
return 1
|
|
except ipa.ipaerror.IPAError, e:
|
|
print "%s" % (e.message)
|
|
return 1
|
|
|
|
print username + " successfully updated"
|
|
return 0
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|