← Back to branch summary

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

« back to all changes in this revision

Viewing changes to setup/install.py

  • Committer: William Grant
  • Date: 2010-07-20 04:52:31 UTC
  • mto: This revision was merged to the branch mainline in revision 1826.
  • Revision ID: grantw@unimelb.edu.au-20100720045231-8ia3uk8nao8zdq1i
Replace cjson with json, or simplejson if json is not available (Python <2.6)

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(os.path.join(os.pardir, 'lib'))
69
 
    sys.path.append(os.pardir)
70
 
    import conf.conf
71
 
    import install_list
72
 
 
73
 
    # Pull the required varibles out of the config
74
 
    ivle_install_dir = conf.conf.ivle_install_dir
75
 
    jail_base = conf.conf.jail_base
76
 
    jail_system = conf.conf.jail_system
77
 
    subjects_base = conf.conf.subjects_base
78
 
    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)
79
91
 
80
92
    # Must be run as root or a dry run  
81
93
    if dry:
82
94
        print "Dry run (no actions will be executed)\n"
83
95
    
84
96
    if not dry and os.geteuid() != 0:
85
 
        print >>sys.stderr, "Must be root to run build"
86
 
        print >>sys.stderr, "(I need to chown)."
 
97
        print >>sys.stderr, "Must be root to run install"
87
98
        return 1
88
 
    
89
 
    # Create the target (install) directory
90
 
    action_mkdir(ivle_install_dir, dry)
91
 
 
92
 
    # Create bin and copy the compiled files there
93
 
    action_mkdir(os.path.join(ivle_install_dir, 'bin'), dry)
94
 
    tramppath = os.path.join(ivle_install_dir, 'bin/trampoline')
95
 
    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)
96
108
    # chown trampoline to root and set setuid bit
97
 
    action_chown_setuid(tramppath, dry)
98
 
 
99
 
    timountpath = os.path.join(ivle_install_dir, 'bin/timount')
100
 
    action_copyfile('timount/timount', timountpath, dry)
101
 
 
102
 
    # Create a scripts directory to put the usrmgt-server in.
103
 
    action_mkdir(os.path.join(ivle_install_dir, 'scripts'), dry)
104
 
    usrmgtpath = os.path.join(ivle_install_dir, 'scripts/usrmgt-server')
105
 
    action_copyfile('scripts/usrmgt-server', usrmgtpath, dry)
106
 
    action_chmod_x(usrmgtpath, dry)
107
 
 
108
 
    # Copy the www and lib directories using the list
109
 
    action_copylist(install_list.list_www, ivle_install_dir, dry)
110
 
    action_copylist(install_list.list_lib, ivle_install_dir, dry)
111
 
    
112
 
    # Copy the php directory
113
 
    forum_dir = "www/php/phpBB3"
114
 
    forum_path = os.path.join(ivle_install_dir, forum_dir)
115
 
    action_copytree(forum_dir, forum_path, dry)
116
 
    print "chown -R www-data:www-data %s" % forum_path
117
 
    if not dry:
118
 
        os.system("chown -R www-data:www-data %s" % forum_path)
119
 
 
120
 
    if not nojail:
121
 
        # Copy the local jail directory built by the build action
122
 
        # to the jail_system directory (it will be used to help build
123
 
        # all the students' jails).
124
 
        action_copytree('jail', jail_system, dry)
125
 
    if not nosubjects:
126
 
        # Copy the subjects and exercises directories across
127
 
        action_copylist(install_list.list_subjects, subjects_base, dry,
128
 
            srcdir="./subjects")
129
 
        action_copylist(install_list.list_exercises, exercises_base, dry,
130
 
            srcdir="./exercises")
131
 
 
132
 
    # Append IVLE path to ivle.pth in python site packages
133
 
    # (Unless it's already there)
134
 
    ivle_pth = os.path.join(sys.prefix,
135
 
        "lib/python%s/site-packages/ivle.pth" % PYTHON_VERSION)
136
 
    ivle_www = os.path.join(ivle_install_dir, "www")
137
 
    ivle_lib = os.path.join(ivle_install_dir, "lib")
138
 
    write_ivle_pth = True
139
 
    write_ivle_lib_pth = True
140
 
    try:
141
 
        file = open(ivle_pth, 'r')
142
 
        for line in file:
143
 
            if line.strip() == ivle_www:
144
 
                write_ivle_pth = False
145
 
            elif line.strip() == ivle_lib:
146
 
                write_ivle_lib_pth = False
147
 
        file.close()
148
 
    except (IOError, OSError):
149
 
        pass
150
 
    if write_ivle_pth:
151
 
        action_append(ivle_pth, ivle_www)
152
 
    if write_ivle_lib_pth:
153
 
        action_append(ivle_pth, ivle_lib)
154
 
 
155
 
 
156
 
    # Create the ivle working revision record file
157
 
    action_mkdir(os.path.join(ivle_install_dir, 'version'), dry)
158
 
    ivle_revision_record_file = os.path.join(ivle_install_dir, 'version/ivle-revision.txt')
159
 
    if not dry:
160
 
        try:
161
 
            conf = open(ivle_revision_record_file, "w")
162
 
 
163
 
            conf.write( "# IVLE code revision listing generated by running 'svn status -v ..' from " + os.getcwd() + "\n#\n\n")
164
 
 
165
 
            conf.close()
166
 
        except IOError, (errno, strerror):
167
 
            print "IO error(%s): %s" % (errno, strerror)
168
 
            sys.exit(1)
169
 
 
170
 
        os.system("svn status -v . >> %s" % ivle_revision_record_file)
171
 
 
172
 
    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)
173
126
 
174
127
    return 0
175
128