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

« back to all changes in this revision

Viewing changes to ivle/util.py

  • Committer: Nick Chadwick
  • Date: 2009-02-25 16:39:59 UTC
  • mto: (1099.1.227 exercise-ui)
  • mto: This revision was merged to the branch mainline in revision 1162.
  • Revision ID: chadnickbok@gmail.com-20090225163959-3hsedlesu2pjm12e
Added a few extra files. I'm not sure if they're necessary, and I don't
have time to find out.

Show diffs side-by-side

added added

removed removed

Lines of Context:
20
20
# Date: 12/12/2007
21
21
 
22
22
# Contains common utility functions.
 
23
# Also initialises mime types library. You must import util before using
 
24
# Python's builtin mimetypes module to make sure local settings are applied.
23
25
 
24
26
import os
25
 
import sys
26
 
import stat
 
27
import mimetypes
 
28
import datetime
 
29
 
 
30
import ivle.conf
 
31
import ivle.conf.mimetypes
27
32
 
28
33
class IVLEError(Exception):
29
 
    """Legacy general IVLE exception.
30
 
 
31
 
    This is the old "standard" exception class for IVLE errors. It is only
32
 
    used in fileservice, and should not be used in any new code.
33
 
    """
34
 
 
35
 
    message = None
36
 
 
 
34
    """
 
35
    This is the "standard" exception class for IVLE errors.
 
36
    It is the ONLY exception which should propagate to the top - it will then
 
37
    be displayed to the user as an HTTP error with the given code.
 
38
 
 
39
    All other exceptions are considered IVLE bugs and should not occur
 
40
    (they will display a traceback).
 
41
 
 
42
    This error should not be raised directly. Call req.throw_error.
 
43
    """
37
44
    def __init__(self, httpcode, message=None):
38
45
        self.httpcode = httpcode
39
46
        self.message = message
40
47
        self.args = (httpcode, message)
41
48
 
42
49
class IVLEJailError(Exception):
43
 
    """Exception proxying an in-jail error.
44
 
 
 
50
    """
45
51
    This exception indicates an error that occurred inside an IVLE CGI script
46
52
    inside the jail. It should never be raised directly - only by the 
47
53
    interpreter.
64
70
    def __repr__(self):
65
71
        return "<Fake %s %s>"%(self.type, self.name)
66
72
 
 
73
 
 
74
def make_path(path):
 
75
    """Given a path relative to the IVLE root, makes the path relative to the
 
76
    site root using ivle.conf.root_dir. This path can be used in URLs sent to
 
77
    the client."""
 
78
    return os.path.join(ivle.conf.root_dir, path)
 
79
 
 
80
def make_local_path(path):
 
81
    """Given a path relative to the IVLE root, on the local file system, makes
 
82
    the path relative to the root using ivle.conf.share_path. This path can
 
83
    be used in reading files from the local file system."""
 
84
    return os.path.join(ivle.conf.share_path, 'www', path)
 
85
 
 
86
def unmake_path(path):
 
87
    """Given a path relative to the site root, makes the path relative to the
 
88
    IVLE root by removing ivle.conf.root_dir if it appears at the beginning. If
 
89
    it does not appear at the beginning, returns path unchanged. Also
 
90
    normalises the path."""
 
91
    path = os.path.normpath(path)
 
92
    root = os.path.normpath(ivle.conf.root_dir)
 
93
 
 
94
    if path.startswith(root):
 
95
        path = path[len(root):]
 
96
        # Take out the slash as well
 
97
        if len(path) > 0 and path[0] == os.sep:
 
98
            path = path[1:]
 
99
 
 
100
    return path
 
101
 
67
102
def split_path(path):
68
103
    """Given a path, returns a tuple consisting of the top-level directory in
69
104
    the path, and the rest of the path. Note that both items in the tuple will
99
134
    else:
100
135
        return tuple(splitpath)
101
136
 
102
 
def relpath(path, start=os.path.curdir):
103
 
    """Return a relative version of a path.
104
 
    XXX Backported from Python 2.6 posixpath.py.
