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

« back to all changes in this revision

Viewing changes to lib/common/util.py

  • Committer: William Grant
  • Date: 2009-01-13 01:36:15 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:1123
Merge setup-refactor branch. This completely breaks existing installations;
every path (both filesystem and Python) has changed. Do not upgrade without
knowing what you are doing.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# IVLE - Informatics Virtual Learning Environment
2
 
# Copyright (C) 2007-2008 The University of Melbourne
3
 
#
4
 
# This program is free software; you can redistribute it and/or modify
5
 
# it under the terms of the GNU General Public License as published by
6
 
# the Free Software Foundation; either version 2 of the License, or
7
 
# (at your option) any later version.
8
 
#
9
 
# This program is distributed in the hope that it will be useful,
10
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
# GNU General Public License for more details.
13
 
#
14
 
# You should have received a copy of the GNU General Public License
15
 
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
 
 
18
 
# Module: common.util
19
 
# Author: Matt Giuca
20
 
# Date: 12/12/2007
21
 
 
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.
25
 
 
26
 
import os
27
 
import mimetypes
28
 
import datetime
29
 
 
30
 
import conf
31
 
import conf.mimetypes
32
 
 
33
 
class IVLEError(Exception):
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
 
    """
44
 
    def __init__(self, httpcode, message=None):
45
 
        self.httpcode = httpcode
46
 
        self.message = message
47
 
        self.args = (httpcode, message)
48
 
 
49
 
class IVLEJailError(Exception):
50
 
    """
51
 
    This exception indicates an error that occurred inside an IVLE CGI script
52
 
    inside the jail. It should never be raised directly - only by the 
53
 
    interpreter.
54
 
 
55
 
    Information will be retrieved from it, and then treated as a normal
56
 
    error.
57
 
    """
58
 
    def __init__(self, type_str, message, info):
59
 
        self.type_str = type_str
60
 
        self.message = message
61
 
        self.info = info
62
 
 
63
 
class FakeObject(object):
64
 
    """ A representation of an object that can't be Pickled """
65
 
    def __init__(self, type, name, attrib={}):
66
 
        self.type = type
67
 
        self.name = name
68
 
        self.attrib = attrib
69
 
 
70
 
    def __repr__(self):
71
 
        return "<Fake %s %s>"%(self.type, self.name)
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 conf.root_dir. This path can be used in URLs sent to the
77
 
    client."""
78
 
    return os.path.join(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 conf.ivle_install_dir. This path can
83
 
    be used in reading files from the local file system."""
