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

1092.1.16 by William Grant
Reimplement setup/buildjail.sh in Python. This means that sites can configure
1
# IVLE - Informatics Virtual Learning Environment
2
# Copyright (C) 2007-2009 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
'''Debian-specific bits of the jail building process.'''
19
20
import os
21
import subprocess
22
23
from ivle.jailbuilder.exceptions import JailBuildError
24
25
def debootstrap_create_jail(release, path, components=['main', 'universe'],
26
                            mirror=None):
27
    """Create a Debian-based jail using debootstrap."""
28
    if mirror is None:
29
        mirror = 'http://archive.ubuntu.com/ubuntu'
30
    ec = os.spawnvp(os.P_WAIT, 'debootstrap',
31
              ['debootstrap', '--components=' + ','.join(components),
32
               '--include=ubuntu-keyring', '--variant=minbase',
33
               release, path, mirror])
34
    if ec != 0:
35
        raise JailBuildError('debootstrap failed with code %d' % ec)
36
37
    mangle_sources_list(path, clobber=True)
38
    mangle_sources_list(path, lines=[
39
            'deb %s %s%s %s' % (mirror, release, pocket, ' '.join(components))
40
            for pocket in ('', '-updates', '-security')])
41
42
def mangle_sources_list(path, lines=[], clobber=False):
43
    """Add lines to a sources.list, optionally clobbering an existing one."""
44
    f = open(os.path.join(path, 'etc/apt/sources.list'),
45
             'w' if clobber else 'a')
46
47
    for line in lines:
48
        f.write(line + '\n')
49
    f.close()
50
51
def _execute_in_chroot(path, argv, stdin=None):
52
    """Execute a binary in a chroot, with optional stdin."""
53
    pid = subprocess.Popen(['chroot', path] + argv, stdin=subprocess.PIPE)
54
    pid.communicate(stdin)
55
    if pid.returncode != 0:
56
        raise JailBuildError('command failed with code %d' % pid.returncode)
57
58
def apt_update_cache(path):
59
    """Update the apt cache in a chroot."""
60
    _execute_in_chroot(path, ['apt-get', '-y', 'update'])
61
62
def apt_upgrade(path):
63
    """Upgrade apt packages in a chroot."""
64
    _execute_in_chroot(path, ['apt-get', '-y', 'upgrade'])
65
66
def apt_install(path, packages=None):
67
    """Install apt packages in a chroot."""
68
    if not packages:
69
        return
70
    _execute_in_chroot(path, ['apt-get', '-y', 'install'] + packages)
71
72
def apt_clean(path):
73
    """Clean the apt package cache in a chroot."""
74
    _execute_in_chroot(path, ['apt-get', 'clean'])
75
76
def apt_add_key(path, key_text):
77
    """Add a key to authenticate apt repositories in a chroot."""
78
    _execute_in_chroot(path, ['apt-key', 'add', '-'], stdin=key_text)