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

« back to all changes in this revision

Viewing changes to www/apps/fileservice/action.py

  • Committer: drtomc
  • Date: 2008-01-22 04:07:52 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:264
Fix the "Aaarrgghh! ^C doesn't work problem". The multiple console process
problem should go away too. Next: run as a proper daemon.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# IVLE
 
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: File Service / Action
 
19
# Author: Matt Giuca
 
20
# Date: 10/1/2008
 
21
 
 
22
# Handles actions requested by the client as part of the 2-stage process of
 
23
# fileservice (the second part being the return listing).
 
24
 
 
25
### Actions ###
 
26
 
 
27
# The most important argument is "action". This determines which action is
 
28
# taken. Note that action, and all other arguments, are ignored unless the
 
29
# request is a POST request. The other arguments depend upon the action.
 
30
# Note that paths are often specified as arguments. Paths that begin with a
 
31
# slash are taken relative to the user's home directory (the top-level
 
32
# directory seen when fileservice has no arguments or path). Paths without a
 
33
# slash are taken relative to the specified path.
 
34
 
 
35
# action=remove: Delete a file(s) or directory(s) (recursively).
 
36
#       path:   The path to the file or directory to delete. Can be specified
 
37
#               multiple times.
 
38
#
 
39
# action=move: Move or rename a file or directory.
 
40
#       from:   The path to the file or directory to be renamed.
 
41
#       to:     The path of the target filename. Error if the file already
 
42
#               exists.
 
43
#
 
44
# action=putfile: Upload a file to the student workspace, and optionally
 
45
#               accept zip files which will be unpacked.
 
46
#       path:   The path to the file to be written. If it exists, will
 
47
#               overwrite. Error if the target file is a directory.
 
48
#       data:   Bytes to be written to the file verbatim. May either be
 
49
#               a string variable or a file upload.
 
50
#       unpack: Optional. If "true", and the data is a valid ZIP file,
 
51
#               will create a directory instead and unpack the ZIP file
 
52
#               into it.
 
53
#
 
54
# action=putfiles: Upload multiple files to the student workspace, and
 
55
#                 optionally accept zip files which will be unpacked.
 
56
#       path:   The path to the DIRECTORY to place files in. Must not be a
 
57
#               file.
 
58
#       data:   A file upload (may not be a simple string). The filename
 
59
#               will be used to determine the target filename within
 
60
#               the given path.
 
61
#       unpack: Optional. If "true", if any data is a valid ZIP file,
 
62
#               will create a directory instead and unpack the ZIP file
 
63
#               into it.
 
64
#
 
65
# The differences between putfile and putfiles are:
 
66
# * putfile can only accept a single file.
 
67
# * putfile can accept string data, doesn't have to be a file upload.
 
68
# * putfile ignores the upload filename, the entire filename is specified on
 
69
#       path. putfiles calls files after the name on the user's machine.
 
70
#
 
71
# Clipboard-based actions. Cut/copy/paste work in the same way as modern
 
72
# file browsers, by keeping a server-side clipboard of files that have been
 
73
# cut and copied. The clipboard is stored in the session data, so it persists
 
74
# across navigation, tabs and browser windows, but not across browser
 
75
# sessions.
 
76
 
77
# action=copy: Write file(s) to the session-based clipboard. Overrides any
 
78
#               existing clipboard data. Does not actually copy the file.
 
79
#               The files are physically copied when the clipboard is pasted.
 
80
#       path:   The path to the file or directory to copy. Can be specified
 
81
#               multiple times.
 
82
 
83
# action=cut: Write file(s) to the session-based clipboard. Overrides any
 
84
#               existing clipboard data. Does not actually move the file.
 
85
#               The files are physically moved when the clipboard is pasted.
 
86
#       path:   The path to the file or directory to cut. Can be specified
 
87
#               multiple times.
 
88
 
89
# action=paste: Copy or move the files stored in the clipboard. Clears the
 
90
#               clipboard. The files are copied or moved to a specified dir.
 
91
#       path:   The path to the DIRECTORY to paste the files to. Must not
 
92
#               be a file.
 
93
#
 
94
# Subversion actions.
 
95
# action=svnadd: Add an existing file(s) to version control.
 
96
#       path:   The path to the file to be added. Can be specified multiple
 
97
#               times.
 
98
#
 
99
# action=svnrevert: Revert a file(s) to its state as of the current revision
 
100
#               / undo local edits.
 
101
#       path:   The path to the file to be reverted. Can be specified multiple
 
102
#               times.
 
103
#
 
104
# action=svnupdate: Bring a file up to date with the head revision.
 
105
#       path:   The path to the file to be updated. Only one file may be
 
106
#               specified.
 
107
#
 
108
# action=svncommit: Commit a file(s) or directory(s) to the repository.
 