105
 
    """
106
 
 
107
 
    if not path:
108
 
        raise ValueError("no path specified")
109
 
 
110
 
    start_list = os.path.abspath(start).split(os.path.sep)
111
 
    path_list = os.path.abspath(path).split(os.path.sep)
112
 
 
113
 
    # Work out how much of the filepath is shared by start and path.
114
 
    i = len(os.path.commonprefix([start_list, path_list]))
115
 
 
116
 
    rel_list = [os.path.pardir] * (len(start_list)-i) + path_list[i:]
117
 
    if not rel_list:
118
 
        return os.path.curdir
119
 
    return os.path.join(*rel_list)
 
137
def open_exercise_file(exercisename):
 
138
    """Given an exercise name, opens the corresponding XML file for reading.
 
139
    Returns None if the exercise file was not found.
 
140
    (For tutorials / worksheets).
 
141
    """
 
142
    # First normalise the path
 
143
    exercisename = os.path.normpath(exercisename)
 
144
    # Now if it begins with ".." or separator, then it's illegal
 
145
    if exercisename.startswith("..") or exercisename.startswith(os.sep):
 
146
        exercisefile = None
 
147
    else:
 
148
        exercisefile = os.path.join(ivle.conf.exercises_base, exercisename)
 
149
 
 
150
    try:
 
151
        return open(exercisefile)
 
152
    except (TypeError, IOError):    # TypeError if exercisefile == None
 
153
        return None
 
154
 
 
155
# Initialise mime types library
 
156
mimetypes.init()
 
157
for (ext, mimetype) in ivle.conf.mimetypes.additional_mime_types.items():
 
158
    mimetypes.add_type(mimetype, ext)
 
159
 
 
160
def nice_filetype(filename):
 
161
    """Given a filename or basename, returns a "friendly" name for that
 
162
    file's type.
 
163
    eg. nice_mimetype("file.py") == "Python source code".
 
164
        nice_filetype("file.bzg") == "BZG file".
 
165
        nice_filetype("directory/") == "Directory".
 
166
    """
 
167
    if filename[-1] == os.sep:
 
168
        return "Directory"
 
169
    else:
 
170
        try:
 
171
            return ivle.conf.mimetypes.nice_mimetypes[
 
172
                mimetypes.guess_type(filename)[0]]
 
173
        except KeyError:
 
174
            filename = os.path.basename(filename)
 
175
            try:
 
176
                return filename[filename.rindex('.')+1:].upper() + " file"
 
177
            except ValueError:
 
178
                return "File"
 
179
 
 
180
def get_terms_of_service():
 
181
    """
 
182
    Sends the Terms of Service document to the req object.
 
183
    This consults conf to find out where the TOS is located on disk, and sends
 
184
    that. If it isn't found, it sends a generic message explaining to admins
 
185
    how to create a real one.
 
186
    """
 
187
    try:
 
188
        return open(ivle.conf.tos_path).read()
 
189
    except IOError:
 
190
        return """<h1>Terms of Service</h1>
 
191
<p><b>*** SAMPLE ONLY ***</b></p>
 
192
<p>This is the text of the IVLE Terms of Service.</p>
 
193
<p>The administrator should create a license file with an appropriate
 
194
"Terms of Service" license for your organisation.</p>
 
195
<h2>Instructions for Administrators</h2>
 
196
<p>You are seeing this message because you have not configured a Terms of
 
197
Service document.</p>
 
198
<p>When you configured IVLE, you specified a path to the Terms of Service
 
199
document (this is found in <b><tt>ivle/conf/conf.py</tt></b> under
 
200
"<tt>tos_path</tt>").</p>
 
201
<p>Create a file at this location; an HTML file with the appropriately-worded
 
202
license.</p>
 
203
<p>This should be a normal XHTML file, except it should not contain
 
204
<tt>html</tt>, <tt>head</tt> or <tt>body</tt> elements - it should
 
205
just be the contents of a body element (IVLE will wrap it accordingly).</p>
 
206
<p>This will automatically be used as the license text instead of this
 
207
placeholder text.</p>
 
208
"""
 
209
 
 
210
def parse_iso8601(str):
 
211
    """Parses ISO8601 string into a datetime object."""
 
212
    # FIXME: Terrific hack that means we only accept the format that is 
 
213
    # produced by json.js module when it encodes date objects.
 
214
    return datetime.datetime.strptime(str, "%Y-%m-%dT%H:%M:%SZ")
120
215
 
121
216
def incomplete_utf8_sequence(byteseq):
122
 
    """Calculate the completeness of a UTF-8 encoded string.
123
 
 
 
