← Back to branch summary

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

« back to all changes in this revision

Viewing changes to remakeuser.py

  • Committer: mattgiuca
  • Date: 2008-01-09 21:33:56 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:144
Trunk, and all subdirectories with Python files:
    Added to svn:ignore all *.pyc *.pyo, to avoid compiled files
    showing up in svn st / diff / commit lists.
    Added to svn:ignore trampoline/trampoline.

Show diffs side-by-side

added added

removed removed

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