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

418 by mattgiuca
Renamed trunk/console to trunk/scripts. We are now able to put more scripts in
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: RemakeUser
20
# Author:  Matt Giuca
21
# Date:    5/2/2008
22
23
# Repairs a user's jail, without actually writing to the DB.
24
# (This asks for less input than makeuser).
25
# This script wraps common.makeuser.
26
27
import sys
28
import os
29
import os.path
30
import pwd
31
# Import modules from the website is tricky since they're in the www
32
# directory.
33
sys.path.append(os.path.join(os.getcwd(), 'www'))
34
import conf
35
import common.makeuser
636 by drtomc
remakeuser: use the database, not the password file.
36
import common.db
418 by mattgiuca
Renamed trunk/console to trunk/scripts. We are now able to put more scripts in
37
38
if len(sys.argv) <= 1:
636 by drtomc
remakeuser: use the database, not the password file.
39
    print "Usage: python remakeuser.py <login>"
418 by mattgiuca
Renamed trunk/console to trunk/scripts. We are now able to put more scripts in
40
    sys.exit()
41
42
if os.getuid() != 0:
43
    print "Must run remakeuser.py as root."
44
    sys.exit()
45
636 by drtomc
remakeuser: use the database, not the password file.
46
login = sys.argv[1]
418 by mattgiuca
Renamed trunk/console to trunk/scripts. We are now able to put more scripts in
47
48
try:
636 by drtomc
remakeuser: use the database, not the password file.
49
    # Resolve the user's login into a UID
418 by mattgiuca
Renamed trunk/console to trunk/scripts. We are now able to put more scripts in
50
    # Create the user if it does not exist
51
    try:
636 by drtomc
remakeuser: use the database, not the password file.
52
        conn = common.db.DB()
53
        uid = conn.get_single({'login':login}, 'login', ['unixid'], ['login'])
418 by mattgiuca
Renamed trunk/console to trunk/scripts. We are now able to put more scripts in
54
    except KeyError:
636 by drtomc
remakeuser: use the database, not the password file.
55
        raise Exception("User does not have a unixid in the database")
418 by mattgiuca
Renamed trunk/console to trunk/scripts. We are now able to put more scripts in
56
    # Remake the user's jail
636 by drtomc
remakeuser: use the database, not the password file.
57
    common.makeuser.make_jail(login, uid)
418 by mattgiuca
Renamed trunk/console to trunk/scripts. We are now able to put more scripts in
58
except Exception, message:
59
    print "Error: " + str(message)
60
    sys.exit(1)
61
636 by drtomc
remakeuser: use the database, not the password file.
62
print "Successfully recreated user %s's jail." % login