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

657 by drtomc
serve: use the trampoline to serve all files.
1
#!/usr/bin/python
2
3
# IVLE - Informatics Virtual Learning Environment
4
# Copyright (C) 2007-2008 The University of Melbourne
5
#
6
# This program is free software; you can redistribute it and/or modify
7
# it under the terms of the GNU General Public License as published by
8
# the Free Software Foundation; either version 2 of the License, or
9
# (at your option) any later version.
10
#
11
# This program is distributed in the hope that it will be useful,
12
# but WITHOUT ANY WARRANTY; without even the implied warranty of
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
# GNU General Public License for more details.
15
#
16
# You should have received a copy of the GNU General Public License
17
# along with this program; if not, write to the Free Software
18
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
20
# Script: serveservice
21
# Author: Thomas Conway
22
# Date:   6/3/2007
23
24
# A CGI script for serving files.
25
26
import mimetypes
27
import os
672 by mattgiuca
serveservice: Added missing "import conf".
28
import conf
720 by dcoles
download: Fixing the download button.
29
import StringIO
30
import urlparse
657 by drtomc
serve: use the trampoline to serve all files.
31
720 by dcoles
download: Fixing the download button.
32
from common import (cgirequest, studpath)
33
from common import zip as zipmod
657 by drtomc
serve: use the trampoline to serve all files.
34
35
req = cgirequest.CGIRequest()
720 by dcoles
download: Fixing the download button.
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,
722 by dcoles
s/serveservice: Show the url rather than the local path on not-found error
67
            "The specified file (%s) does not exist." % urlpath)
720 by dcoles
download: Fixing the download button.
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:
721 by dcoles
download/serve: Removed the setting of mime-type here. If it's a download file
102
    #req.content_type = default_mimetype
720 by dcoles
download: Fixing the download button.
103
    req.sendfile(filename)