102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
import sys, ldap, termios
|
|
from getopt import getopt
|
|
from ceo import members, terms, uwldap
|
|
import ceo.ldapi as ldapi
|
|
|
|
shortopts = [
|
|
]
|
|
|
|
longopts = [
|
|
]
|
|
|
|
def start():
|
|
(opts, args) = getopt(sys.argv[1:], shortopts, longopts)
|
|
if len(args) >= 1:
|
|
if args[0] in commands:
|
|
commands[args[0]](args[1:])
|
|
else:
|
|
print "Invalid command '%s'" % args[0]
|
|
|
|
def help():
|
|
print 'Available commands:'
|
|
for c in commands:
|
|
print ' %s' % c
|
|
|
|
def memberlist(args):
|
|
mlist = members.list_term(terms.current())
|
|
dns = mlist.keys()
|
|
dns.sort()
|
|
for dn in dns:
|
|
member = mlist[dn]
|
|
print '%s %s %s' % (
|
|
member['uid'][0].ljust(12),
|
|
member['cn'][0].ljust(30),
|
|
member.get('program', [''])[0]
|
|
)
|
|
|
|
def updateprogram(args):
|
|
mlist = members.list_all().items()
|
|
uwl = ldap.initialize(uwldap.uri())
|
|
fd = sys.stdin.fileno()
|
|
for (dn, member) in mlist:
|
|
uid = member['uid'][0]
|
|
user = uwl.search_s(uwldap.base(), ldap.SCOPE_SUBTREE,
|
|
'(uid=%s)' % ldapi.escape(uid))
|
|
if len(user) == 0:
|
|
continue
|
|
user = user[0][1]
|
|
oldprog = member.get('program', [''])[0]
|
|
newprog = user.get('ou', [''])[0]
|
|
if oldprog == newprog:
|
|
continue
|
|
sys.stdout.write("%s: '%s' => '%s'? (y/n) " % (uid, oldprog, newprog))
|
|
new = old = termios.tcgetattr(fd)
|
|
new[3] = new[3] & ~termios.ICANON
|
|
try:
|
|
termios.tcsetattr(fd, termios.TCSANOW, new)
|
|
try:
|
|
if sys.stdin.read(1) != 'y':
|
|
continue
|
|
except KeyboardInterrupt:
|
|
return ''
|
|
finally:
|
|
print ''
|
|
termios.tcsetattr(fd, termios.TCSANOW, old)
|
|
old = new = {}
|
|
if oldprog != '':
|
|
old = {'program': [oldprog]}
|
|
if newprog != '':
|
|
new = {'program': [newprog]}
|
|
mlist = ldapi.make_modlist(old, new)
|
|
# TODO: don't use members.ld directly
|
|
members.ld.modify_s(dn, mlist)
|
|
|
|
def expiredaccounts(args):
|
|
send_email = False
|
|
if len(args) == 1 and args[0] == '--email':
|
|
sys.stderr.write("If you want to send an account expiration notice to " \
|
|
"these users then type 'Yes, do this' and hit enter\n")
|
|
if raw_input() == 'Yes, do this':
|
|
send_email = True
|
|
uwl = ldap.initialize(uwldap.uri())
|
|
mlist = members.expired_accounts()
|
|
for member in mlist.values():
|
|
uid = member['uid'][0]
|
|
name = member['cn'][0]
|
|
email = None
|
|
if send_email:
|
|
members.send_account_expired_email(name, uid)
|
|
user = uwl.search_s(uwldap.base(), ldap.SCOPE_SUBTREE,
|
|
'(uid=%s)' % ldapi.escape(uid))
|
|
if len(user) > 0 and 'mailLocalAddress' in user[0][1]:
|
|
email = user[0][1]['mailLocalAddress'][0]
|
|
members.send_account_expired_email(name, email)
|
|
print '%s %s' % (uid.ljust(12), name.ljust(30))
|
|
|
|
# list of commands
|
|
commands = {
|
|
'memberlist' : memberlist,
|
|
'updateprogram' : updateprogram,
|
|
'expiredaccounts' : expiredaccounts,
|
|
}
|