109
#       path:   The path to the file or directory to be committed. Can be
 
110
#               specified multiple times. Directories are committed
 
111
#               recursively.
 
112
#       logmsg: Text of the log message. Optional. There is a default log
 
113
#               message if unspecified.
 
114
 
115
# TODO: Implement the following actions:
 
116
#   putfiles, svnrevert, svnupdate, svncommit
 
117
# TODO: Implement ZIP unpacking in putfile and putfiles.
 
118
# TODO: svnupdate needs a digest to tell the user the files that were updated.
 
119
#   This can be implemented by some message passing between action and
 
120
#   listing, and having the digest included in the listing. (Problem if
 
121
#   the listing is not a directory, but we could make it an error to do an
 
122
#   update if the path is not a directory).
 
123
 
 
124
import os
 
125
import shutil
 
126
 
 
127
import pysvn
 
128
 
 
129
from common import (util, studpath)
 
130
 
 
131
# Make a Subversion client object
 
132
svnclient = pysvn.Client()
 
133
 
 
134
DEFAULT_LOGMESSAGE = "No log message supplied."
 
135
 
 
136
# Mime types
 
137
# application/json is the "best" content type but is not good for
 
138
# debugging because Firefox just tries to download it
 
139
mime_dirlisting = "text/html"
 
140
#mime_dirlisting = "application/json"
 
141
 
 
142
class ActionError(Exception):
 
143
    """Represents an error processing an action. This can be
 
144
    raised by any of the action functions, and will be caught
 
145
    by the top-level handler, put into the HTTP response field,
 
146
    and continue.
 
147
 
 
148
    Important Security Consideration: The message passed to this
 
149
    exception will be relayed to the client.
 
150
    """
 
151
    pass
 
152
 
 
153
def handle_action(req, action, fields):
 
154
    """Perform the "action" part of the response.
 
155
    This function should only be called if the response is a POST.
 
156
    This performs the action's side-effect on the server. If unsuccessful,
 
157
    writes the X-IVLE-Action-Error header to the request object. Otherwise,
 
158
    does not touch the request object. Does NOT write any bytes in response.
 
159
 
 
160
    May throw an ActionError. The caller should put this string into the
 
161
    X-IVLE-Action-Error header, and then continue normally.
 
162
 
 
163
    action: String, the action requested. Not sanitised.
 
164
    fields: FieldStorage object containing all arguments passed.
 
165
    """
 
166
    global actions_table        # Table of function objects
 
167
    try:
 
168
        action = actions_table[action]
 
169
    except KeyError:
 
170
        # Default, just send an error but then continue
 
171
        raise ActionError("Unknown action")
 
172
    action(req, fields)
 
173
 
 
174
def actionpath_to_local(req, path):
 
175
    """Determines the local path upon which an action is intended to act.
 
176
    Note that fileservice actions accept two paths: the request path,
 
177
    and the "path" argument given to the action.
 
178
    According to the rules, if the "path" argument begins with a '/' it is
 
179
    relative to the user's home; if it does not, it is relative to the
 
180
    supplied path.
 
181
 
 
182
    This resolves the path, given the request and path argument.
 
183
 
 
184
    May raise an ActionError("Invalid path"). The caller is expected to
 
185
    let this fall through to the top-level handler, where it will be
 
186
    put into the HTTP response field. Never returns None.
 
187
 
 
188
    Does not mutate req.
 
189
    """
 
190
    if path is None:
 
191
        path = req.path
 
192
    elif len(path) > 0 and path[0] == os.sep:
 
193
        # Relative to student home
 
194
        path = path[1:]
 
195
    else:
 
196
        # Relative to req.path
 
197
        path = os.path.join(req.path, path)
 
198
 
 
199
    _, r = studpath.url_to_local(path)
 
200
    if r is None:
 
201
        raise ActionError("Invalid path")
 
202
    return r
 
203
 
 
204
def movefile(req, frompath, topath, copy=False):
 
205
    """Performs a file move, resolving filenames, checking for any errors,
 
206
    and throwing ActionErrors if necessary. Can also be used to do a copy
 
207
    operation instead.
 
208
 
 
209
    frompath and topath are straight paths from the client. Will be checked.
 
210
    """
 
211
    # TODO: Do an SVN mv if the file is versioned.
 
212
    # TODO: Disallow tampering with student's home directory
 
213
    if frompath is None or topath is None:
 
214
        raise ActionError("Required field missing")
 
215
    frompath = actionpath_to_local(req, frompath)
 
216
    topath = actionpath_to_local(req, topath)
 
217
    if not os.path.exists(frompath):
 
218
        raise ActionError("The source file does not exist")
 
219
    if os.path.exists(topath):
 
220
        if frompath == topath:
 
221
            raise ActionError("Source and destination are the same")
 
222
        raise ActionError("Another file already exists with that name")
 
