2
# IVLE - Informatics Virtual Learning Environment
3
# Copyright (C) 2007-2008 The University of Melbourne
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.
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.
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
19
# Program: RemakeAllUsers
23
# Script to rebuild jails for all users on the system.
24
# Requires root to run.
32
import common.makeuser
35
# Import modules from the website is tricky since they're in the www
37
sys.path.append(os.path.join(os.getcwd(), 'www'))
39
p = optparse.OptionParser()
40
p.add_option('--incremental', '-i', action='store_true')
41
options, arguments = p.parse_args()
44
print "Must run remakeallusers.py as root."
50
res = db.get_all('login', ['login', 'unixid'])
52
return (flds['login'], flds['unixid'])
53
uids = dict(map(repack,res))
55
except Exception, message:
56
print "Error: " + str(message)
59
# First check that our __templateuser__ directory exits, if not create it
60
templateuserdir = os.path.join(conf.jail_base, '__templateuser__')
61
stagingdir = templatedir = os.path.join(conf.jail_base, '__staging__')
62
if not os.path.isdir(templateuserdir):
63
os.mkdir(templateuserdir)
65
# Generate manifest of files that differ between template user and staging
66
if options.incremental:
67
print "Incremental Rebuild selected"
68
manifest = common.makeuser.generate_manifest(stagingdir, templateuserdir)
69
print "Generated change manifest"
70
print "Adding/Updating: %s\nRemoving: %s" % manifest
72
# Force a full rebuild
75
list.sort(key=lambda user: user.login)
80
# Resolve the user's login into a UID
81
# Create the user if it does not exist
85
raise Exception("User %s does not have a unixid in the database"
87
# Remake the user's jail
88
common.makeuser.make_jail(login, uid, manifest=manifest)
89
except Exception, message:
90
print "Error: " + str(message)
93
print "Successfully recreated user %s's jail." % login
95
# Update the template user directory
96
print "Updating templateuser directory"
97
shutil.rmtree(templateuserdir)
98
#common.makeuser.linktree(stagingdir, templateuserdir)
99
common.makeuser.make_jail('__templateuser__', 0, manifest=manifest)
100
shutil.rmtree(os.path.join(templateuserdir, 'home'))