~azzar1/unity/add-show-desktop-key

1079 by William Grant
Merge setup-refactor branch. This completely breaks existing installations;
1
#!/usr/bin/env python
2
# IVLE - Informatics Virtual Learning Environment
3
# Copyright (C) 2007-2008 The University of Melbourne
4
#
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
9
#
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
# GNU General Public License for more details.
14
#
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
19
# Program: RemakeAllUsers
20
# Author:  Matt Giuca
21
# Date:    3/3/2008
22
23
# Script to rebuild jails for users on the system.
24
# One can specify a single username, or that all should be rebuilt.
25
# Requires root to run.
26
27
import sys
28
import os
29
import optparse
30
import logging
31
32
import ivle.db
33
import ivle.makeuser
34
35
p = optparse.OptionParser()
36
p.add_option('--verbose', '-v', action='store_true')
37
p.add_option('--all', '-a', action='store_true')
38
options, arguments = p.parse_args()
39
40
level = logging.DEBUG if options.verbose else logging.WARNING
41
42
logging.basicConfig(format='%(asctime)s %(levelname)s %(message)s',
43
                        level=level)
44
45
46
if os.getuid() != 0:
47
    print >> sys.stderr, "%s must be run as root" % \
48
                          os.path.basename(sys.argv[0])
49
    sys.exit(1)
50
51
try:
52
    db = ivle.db.DB()
53
    if options.all:
54
        list = db.get_users()
55
    else:
56
        if len(arguments) == 0:
57
            print >> sys.stderr, "must be run with -a or a username"
58
            sys.exit(1)
59
        list = [db.get_user(arguments[0])]
60
    res = db.get_all('login', ['login', 'unixid'])
61
    def repack(flds):
62
        return (flds['login'], flds['unixid'])
63
    uids = dict(map(repack,res))
64
except Exception, message:
65
    logging.error(str(message))
66
    sys.exit(1)
67
68
logging.info("rebuild started")
69
70
list.sort(key=lambda user: user.login)
71
for user in list:
72
    login = user.login
73
74
    try:
75
        # Resolve the user's login into a UID
76
        try:
77
            uid = uids[login]
78
        except KeyError:
79
            raise Exception("user %s does not have a unixid in the database"
80
                % login)
81
        # Remake the user's jail
82
        ivle.makeuser.make_jail(login, uid)
83
    except Exception, message:
84
        logging.warning(str(message))
85
        continue
86
87
    logging.debug("recreated user %s's jail." % login)
88
    
89
logging.info("rebuild completed successfully")