217
    """
 
218
    str -> int
124
219
    Given a UTF-8-encoded byte sequence (str), returns the number of bytes at
125
220
    the end of the string which comprise an incomplete UTF-8 character
126
221
    sequence.
198
293
        return count
199
294
 
200
295
def object_to_dict(attrnames, obj):
201
 
    """Convert an object into a dictionary.
202
 
 
203
 
    This takes a shallow copy of the object.
204
 
 
205
 
    @param attrnames: Set (or iterable) of names of attributes to be copied
206
 
                      into the dictionary. (We don't auto-lookup, because this
207
 
                      function needs to be used on magical objects).
 
296
    """
 
297
    Convert an object into a dictionary. This takes a shallow copy of the
 
298
    object.
 
299
    attrnames: Set (or iterable) of names of attributes to be copied into the
 
300
        dictionary. (We don't auto-lookup, because this function needs to be
 
301
        used on magical objects).
208
302
    """
209
303
    return dict((k, getattr(obj, k))
210
304
        for k in attrnames if not k.startswith('_'))
211
 
 
212
 
def safe_rmtree(path, ignore_errors=False, onerror=None):
213
 
    """Recursively delete a directory tree.
214
 
 
215
 
    Copied from shutil.rmtree from Python 2.6, which does not follow symbolic
216
 
    links (it is otherwise unsafe to call as root on untrusted directories; do
217
 
    not use shutil.rmtree in this case, as you may be running Python 2.5).
218
 
 
219
 
    If ignore_errors is set, errors are ignored; otherwise, if onerror
220
 
    is set, it is called to handle the error with arguments (func,
221
 
    path, exc_info) where func is os.listdir, os.remove, or os.rmdir;
222
 
    path is the argument to that function that caused it to fail; and
223
 
    exc_info is a tuple returned by sys.exc_info().  If ignore_errors
224
 
    is false and onerror is None, an exception is raised.
225
 
 
226
 
    """
227
 
    if ignore_errors:
228
 
        def onerror(*args):
229
 
            pass
230
 
    elif onerror is None:
231
 
        def onerror(*args):
232
 
            raise
233
 
    try:
234
 
        if os.path.islink(path):
235
 
            # symlinks to directories are forbidden, see bug #1669
236
 
            raise OSError("Cannot call safe_rmtree on a symbolic link")
237
 
    except OSError:
238
 
        onerror(os.path.islink, path, sys.exc_info())
239
 
        # can't continue even if onerror hook returns
240
 
        return
241
 
    names = []
242
 
    try:
243
 
        names = os.listdir(path)
244
 
    except os.error, err:
245
 
        onerror(os.listdir, path, sys.exc_info())
246
 
    for name in names:
247
 
        fullname = os.path.join(path, name)
248
 
        try:
249
 
            mode = os.lstat(fullname).st_mode
250
 
        except os.error:
251
 
            mode = 0
252
 
        if stat.S_ISDIR(mode):
253
 
            safe_rmtree(fullname, ignore_errors, onerror)
254
 
        else:
255
 
            try:
256
 
                os.remove(fullname)
257
 
            except os.error, err:
258
 
                onerror(os.remove, fullname, sys.exc_info())
259
 
    try:
260
 
        os.rmdir(path)
261
 
    except os.error:
262
 
        onerror(os.rmdir, path, sys.exc_info())
263
 
 
264
 
def format_submission_principal(user, principal):
265
 
    """Render a list of users to fit in the offering project listing.
266
 
 
267
 
    Given a user and a list of submitters, returns 'solo' if the
268
 
    only submitter is the user, or a string of the form
269
 
    'with A, B and C' if there are any other submitters.
270
 
 
271
 
    If submitters is None, we assume that the list of members could
272
 
    not be determined, so we just return 'group'.
273
 
    """
274
 
    if principal is None:
275
 
        return 'group'
276
 
 
277
 
    if principal is user:
278
 
        return 'solo'
279
 
 
280
 
    display_names = sorted(
281
 
        member.display_name for member in principal.members
282
 
        if member is not user)
283
 
 
284
 
    if len(display_names) == 0:
285
 
        return 'solo (%s)' % principal.name
286
 
    elif len(display_names) == 1:
287
 
        return 'with %s (%s)' % (display_names[0], principal.name)
288
 
    elif len(display_names) > 5:
289
 
        return 'with %d others (%s)' % (len(display_names), principal.name)
290
 
    else:
291
 
        return 'with %s and %s (%s)' % (', '.join(display_names[:-1]),
292
 
                                        display_names[-1], principal.name)