← Back to branch summary

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

« back to all changes in this revision

Viewing changes to setup/install.py

Merge from object-publishing.

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup/install.py
19
19
# Author: Matt Giuca, Refactored by David Coles
20
20
# Date:   03/07/2008
21
21
 
22
 
# setup/build.py
23
 
# Compiles all files and sets up a jail template in the source directory.
24
 
# Details:
25
 
# Compiles (GCC) trampoline/trampoline.c to trampoline/trampoline.
26
 
# Creates jail/.
27
 
# Creates standard subdirs inside the jail, eg bin, opt, home, tmp.
28
 
# Copies console/ to a location within the jail.
29
 
# Copies OS programs and files to corresponding locations within the jail
30
 
#   (eg. python and Python libs, ld.so, etc).
31
 
# Generates .pyc files for all the IVLE .py files.
32
 
 
33
22
import optparse
34
23
import os
35
 
from setuputil import *
 
24
import sys
 
25
import functools
 
26
import distutils.sysconfig
 
27
 
 
28
from setup import util
36
29
 
37
30
def install(args):
38
31
    usage = """usage: %prog install [options]
39
32
(Requires root)
40
33
Create target install directory ($target).
41
34
Create $target/bin.
42
 
Copy trampoline/trampoline to $target/bin.
43
 
Copy timount/timount to $target/bin.
 
35
Copy bin/trampoline/trampoline to $target/bin.
 
36
Copy bin/timount/timount to $target/bin.
44
37
chown and chmod the installed trampoline.
45
 
Copy www/ to $target.
46
 
Copy jail/ to jail_system directory (unless --nojail specified).
47
 
Copy subjects/ to subjects directory (unless --nosubjects specified).
48
38
"""
49
39
 
50
40
    # Parse arguments
52
42
    parser.add_option("-n", "--dry",
53
43
        action="store_true", dest="dry",
54
44
        help="Print out the actions but don't do anything.")
55
 
    parser.add_option("-J", "--nojail",
56
 
        action="store_true", dest="nojail",
57
 
        help="Don't copy jail/ to jail_system directory")
58
 
    parser.add_option("-S", "--nosubjects",
59
 
        action="store_true", dest="nosubjects",
60
 
        help="Don't copy subject/ to subjects directory.")
 
45
    parser.add_option("--root",
 
46
        action="store", dest="rootdir",
 
47
        help="Install into a different root directory.",
 
48
        default='/')
 
49
    parser.add_option("--prefix",
 
50
        action="store", dest="prefix",
 
51
        help="Base prefix to install IVLE into (default: /usr/local).",
 
52
        default='/usr/local')
 
53
    parser.add_option("--python-site-packages",
 
54
        action="store", dest="python_site_packages",
 
55
        help="Path to Python site packages directory.",
 
56
        default=None)
61
57
    (options, args) = parser.parse_args(args)
62
58
 
 
59
    # Prefix must be absolute (not really necessary, but since a relative
 
60
    # prefix will be taken relative to *root* not working directory, it is
 
61
    # confusing if we allow it).
 
62
    if options.prefix[:1] not in (os.path.sep, os.path.altsep):
 
63
        print >>sys.stderr, """prefix must be an absolute path.
 
64
    (This will be interpreted relative to root, so provide --root=. if you
 
65
    want a path relative to the working directory)."""
 
66
        return 1
 
67
 
 
68
    # Calculate python_site_packages using the supplied prefix
 
69
    if options.python_site_packages is None:
 
70
        options.python_site_packages = distutils.sysconfig.get_python_lib(
 
71
            prefix=options.prefix)
 
72
 
63
73
    # Call the real function
64
 
    __install(options.dry, options.nojail, options.nosubjects)
65
 
 
66
 
def __install(dry=False,nojail=False,nosubjects=False):
67
 
    # Importing configuration is a little tricky
68
 
    sys.path.append('lib')
69
 
    import conf.conf
70
 
    import install_list
71
 
 
72
 
    # Pull the required varibles out of the config
73
 
    ivle_install_dir = conf.conf.ivle_install_dir
74
 
    jail_base = conf.conf.jail_base
75
 
    jail_system = conf.conf.jail_system
76
 
    subjects_base = conf.conf.subjects_base
77
 
    exercises_base = conf.conf.exercises_base
 
74
    return __install(prefix=options.prefix,
 
75
                     python_site_packages=options.python_site_packages,
 
76
                     dry=options.dry, rootdir=options.rootdir)
 
77
 
 
78
def __install(prefix, python_site_packages, dry=False, rootdir=None):
 
79
    install_list = util.InstallList()
 
80
 
 
81
    # We need to apply make_install_path with the rootdir to an awful lot of
 
82
    # config variables, so make it easy:
 
83
    mip = functools.partial(util.make_install_path, rootdir)
 
84
 
 
85
    # Compute the lib_path, share_path and bin_path (copied from
 
86
    # ivle/conf/conf.py).
 
87
    lib_path = mip(os.path.join(prefix, 'lib/ivle'))
 
88
    share_path = mip(os.path.join(prefix, 'share/ivle'))
 
89
    bin_path = mip(os.path.join(prefix, 'bin'))
 
90
    python_site_packages = mip(python_site_packages)
78
91
 
79
92
    # Must be run as root or a dry run  
80
93
    if dry:
81
94
        print "Dry run (no actions will be executed)\n"
82
95
    
83
96
    if not dry and os.geteuid() != 0:
84
 
        print >>sys.stderr, "Must be root to run build"
85
 
        print >>sys.stderr, "(I need to chown)."
 
97
        print >>sys.stderr, "Must be root to run install"
86
98
        return 1
87
 
    
88
 
    # Create the target (install) directory
89
 
    action_mkdir(ivle_install_dir, dry)
