← Back to branch summary

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

« back to all changes in this revision

Viewing changes to remakeallusers.py

  • Committer: wagrant
  • Date: 2008-07-21 00:03:16 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:914
remakeuser.py: Destroy!
remakeallusers.py: Rename to remakeuser.py. It now takes a username to
      rebuild, or -a to rebuild everyone.

Show diffs side-by-side

added added

removed removed

Lines of Context:
remakeallusers.py
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 all users on the system.
24
 
# Requires root to run.
25
 
 
26
 
import sys
27
 
import os
28
 
import common.db
29
 
import common.makeuser
30
 
import optparse
31
 
import logging
32
 
 
33
 
p = optparse.OptionParser()
34
 
p.add_option('--verbose', '-v', action='store_true')
35
 
options, arguments = p.parse_args()
36
 
 
37
 
level = logging.DEBUG if options.verbose else logging.WARNING
38
 
 
39
 
logging.basicConfig(format='%(asctime)s %(levelname)s %(message)s',
40
 
                        level=level)
41
 
 
42
 
 
43
 
if os.getuid() != 0:
44
 
    print >> sys.stderr, "%s must be run as root" % sys.argv[0]
45
 
    sys.exit(1)
46
 
 
47
 
try:
48
 
    db = common.db.DB()
49
 
    list = db.get_users()
50
 
    res = db.get_all('login', ['login', 'unixid'])
51
 
    def repack(flds):
52
 
        return (flds['login'], flds['unixid'])
53
 
    uids = dict(map(repack,res))
54
 
except Exception, message:
55
 
    logging.error(str(message))
56
 
    sys.exit(1)
57
 
 
58
 
logging.info("rebuild started")
59
 
 
60
 
list.sort(key=lambda user: user.login)
61
 
for user in list:
62
 
    login = user.login
63
 
 
64
 
    try:
65
 
        # Resolve the user's login into a UID
66
 
        try:
67
 
            uid = uids[login]
68
 
        except KeyError:
69
 
            raise Exception("user %s does not have a unixid in the database"
70
 
                % login)
71
 
        # Remake the user's jail
72
 
        common.makeuser.make_jail(login, uid)
73
 
    except Exception, message:
74
 
        logging.warning(str(message))
75
 
        continue
76
 
 
77
 
    logging.debug("recreated user %s's jail." % login)
78
 
    
79
 
logging.info("rebuild completed successfully")