← Back to branch summary

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

« back to all changes in this revision

Viewing changes to remakeallusers.py

  • Committer: mattgiuca
  • Date: 2007-12-06 22:38:46 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:9
Added prototypes directory. Added ajax file browser proof-of-concept demo.

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 os.path
30
 
import pwd
31
 
import conf
32
 
import common.makeuser
33
 
# Import modules from the website is tricky since they're in the www
34
 
# directory.
35
 
sys.path.append(os.path.join(os.getcwd(), 'www'))
36
 
 
37
 
if os.getuid() != 0:
38
 
    print "Must run remakeallusers.py as root."
39
 
    sys.exit()
40
 
 
41
 
try:
42
 
    db = common.db.DB()
43
 
    list = db.get_users()
44
 
    res = db.get_all('login', ['login', 'unixid'])
45
 
    def repack(flds):
46
 
        return (flds['login'], flds['unixid'])
47
 
    uids = dict(map(repack,res))
48
 
 
49
 
    return uids[login]
50
 
 
51
 
except Exception, message:
52
 
    print "Error: " + str(message)
53
 
    sys.exit(1)
54
 
 
55
 
list.sort(key=lambda user: user.login)
56
 
for user in list:
57
 
    login = user.login
58
 
 
59
 
    try:
60
 
        # Resolve the user's login into a UID
61
 
        # Create the user if it does not exist
62
 
        try:
63
 
            uid = uids[login]
64
 
        except KeyError:
65
 
            raise Exception("User %s does not have a unixid in the database"
66
 
                % login)
67
 
        # Remake the user's jail
68
 
        common.makeuser.make_jail(login, uid)
69
 
    except Exception, message:
70
 
        print "Error: " + str(message)
71
 
        continue
72
 
 
73
 
    print "Successfully recreated user %s's jail." % login