90
 
 
91
 
    # Create bin and copy the compiled files there
92
 
    action_mkdir(os.path.join(ivle_install_dir, 'bin'), dry)
93
 
    tramppath = os.path.join(ivle_install_dir, 'bin/trampoline')
94
 
    action_copyfile('trampoline/trampoline', tramppath, dry)
 
99
 
 
100
    # Create the config directory.
 
101
    util.action_mkdir(mip('/etc/ivle/plugins.d'), dry)
 
102
 
 
103
    # Create lib and copy the compiled files there
 
104
    util.action_mkdir(lib_path, dry)
 
105
 
 
106
    tramppath = os.path.join(lib_path, 'trampoline')
 
107
    util.action_copyfile('bin/trampoline/trampoline', tramppath, dry)
95
108
    # chown trampoline to root and set setuid bit
96
 
    action_chown_setuid(tramppath, dry)
97
 
 
98
 
    timountpath = os.path.join(ivle_install_dir, 'bin/timount')
99
 
    action_copyfile('timount/timount', timountpath, dry)
100
 
 
101
 
    # Create a scripts directory to put the usrmgt-server in.
102
 
    action_mkdir(os.path.join(ivle_install_dir, 'scripts'), dry)
103
 
    usrmgtpath = os.path.join(ivle_install_dir, 'scripts/usrmgt-server')
104
 
    action_copyfile('scripts/usrmgt-server', usrmgtpath, dry)
105
 
    action_chmod_x(usrmgtpath, dry)
106
 
 
107
 
    # Copy the www and lib directories using the list
108
 
    action_copylist(install_list.list_www, ivle_install_dir, dry)
109
 
    action_copylist(install_list.list_lib, ivle_install_dir, dry)
110
 
    
111
 
    # Copy the php directory
112
 
    forum_dir = "www/php/phpBB3"
113
 
    forum_path = os.path.join(ivle_install_dir, forum_dir)
114
 
    action_copytree(forum_dir, forum_path, dry)
115
 
    print "chown -R www-data:www-data %s" % forum_path
116
 
    if not dry:
117
 
        os.system("chown -R www-data:www-data %s" % forum_path)
118
 
 
119
 
    if not nojail:
120
 
        # Copy the local jail directory built by the build action
121
 
        # to the jail_system directory (it will be used to help build
122
 
        # all the students' jails).
123
 
        action_copytree('jail', jail_system, dry)
124
 
    if not nosubjects:
125
 
        # Copy the subjects and exercises directories across
126
 
        action_copylist(install_list.list_subjects, subjects_base, dry,
127
 
            srcdir="./subjects")
128
 
        action_copylist(install_list.list_exercises, exercises_base, dry,
129
 
            srcdir="./exercises")
130
 
 
131
 
    # Append IVLE path to ivle.pth in python site packages
132
 
    # (Unless it's already there)
133
 
    ivle_pth = os.path.join(sys.prefix,
134
 
        "lib/python%s/site-packages/ivle.pth" % PYTHON_VERSION)
135
 
    ivle_www = os.path.join(ivle_install_dir, "www")
136
 
    ivle_lib = os.path.join(ivle_install_dir, "lib")
137
 
    write_ivle_pth = True
138
 
    write_ivle_lib_pth = True
139
 
    try:
140
 
        file = open(ivle_pth, 'r')
141
 
        for line in file:
142
 
            if line.strip() == ivle_www:
143
 
                write_ivle_pth = False
144
 
            elif line.strip() == ivle_lib:
145
 
                write_ivle_lib_pth = False
146
 
        file.close()
147
 
    except (IOError, OSError):
148
 
        pass
149
 
    if write_ivle_pth:
150
 
        action_append(ivle_pth, ivle_www)
151
 
    if write_ivle_lib_pth:
152
 
        action_append(ivle_pth, ivle_lib)
153
 
 
154
 
 
155
 
    # Create the ivle working revision record file
156
 
    action_mkdir(os.path.join(ivle_install_dir, 'version'), dry)
157
 
    ivle_revision_record_file = os.path.join(ivle_install_dir, 'version/ivle-revision.txt')
158
 
    if not dry:
159
 
        try:
160
 
            conf = open(ivle_revision_record_file, "w")
161
 
 
162
 
            conf.write( "# IVLE code revision listing generated by running 'svn status -v ..' from " + os.getcwd() + "\n#\n\n")
163
 
 
164
 
            conf.close()
165
 
        except IOError, (errno, strerror):
166
 
            print "IO error(%s): %s" % (errno, strerror)
167
 
            sys.exit(1)
168
 
 
169
 
        os.system("svn status -v . >> %s" % ivle_revision_record_file)
170
 
 
171
 
    print "Wrote IVLE code revision status to %s" % ivle_revision_record_file
 
109
    util.action_chown_setuid(tramppath, dry)
 
110
 
 
111
    timountpath = os.path.join(lib_path, 'timount')
 
112
    util.action_copyfile('bin/timount/timount', timountpath, dry)
 
113
 
 
114
    # Copy in the services (only usrmgt-server is needed on the host, but
 
115
    # the jail build requires the rest).
 
116
    util.action_copylist(install_list.list_services, share_path, dry)
 
117
    usrmgtpath = os.path.join(share_path, 'services/usrmgt-server')
 
118
    util.action_chmod_x(usrmgtpath, dry)
 
119
 
 
120
    # Copy the user-executable binaries using the list.
 
121
    util.action_copylist(install_list.list_user_binaries, bin_path, dry,
 
122
                         onlybasename=True)
 
123
 
 
124
    # Copy the lib directory (using the list)
 
125
    util.action_copylist(install_list.list_ivle_lib, python_site_packages, dry)
172
126
 
173
127
    return 0
174
128