← Back to branch summary

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

« back to all changes in this revision

Viewing changes to www/common/makeuser.py

  • Committer: stevenbird
  • Date: 2008-02-04 04:22:44 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:400
first commit of homepage at ivle.sourceforge.net, including screenshots; just web-presence at this stage

Show diffs side-by-side

added added

removed removed

Lines of Context:
www/common/makeuser.py
 
1
# IVLE - Informatics Virtual Learning Environment
 
2
# Copyright (C) 2007-2008 The University of Melbourne
 
3
#
 
4
# This program is free software; you can redistribute it and/or modify
 
5
# it under the terms of the GNU General Public License as published by
 
6
# the Free Software Foundation; either version 2 of the License, or
 
7
# (at your option) any later version.
 
8
#
 
9
# This program is distributed in the hope that it will be useful,
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
# GNU General Public License for more details.
 
13
#
 
14
# You should have received a copy of the GNU General Public License
 
15
# along with this program; if not, write to the Free Software
 
16
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
17
 
 
18
# Module: MakeUser
 
19
# Author: Matt Giuca
 
20
# Date:   1/2/2008
 
21
 
 
22
# Allows creation of users. This sets up the following:
 
23
# * User's jail and home directory within the jail.
 
24
# * Subversion repository (TODO)
 
25
# * Check out Subversion workspace into jail (TODO)
 
26
# * Database details for user
 
27
# * Unix user account
 
28
 
 
29
# TODO: Sanitize login name and other fields.
 
30
 
 
31
import os
 
32
import shutil
 
33
 
 
34
import conf
 
35
import db
 
36
 
 
37
def makeuser(username, password, nick, fullname, rolenm, studentid):
 
38
    """Creates a new user on a pre-installed system.
 
39
    Sets up the following:
 
40
    * User's jail and home directory within the jail.
 
41
    * Subversion repository
 
42
    * Check out Subversion workspace into jail
 
43
    * Database details for user
 
44
    * Unix user account
 
45
    """
 
46
    homedir = make_jail(username)
 
47
    make_user_db(username, password, nick, fullname, rolenm, studentid)
 
48
    # TODO: -p password (need to use crypt)
 
49
    if os.system("useradd -d %s '%s'" % (homedir, username)) != 0:
 
50
        raise Exception("Failed to add Unix user account")
 
51
 
 
52
def make_jail(username):
 
53
    """Creates a new user's jail space, in the jail directory as configured in
 
54
    conf.py.
 
55
 
 
56
    This expects there to be a "template" directory within the jail root which
 
57
    contains all the files for a sample student jail. It creates the student's
 
58
    directory in the jail root, by making a hard-link copy of every file in the
 
59
    template directory, recursively.
 
60
 
 
61
    Returns the path to the user's home directory.
 
62
    """
 
63
    templatedir = os.path.join(conf.jail_base, 'template')
 
64
    if not os.path.isdir(templatedir):
 
65
        raise Exception("Template jail directory does not exist: " +
 
66
            templatedir)
 
67
    userdir = os.path.join(conf.jail_base, username)
 
68
    if os.path.exists(userdir):
 
69
        raise Exception("User's jail directory already exists: " +
 
70
            userdir)
 
71
 
 
72
    # Hard-link (copy aliasing) the entire tree over
 
73
    linktree(templatedir, userdir)
 
74
 
 
75
    # Set up the user's home directory
 
76
    homedir = os.path.join(userdir, 'home', username)
 
77
    os.mkdir(homedir)
 
78
    return homedir
 
79
 
 
80
def linktree(src, dst):
 
81
    """Recursively hard-link a directory tree using os.link().
 
82
 
 
83
    The destination directory must not already exist.
 
84
    If exception(s) occur, an Error is raised with a list of reasons.
 
85
 
 
86
    Symlinks are preserved (in fact, hard links are created which point to the
 
87
    symlinks).
 
88
 
 
89
    Code heavily based upon shutil.copytree from Python 2.5 library.
 
90
    """
 
91
    names = os.listdir(src)
 
92
    os.makedirs(dst)
 
93
    errors = []
 
94
    for name in names:
 
95
        srcname = os.path.join(src, name)
 
96
        dstname = os.path.join(dst, name)
 
97
        try:
 
98
            if os.path.isdir(srcname):
 
99
                linktree(srcname, dstname)
 
100
            else:
 
101
                os.link(srcname, dstname)
 
102
            # XXX What about devices, sockets etc.?
 
103
        except (IOError, os.error), why:
 
104
            errors.append((srcname, dstname, str(why)))
 
105
        # catch the Error from the recursive copytree so that we can
 
106
        # continue with other files
 
107
        except Exception, err:
 
108
            errors.append(err.args[0])
 
109
    try:
 
110
        shutil.copystat(src, dst)
 
111
    except WindowsError:
 
112
        # can't copy file access times on Windows
 
113
        pass
 
114
    except OSError, why:
 
115
        errors.extend((src, dst, str(why)))
 
116
    if errors:
 
117
        raise Exception, errors
 
118
 
 
119
def make_user_db(login, password, nick, fullname, rolenm, studentid):
 
120
    """Creates a user's entry in the database, filling in all the fields."""
 
121
    dbconn = db.DB()
 
122
    dbconn.create_user(login, password, nick, fullname, rolenm, studentid)
 
123
    dbconn.close()