84
 
    return os.path.join(conf.ivle_install_dir, '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 conf.root_dir if it appears at the beginning. If it
89
 
    does not appear at the beginning, returns path unchanged. Also normalises
90
 
    the path."""
91
 
    path = os.path.normpath(path)
92
 
    root = os.path.normpath(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
 
 
102
 
def split_path(path):
103
 
    """Given a path, returns a tuple consisting of the top-level directory in
104
 
    the path, and the rest of the path. Note that both items in the tuple will
105
 
    NOT begin with a slash, regardless of whether the original path did. Also
106
 
    normalises the path.
107
 
 
108
 
    Always returns a pair of strings, except for one special case, in which
109
 
    the path is completely empty (or just a single slash). In this case the
110
 
    return value will be (None, ''). But still always returns a pair.
111
 
 
112
 
    Examples:
113
 
 
114
 
    >>> split_path("")
115
 
    (None, '')
116
 
    >>> split_path("/")
117
 
    (None, '')
118
 
    >>> split_path("home")
119
 
    ('home', '')
120
 
    >>> split_path("home/docs/files")
121
 
    ('home', 'docs/files')
122
 
    >>> split_path("//home/docs/files")
123
 
    ('', 'home/docs/files')
124
 
    """
125
 
    path = os.path.normpath(path)
126
 
    # Ignore the opening slash
127
 
    if path.startswith(os.sep):
128
 
        path = path[len(os.sep):]
129
 
    if path == '' or path == '.':
130
 
        return (None, '')
131
 
    splitpath = path.split(os.sep, 1)
132
 
    if len(splitpath) == 1:
133
 
        return (splitpath[0], '')
134
 
    else:
135
 
        return tuple(splitpath)
136
 
 
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(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 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 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 send_terms_of_service(req):
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
 
        req.sendfile(conf.tos_path)
189
 
    except IOError:
190
 
        req.write(
191
 
"""<h1>Terms of Service</h1>
192
 
<p><b>*** SAMPLE ONLY ***</b></p>
193
 
<p>This is the text of the IVLE Terms of Service.</p>
194
 
<p>The administrator should create a license file with an appropriate
195
 
"Terms of Service" license for your organisation.</p>
196
 
<h2>Instructions for Administrators</h2>
197
 
<p>You are seeing this message because you have not configured a Terms of
198
 
Service document.</p>
199
 
<p>When you configured IVLE, you specified a path to the Terms of Service
200
 
document (this is found in <b><tt>lib/conf/conf.py</tt></b> under
201
 
"<tt>tos_path</tt>").</p>
202
 
<p>Create a file at this location; an HTML file with the appropriately-worded
203
 
license.</p>
204
 
<p>This should be a normal XHTML file, except it should not contain
205
 
<tt>html</tt>, <tt>head</tt> or <tt>body</tt> elements - it should
206
 
just be the contents of a body element (IVLE will wrap it accordingly).</p>
207
 
<p>This will automatically be used as the license text instead of this
208
 
placeholder text.</p>
209
 
""")
210
 
 
211
 
def parse_iso8601(str):
212
 
    """Parses ISO8601 string into a datetime object."""
213
 
    # FIXME: Terrific hack that means we only accept the format that is 
214
 
    # produced by json.js module when it encodes date objects.
215
 
    return datetime.datetime.strptime(str, "%Y-%m-%dT%H:%M:%SZ")
216
 
 
217
 
def incomplete_utf8_sequence(byteseq):
218
 
    """
219
 
    str -> int
220
 
    Given a UTF-8-encoded byte sequence (str), returns the number of bytes at
221
 
    the end of the string which comprise an incomplete UTF-8 character
222
 
    sequence.
223
 
 
224
 
    If the string is empty or ends with a complete character OR INVALID
225
 
    sequence, returns 0.
226
 
    Otherwise, returns 1-3 indicating the number of bytes in the final
227
 
    incomplete (but valid) character sequence.
228
 
 
229
 
    Does not check any bytes before the final sequence for correctness.
230
 
 
231
 
    >>> incomplete_utf8_sequence("")
232
 
    0
233
 
    >>> incomplete_utf8_sequence("xy")
234
 
    0
235
 
    >>> incomplete_utf8_sequence("xy\xc3\xbc")
236
 
    0
237
 
    >>> incomplete_utf8_sequence("\xc3")
238
 
    1
239
 
    >>> incomplete_utf8_sequence("\xbc\xc3")
240
 
    1
241
 
    >>> incomplete_utf8_sequence("xy\xbc\xc3")
242
 
    1
243
 
    >>> incomplete_utf8_sequence("xy\xe0\xa0")
244
 
    2
245
 
    >>> incomplete_utf8_sequence("xy\xf4")
246
 
    1
247
 
    >>> incomplete_utf8_sequence("xy\xf4\x8f")
248
 
    2
249
 
    >>> incomplete_utf8_sequence("xy\xf4\x8f\xa0")
250
 
    3
251
 
    """
252
 
    count = 0
253
 
    expect = None
254
 
    for b in byteseq[::-1]:
255
 
        b = ord(b)
256
 
        count += 1
257
 
        if b & 0x80 == 0x0:
258
 
            # 0xxxxxxx (single-byte character)
259
 
            expect = 1
260
 
            break
261
 
        elif b & 0xc0 == 0x80:
262
 
            # 10xxxxxx (subsequent byte)
263
 
            pass
264
 
        elif b & 0xe0 == 0xc0:
265
 
            # 110xxxxx (start of 2-byte sequence)
266
 
            expect = 2
267
 
            break
268
 
        elif b & 0xf0 == 0xe0:
269
 
            # 1110xxxx (start of 3-byte sequence)
270
 
            expect = 3
271
 
            break
272
 
        elif b & 0xf8 == 0xf0:
273
 
            # 11110xxx (start of 4-byte sequence)
274
 
            expect = 4
275
 
            break
276
 
        else:
277
 
            # Invalid byte
278
 
            return 0
279
 
 
280
 
        if count >= 4:
281
 
            # Seen too many "subsequent bytes", invalid
282
 
            return 0
283
 
 
284
 
    if expect is None:
285
 
        # We never saw a "first byte", invalid
286
 
        return 0
287
 
 
288
 
    # We now know expect and count
289
 
    if count >= expect:
290
 
        # Complete, or we saw an invalid sequence
291
 
        return 0
292
 
    elif count < expect:
293
 
        # Incomplete
294
 
        return count