← Back to branch summary

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

« back to all changes in this revision

Viewing changes to scripts/serveservice

  • Committer: dcoles
  • Date: 2008-03-31 01:10:43 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:720
download: Fixing the download button.
    We now run download in the jail so that we don't encounter permission 
    problems.  Since this is very similar to "serve" we now extend serveservice 
    to do the work of download and use the serve code to do download.

    w/a/s/__init__.py: Overloaded servefile to support downloads
    w/a/d/__init__.py: Use servefile from server rather than old implementaion
    s/serveservice: Added support for download of directories (as zip files)
    l/c/zip.py: Now uses paths not ivle_urls

Show diffs side-by-side

added added

removed removed

Lines of Context:
scripts/serveservice
26
26
import mimetypes
27
27
import os
28
28
import conf
 
29
import StringIO
 
30
import urlparse
29
31
 
30
 
from common import cgirequest
31
 
from common import studpath
 
32
from common import (cgirequest, studpath)
 
33
from common import zip as zipmod
32
34
 
33
35
req = cgirequest.CGIRequest()
34
 
filename = studpath.url_to_jailpaths(req.path)[2]
35
 
 
36
 
if not os.access(filename, os.R_OK):
37
 
    req.throw_error(req.HTTP_NOT_FOUND, "The specified file does not exist.")
38
 
 
39
 
(type, _) = mimetypes.guess_type(filename)
40
 
if type is None:
41
 
    type = conf.mimetypes.default_mimetype
42
 
 
43
 
req.content_type = type
44
 
req.sendfile(filename)
 
36
 
 
37
# Work out the parts of the URL
 
38
url = urlparse.urlparse(req.path)
 
39
querystr = url[4]
 
40
urlpath = url[2]
 
41
filename = studpath.url_to_jailpaths(urlpath)[2]
 
42
 
 
43
default_mimetype = "application/octet-stream"
 
44
zip_mimetype = "application/zip"
 
45
 
 
46
zipmode = False
 
47
zipbasepath = None
 
48
zipfilename = None
 
49
path = None
 
50
 
 
51
# If any "path=" variables have been supplied, bring these into a list and
 
52
# make a zip file instead.
 
53
fields = req.get_fieldstorage()
 
54
paths = fields.getlist("path")
 
55
if len(paths) > 0:
 
56
    zipmode = True
 
57
    zipbasepath = filename
 
58
    zipfilename = os.path.basename(zipbasepath)
 
59
    #for i in range(0, len(paths)):
 
60
        #paths[i] = paths[i].value
 
61
else:
 
62
    if filename is None:
 
63
        req.throw_error(req.HTTP_NOT_FOUND,
 
64
            "The path specified is invalid.")
 
65
    elif not os.access(filename, os.R_OK):
 
66
        req.throw_error(req.HTTP_NOT_FOUND,
 
67
            "The specified file (%s) does not exist." % path)
 
68
     # If it's a directory, serve as a zip file
 
69
    if os.path.isdir(filename):
 
70
        zipmode = True
 
71
        # Zip it from the perspective of its own parent.
 
72
        # That way it will be a directory in the top level of the zip
 
73
        # file.
 
74
        if filename[-1] == os.sep: filename = filename[:-1]
 
75
        splitpath = filename.rsplit(os.sep, 1)
 
76
        if len(splitpath) == 1:
 
77
            zipbasepath = ''
 
78
            paths = [filename]
 
79
        else:
 
80
            zipbasepath = splitpath[0]
 
81
            paths = [splitpath[1]]
 
82
        zipfilename = paths[0]
 
83
 
 
84
if zipmode:
 
85
    req.content_type = zip_mimetype
 
86
    # zipfilename is some filename. Strip trailing slash or extension,
 
87
    # and add ".zip".
 
88
    if zipfilename == '':
 
89
        zipfilename = "files"
 
90
    elif zipfilename[-1] == '/':
 
91
        zipfilename = zipfilename[:-1]
 
92
    elif '.' in zipfilename:
 
93
        zipfilename = zipfilename[:zipfilename.rindex('.')]
 
94
    zipfilename += ".zip"
 
95
    req.headers_out["Content-Disposition"] = ("attachment; filename=" +   
 
96
        zipfilename)
 
97
    zipfile = StringIO.StringIO()
 
98
    zipmod.make_zip(zipbasepath, paths, zipfile)
 
99
        
 
100
    req.write(zipfile.getvalue())
 
101
else:
 
102
    req.content_type = default_mimetype
 
103
    req.sendfile(filename)