← Back to branch summary

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

« back to all changes in this revision

Viewing changes to setup/util.py

  • Committer: David Coles
  • Date: 2010-08-30 03:26:13 UTC
  • Revision ID: coles.david@gmail.com-20100830032613-d14vng0jkelniu3l
python-console: Fix globals broken with new JSON library.

simplejson always returns unicode strings. cJSON would return ordinary strings 
if possible. cPickle.loads() only accepts strings. At present we use pickle 
version 0 so they should all works as ASCII strings. Higher versions of pickle 
are not plain ASCII and are likely to break this and so this should be fixed 
at some point.

Also replaced unconditional exception with one that catches Pickle errors. Not 
sure the best way to report failures of these functions.

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_append', 'action_chown',
 
37
           'action_symlink', 'action_chown',
38
38
           'action_chown_setuid', 'action_chmod_x', 'action_make_private',
39
 
           'query_user', 'filter_mutate', 'get_svn_revision', 'InstallList',
40
 
           'wwwuid']
 
39
           'filter_mutate', 'get_svn_revision', 'InstallList',
 
40
           'make_install_path', 'wwwuid']
41
41
 
42
42
# Determine which Python version (2.4 or 2.5, for example) we are running,
43
43
# and use that as the filename to the Python directory.
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
 
 
203
198
def action_chown_setuid(file, dry):
204
199
    """Chowns a file to root, and sets the setuid bit on the file.
205
200
    Calling this function requires the euid to be root.
235
230
    if not dry:
236
231
        os.chmod(file, stat.S_IRUSR | stat.S_IWUSR)
237
232
 
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
 
 
268
233
def filter_mutate(function, list):
269
234
    """Like built-in filter, but mutates the given list instead of returning a
270
235
    new one. Returns None."""
291
256
# Mime types which will automatically be placed in the list by InstallList.
292
257
installlist_mimetypes = ['text/x-python', 'text/html',
293
258
    'application/x-javascript', 'application/javascript',
294
 
    'text/css', 'image/png', 'image/gif', 'application/xml']
 
259
    'text/css', 'image/png', 'image/gif', 'application/xml', 'text/plain',
 
260
    'text/x-sh']
 
261
# Filenames which will automatically be placed in the list by InstallList.
 
262
whitelist_filenames = ['ivle-spec.conf']
295
263
 
296
264
def build_list_py_files(dir, no_top_level=False):
297
265
    """Builds a list of all py files found in a directory and its
305
273
        filter_mutate(lambda x: x[0] != '.', dirnames)
306
274
        # All *.py files are added to the list
307
275
        pylist += [os.path.join(dirpath, item) for item in filenames
308
 
            if mimetypes.guess_type(item)[0] in installlist_mimetypes]
 
276
            if mimetypes.guess_type(item)[0] in installlist_mimetypes or
 
277
               item in whitelist_filenames]
309
278
    if no_top_level:
310
279
        for i in range(0, len(pylist)):
311
280
            _, pylist[i] = pylist[i].split(os.sep, 1)
312
281
    return pylist
313
282
 
 
283
def make_install_path(rootdir, path):
 
284
    '''Combine an installation root directory and final install path.
 
285
 
 
286
    Normalises path, and joins it to the end of rootdir, removing the leading
 
287
    / to make it relative if required.
 
288
    '''
 
289
    normpath = os.path.normpath(path)
 
290
    if normpath.startswith(os.sep):
 
291
        normpath = normpath[1:]
 
292
    return os.path.join(rootdir, normpath)
 
293
 
314
294
class InstallList(object):
315
 
    # We build two separate lists, by walking www and console
316
 
    list_www = property(lambda self: build_list_py_files('www'))
317
 
 
318
295
    list_ivle_lib = property(lambda self: build_list_py_files('ivle'))
319
296
 
320
297
    list_subjects = property(lambda self: build_list_py_files('subjects',
327
304
        "services/python-console",
328
305
        "services/fileservice",
329
306
        "services/serveservice",
330
 
        "services/interpretservice",
331
307
        "services/usrmgt-server",
332
308
        "services/diffservice",
333
309
        "services/svnlogservice",
 
310
        "services/usrmgt-server", # XXX: Should be in bin/
334
311
    ]
335
312
 
336
313
    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",
337
320
        "bin/ivle-enrol",
338
321
        "bin/ivle-enrolallusers",
 
322
        "bin/ivle-fetchsubmissions",
339
323
        "bin/ivle-listusers",
340
 
        "bin/ivle-makeuser",
341
 
        "bin/ivle-marks",
 
324
        "bin/ivle-loadsampledata",
342
325
        "bin/ivle-mountallusers",
 
326
        "bin/ivle-refreshfilesystem",
343
327
        "bin/ivle-remakeuser",
344
328
        "bin/ivle-showenrolment",
345
329
    ]