← Back to branch summary

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

« back to all changes in this revision

Viewing changes to setup/util.py

Reimplement setup/buildjail.sh in Python. This means that sites can configure
additional repositories and packages to include in the jail, without modifying
the source tree.

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup/util.py
34
34
__all__ = ['PYTHON_VERSION', 'copy_file_to_jail', 'RunError',
35
35
           'action_runprog', 'action_remove', 'action_rename', 'action_mkdir',
36
36
           'action_copytree', 'action_copylist', 'action_copyfile',
37
 
           'action_symlink', 'action_chown',
 
37
           'action_symlink', 'action_append', 'action_chown',
38
38
           'action_chown_setuid', 'action_chmod_x', 'action_make_private',
39
 
           'filter_mutate', 'get_svn_revision', 'InstallList',
 
39
           'query_user', 'filter_mutate', 'get_svn_revision', 'InstallList',
40
40
           'make_install_path', 'wwwuid']
41
41
 
42
42
# Determine which Python version (2.4 or 2.5, for example) we are running,
195
195
    if not dry:
196
196
        os.symlink(src, dst)
197
197
 
 
198
def action_append(ivle_pth, ivle_www):
 
199
    file = open(ivle_pth, 'a+')
 
200
    file.write(ivle_www + '\n')
 
201
    file.close()
 
202
 
198
203
def action_chown_setuid(file, dry):
199
204
    """Chowns a file to root, and sets the setuid bit on the file.
200
205
    Calling this function requires the euid to be root.
230
235
    if not dry:
231
236
        os.chmod(file, stat.S_IRUSR | stat.S_IWUSR)
232
237
 
 
238
def query_user(default, prompt):
 
239
    """Prompts the user for a string, which is read from a line of stdin.
 
240
    Exits silently if EOF is encountered. Returns the string, with spaces
 
241
    removed from the beginning and end.
 
242
 
 
243
    Returns default if a 0-length line (after spaces removed) was read.
 
244
    """
 
245
    if default is None:
 
246
        # A default of None means the value will be computed specially, so we
 
247
        # can't really tell you what it is
 
248
        defaultstr = "computed"
 
249
    elif isinstance(default, basestring):
 
250
        defaultstr = '"%s"' % default
 
251
    else:
 
252
        defaultstr = repr(default)
 
253
    sys.stdout.write('%s\n    (default: %s)\n>' % (prompt, defaultstr))
 
254
    try:
 
255
        val = sys.stdin.readline()
 
256
    except KeyboardInterrupt:
 
257
        # Ctrl+C
 
258
        sys.stdout.write("\n")
 
259
        sys.exit(1)
 
260
    sys.stdout.write("\n")
 
261
    # If EOF, exit
 
262
    if val == '': sys.exit(1)
 
263
    # If empty line, return default
 
264
    val = val.strip()
 
265
    if val == '': return default
 
266
    return val
 
267
 
233
268
def filter_mutate(function, list):
234
269
    """Like built-in filter, but mutates the given list instead of returning a
235
270
    new one. Returns None."""
256
291
# Mime types which will automatically be placed in the list by InstallList.
257
292
installlist_mimetypes = ['text/x-python', 'text/html',
258
293
    'application/x-javascript', 'application/javascript',
259
 
    'text/css', 'image/png', 'image/gif', 'application/xml', 'text/plain',
260
 
    'text/x-sh']
 
294
    'text/css', 'image/png', 'image/gif', 'application/xml']
261
295
# Filenames which will automatically be placed in the list by InstallList.
262
296
whitelist_filenames = ['ivle-spec.conf']
263
297
 
292
326
    return os.path.join(rootdir, normpath)
293
327
 
294
328
class InstallList(object):
 
329
    # We build two separate lists, by walking www and console
 
330
    list_www = property(lambda self: build_list_py_files('www'))
 
331
 
295
332
    list_ivle_lib = property(lambda self: build_list_py_files('ivle'))
296
333
 
297
334
    list_subjects = property(lambda self: build_list_py_files('subjects',
304
341
        "services/python-console",
305
342
        "services/fileservice",
306
343
        "services/serveservice",
 
344
        "services/interpretservice",
307
345
        "services/usrmgt-server",
308
346
        "services/diffservice",
309
347
        "services/svnlogservice",
311
349
    ]
312
350
 
313
351
    list_user_binaries = [
314
 
        "bin/ivle-addexercise",
315
 
        "bin/ivle-adduser",
316
 
        "bin/ivle-buildjail",
317
 
        "bin/ivle-cloneworksheets",
318
 
        "bin/ivle-config",
319
 
        "bin/ivle-createdatadirs",
320
352
        "bin/ivle-enrol",
321
353
        "bin/ivle-enrolallusers",
322
 
        "bin/ivle-fetchsubmissions",
323
354
        "bin/ivle-listusers",
324
 
        "bin/ivle-loadsampledata",
 
355
        "bin/ivle-makeuser",
 
356
        "bin/ivle-marks",
325
357
        "bin/ivle-mountallusers",
326
 
        "bin/ivle-refreshfilesystem",
327
358
        "bin/ivle-remakeuser",
328
359
        "bin/ivle-showenrolment",
 
360
        "bin/ivle-buildjail",
329
361
    ]