← Back to branch summary

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

« back to all changes in this revision

Viewing changes to services/serveservice

  • Committer: mattgiuca
  • Date: 2008-07-21 05:25:48 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:920
userdb: Added to table "offering" fields max_students_per_group and
    max_groups_per_student.

Show diffs side-by-side

added added

removed removed

Lines of Context:
services/serveservice
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
 
# Author: Thomas Conway, Will Grant
21
 
 
22
 
import mimetypes
23
 
import os
24
 
import sys
25
 
import StringIO
26
 
from optparse import OptionParser
27
 
 
28
 
import cjson
29
 
 
30
 
from ivle import zip as zipmod
31
 
import ivle.conf.app.server
32
 
 
33
 
def determine_file_type(filename):
34
 
    filetype = mimetypes.guess_type(filename)[0]
35
 
    if filetype is None:
36
 
         filetype = ivle.conf.mimetypes.default_mimetype
37
 
    return filetype
38
 
 
39
 
def throw_error(message, extra={}):
40
 
    error = {'error': message}
41
 
    error.update(extra)
42
 
    print cjson.encode(error)
43
 
    sys.exit(0)
44
 
 
45
 
parser = OptionParser()
46
 
parser.add_option('-d', '--download', dest='download', action='store_true',
47
 
                  help='force download, not execution, of the paths')
48
 
(options, args) = parser.parse_args()
49
 
 
50
 
# Detect download mode. Download mode zips any multiple selection,
51
 
# and does not execute CGI scripts.
52
 
if options.download:
53
 
    download = True
54
 
    # paths is filled later.
55
 
else:
56
 
    download = False
57
 
    assert len(args) == 1
58
 
 
59
 
default_mimetype = "application/octet-stream"
60
 
zip_mimetype = "application/zip"
61
 
 
62
 
zipmode = False
63
 
zipbasepath = None
64
 
zipfilename = None
65
 
 
66
 
# If multiple paths have been specified, zip them up.
67
 
if len(args) > 1:
68
 
    # Mangle the paths - we want the basename of their dirname in front.
69
 
    paths = []
70
 
    dir = os.path.dirname(args[0])
71
 
    for path in args:
72
 
        assert os.path.dirname(path) == dir
73
 
        paths.append(os.path.join(os.path.basename(dir),
74
 
                                  os.path.basename(path)))
75
 
    dir = os.path.dirname(dir)
76
 
    zipmode = True
77
 
    zipbasepath = dir
78
 
    zipfilename = os.path.basename(zipbasepath)
79
 
else:
80
 
    paths = args
81
 
    filename = paths[0]
82
 
    if not os.access(filename, os.F_OK):
83
 
        # The given path doesn't exist. CGI lets us backtrack and put the path
84
 
        # elements through which we pass into PATH_INFO, so we try that.
85
 
        while not os.access(filename, os.F_OK):
86
 
            filename, path_info_frag = os.path.split(filename)
87
 
 
88
 
        # We now have a file that exists, but is it something that we're allowed
89
 
        # to execute? If not, we should 404 anyway.
90
 
        if determine_file_type(filename) not in ivle.conf.app.server.interpreters:
91
 
            throw_error('not-found')
92
 
 
93
 
    # If it's a directory, serve as a zip file
94
 
    if os.path.isdir(filename):
95
 
        if not download:
96
 
            # Not giving a directory listing - this is visible to everyone.
97
 
            throw_error('is-directory')
98
 
        zipmode = True
99
 
        # Zip it from the perspective of its own parent.
100
 
        # That way it will be a directory in the top level of the zip
101
 
        # file.
102
 
        if filename[-1] == os.sep: filename = filename[:-1]
103
 
        splitpath = filename.rsplit(os.sep, 1)
104
 
        if len(splitpath) == 1:
105
 
            zipbasepath = ''
106
 
            paths = [filename]
107
 
        else:
108
 
            zipbasepath = splitpath[0]
109
 
            paths = [splitpath[1]]
110
 
        zipfilename = filename
111
 
    else:
112
 
        if not download and \
113
 
           determine_file_type(filename) in ivle.conf.app.server.interpreters:
114
 
            throw_error('is-executable', {'path': filename})
115
 
 
116
 
        if (ivle.conf.app.server.blacklist_served_filetypes and \
117
 
                determine_file_type(filename) in \
118
 
                ivle.conf.app.server.served_filetypes_blacklist) or \
119
 
            (ivle.conf.app.server.served_filetypes_whitelist and \
120
 
                determine_file_type(filename) not in \
121
 
                ivle.conf.app.server.served_filetypes_whitelist):
122
 
            throw_error('forbidden')
123
 
 
124
 
if zipmode:
125
 
    # zipfilename is some filename. Strip trailing slash or extension,
126
 
    # and add ".zip".
127
 
    if zipfilename == '':
128
 
        zipfilename = "files"
129
 
    elif zipfilename[-1] == '/':
130
 
        zipfilename = zipfilename[:-1]
131
 
    elif '.' in zipfilename:
132
 
        zipfilename = zipfilename[:zipfilename.rindex('.')]
133
 
    zipfilename += ".zip"
134
 
    #req.headers_out["Content-Disposition"] = ("attachment; filename=" +   
135
 
    #    zipfilename) # TODO
136
 
    zipfile = StringIO.StringIO()
137
 
    zipmod.make_zip(zipbasepath, paths, zipfile)
138
 
 
139
 
    print cjson.encode({'type': zip_mimetype,
140
 
                        'name': zipfilename})
141
 
    print zipfile.getvalue()
142
 
else:
143
 
    print cjson.encode({'type': determine_file_type(filename),
144
 
                        'name': os.path.basename(filename)})
145
 
    print open(filename).read()