← Back to branch summary

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

« back to all changes in this revision

Viewing changes to ivle/jailbuilder/debian.py

  • Committer: William Grant
  • Date: 2010-02-25 03:18:21 UTC
  • Revision ID: grantw@unimelb.edu.au-20100225031821-mi9a2tm5679fht4d
Shuffle things around so that req.user and req.store only construct when actually retrieved, and ensure they're not retrieved for media files. Saves 50ms of DB connection time per request.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/jailbuilder/debian.py
 
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)