223
 
 
224
    try:
 
225
        if copy:
 
226
            if os.path.isdir(frompath):
 
227
                shutil.copytree(frompath, topath)
 
228
            else:
 
229
                shutil.copy2(frompath, topath)
 
230
        else:
 
231
            shutil.move(frompath, topath)
 
232
    except OSError:
 
233
        raise ActionError("Could not move the file specified")
 
234
    except shutil.Error:
 
235
        raise ActionError("Could not move the file specified")
 
236
 
 
237
### ACTIONS ###
 
238
 
 
239
def action_remove(req, fields):
 
240
    # TODO: Do an SVN rm if the file is versioned.
 
241
    # TODO: Disallow removal of student's home directory
 
242
    """Removes a list of files or directories.
 
243
 
 
244
    Reads fields: 'path' (multiple)
 
245
    """
 
246
    paths = fields.getlist('path')
 
247
    goterror = False
 
248
    for path in paths:
 
249
        path = actionpath_to_local(req, path)
 
250
        try:
 
251
            if os.path.isdir(path):
 
252
                shutil.rmtree(path)
 
253
            else:
 
254
                os.remove(path)
 
255
        except OSError:
 
256
            goterror = True
 
257
        except shutil.Error:
 
258
            goterror = True
 
259
    if goterror:
 
260
        if len(paths) == 1:
 
261
            raise ActionError("Could not delete the file specified")
 
262
        else:
 
263
            raise ActionError(
 
264
                "Could not delete one or more of the files specified")
 
265
 
 
266
def action_move(req, fields):
 
267
    # TODO: Do an SVN mv if the file is versioned.
 
268
    # TODO: Disallow tampering with student's home directory
 
269
    """Removes a list of files or directories.
 
270
 
 
271
    Reads fields: 'from', 'to'
 
272
    """
 
273
    frompath = fields.getfirst('from')
 
274
    topath = fields.getfirst('to')
 
275
    movefile(req, frompath, topath)
 
276
 
 
277
def action_putfile(req, fields):
 
278
    """Writes data to a file, overwriting it if it exists and creating it if
 
279
    it doesn't.
 
280
 
 
281
    Reads fields: 'path', 'data' (file upload)
 
282
    """
 
283
    # Important: Data is "None" if the file submitted is empty.
 
284
    path = fields.getfirst('path')
 
285
    data = fields.getfirst('data')
 
286
    if path is None:
 
287
        raise ActionError("Required field missing")
 
288
    path = actionpath_to_local(req, path)
 
289
    if data is not None:
 
290
        data = data.file
 
291
 
 
292
    # Copy the contents of file object 'data' to the path 'path'
 
293
    try:
 
294
        dest = open(path, 'wb')
 
295
        if data is not None:
 
296
            shutil.copyfileobj(data, dest)
 
297
    except OSError:
 
298
        raise ActionError("Could not write to target file")
 
299
 
 
300
def action_copy_or_cut(req, fields, mode):
 
301
    """Marks specified files on the clipboard, stored in the
 
302
    browser session. Sets clipboard for either a cut or copy operation
 
303
    as specified.
 
304
 
 
305
    Reads fields: 'path'
 
306
    """
 
307
    # The clipboard object created conforms to the JSON clipboard
 
308
    # specification given at the top of listing.py.
 
309
    # Note that we do not check for the existence of files here. That is done
 
310
    # in the paste operation.
 
311
    files = fields.getlist('path')
 
312
    files = map(lambda field: field.value, files)
 
313
    clipboard = { "mode" : mode, "base" : req.path, "files" : files }
 
314
    session = req.get_session()
 
315
    session['clipboard'] = clipboard
 
316
    session.save()
 
317
 
 
318
def action_copy(req, fields):
 
319
    """Marks specified files on the clipboard, stored in the
 
320
    browser session. Sets clipboard for a "copy" action.
 
321
 
 
322
    Reads fields: 'path'
 
323
    """
 
324
    action_copy_or_cut(req, fields, "copy")
 
325
 
 
326
def action_cut(req, fields):
 
327
    """Marks specified files on the clipboard, stored in the
 
328
    browser session. Sets clipboard for a "cut" action.
 
329
 
 
330
    Reads fields: 'path'
 
331
    """
 
332
    action_copy_or_cut(req, fields, "cut")
 
333
 
 
334
def action_paste(req, fields):
 
335
    """Performs the copy or move action with the files stored on
 
336
    the clipboard in the browser session. Copies/moves the files
 
337
    to the specified directory. Clears the clipboard.
 
338
 
 
339
    Reads fields: 'path'
 
340
    """
 
341
    errormsg = None
 
342
 
 
343
    todir = fields.getfirst('path')
 
344
    if todir is None:
 
345
        raise ActionError("Required field missing")
 
346
    todir_local = actionpath_to_local(req, todir)
 
347
    if not os.path.isdir(todir_local):
 
348
        raise ActionError("Target is not a directory")
 
349
 
 
350
    session = req.get_session()
 
351
    try:
 
352
        clipboard = session['clipboard']
 
353
        files = clipboard['files']
 
354
        base = clipboard['base']
 
355
        if clipboard['mode'] == "copy":
 
356
            copy = True
 
357
        else:
 
358
            copy = False
 
359
    except KeyError:
 
360
        raise ActionError("Clipboard was empty")
 
361
 
 
362
    errorfiles = []
 
363
    for file in files:
 
364
        # The source must not be interpreted as relative to req.path
 
365
        # Add a slash (relative to top-level)
 
366
        frompath = os.sep + os.path.join(base, file)
 
367
        # The destination is found by taking just the basename of the file
 
368
        topath = os.path.join(todir, os.path.basename(file))
 
369
        try:
 
370
            movefile(req, frompath, topath, copy)
 
371
        except ActionError, message:
 
372
            # Store the error for later; we want to copy as many as possible
 
373
            if errormsg is None:
 
374
                errormsg = message
 
375
            else:
 
376
                # Multiple errors; generic message
 
377
                errormsg = "One or more files could not be pasted"
 
378
            # Add this file to errorfiles; it will be put back on the
 
379
            # clipboard for possible future pasting.
 
380
            errorfiles.append(file)
 
381
    # If errors occured, augment the clipboard and raise ActionError
 
382
    if len(errorfiles) > 0:
 
383
        clipboard['files'] = errorfiles
 
384
        session['clipboard'] = clipboard
 
385
        session.save()
 
386
        raise ActionError(errormsg)
 
387
 
 
388
    # Success: Clear the clipboard
 
389
    del session['clipboard']
 
390
    session.save()
 
391
 
 
392
def action_svnadd(req, fields):
 
393
    """Performs a "svn add" to each file specified.
 
394
 
 
395
    Reads fields: 'path' (multiple)
 
396
    """
 
397
    paths = fields.getlist('path')
 
398
    paths = map(lambda path: actionpath_to_local(req, path), paths)
 
399
 
 
400
    try:
 
401
        svnclient.add(paths, recurse=True, force=True)
 
402
    except pysvn.ClientError:
 
403
        raise ActionError("One or more files could not be added")
 
404
 
 
405
def action_svnupdate(req, fields):
 
406
    """Performs a "svn update" to each file specified.
 
407
 
 
408
    Reads fields: 'path'
 
409
    """
 
410
    path = fields.getfirst('path')
 
411
    if path is None:
 
412
        raise ActionError("Required field missing")
 
413
    path = actionpath_to_local(req, path)
 
414
 
 
415
    try:
 
416
        svnclient.update(path, recurse=True)
 
417
    except pysvn.ClientError:
 
418
        raise ActionError("One or more files could not be updated")
 
419
 
 
420
def action_svnrevert(req, fields):
 
421
    """Performs a "svn revert" to each file specified.
 
422
 
 
423
    Reads fields: 'path' (multiple)
 
424
    """
 
425
    paths = fields.getlist('path')
 
426
    paths = map(lambda path: actionpath_to_local(req, path), paths)
 
427
 
 
428
    try:
 
429
        svnclient.revert(paths, recurse=True)
 
430
    except pysvn.ClientError:
 
431
        raise ActionError("One or more files could not be reverted")
 
432
 
 
433
def action_svncommit(req, fields):
 
434
    """Performs a "svn commit" to each file specified.
 
435
 
 
436
    Reads fields: 'path' (multiple), 'logmsg' (optional)
 
437
    """
 
438
    paths = fields.getlist('path')
 
439
    paths = map(lambda path: actionpath_to_local(req, str(path)), paths)
 
440
    logmsg = str(fields.getfirst('logmsg', DEFAULT_LOGMESSAGE))
 
441
    if logmsg == '': logmsg = DEFAULT_LOGMESSAGE
 
442
 
 
443
    try:
 
444
        svnclient.checkin(paths, logmsg, recurse=True)
 
445
    except pysvn.ClientError:
 
446
        raise ActionError("One or more files could not be committed")
 
447
 
 
448
# Table of all action functions #
 
449
# Each function has the interface f(req, fields).
 
450
 
 
451
actions_table = {
 
452
    "remove" : action_remove,
 
453
    "move" : action_move,
 
454
    "putfile" : action_putfile,
 
455
 
 
456
    "copy" : action_copy,
 
457
    "cut" : action_cut,
 
458
    "paste" : action_paste,
 
459
 
 
460
    "svnadd" : action_svnadd,
 
461
    "svnupdate" : action_svnupdate,
 
462
    "svnrevert" : action_svnrevert,
 
463
    "svncommit" : action_svncommit,
 
464
}