133 lines
4.6 KiB
Python
133 lines
4.6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
| This file is part of the web2py Web Framework
|
|
| Copyrighted by Massimo Di Pierro <mdipierro@cs.depaul.edu>
|
|
| License: LGPLv3 (http://www.gnu.org/licenses/lgpl.html)
|
|
|
|
Facilities to handle file streaming
|
|
------------------------------------
|
|
"""
|
|
|
|
import os
|
|
import stat
|
|
import time
|
|
import re
|
|
import errno
|
|
from gluon.http import HTTP
|
|
from gluon.contenttype import contenttype
|
|
from gluon._compat import PY2
|
|
|
|
|
|
regex_start_range = re.compile('\d+(?=\-)')
|
|
regex_stop_range = re.compile('(?<=\-)\d+')
|
|
|
|
DEFAULT_CHUNK_SIZE = 64 * 1024
|
|
|
|
def streamer(stream, chunk_size=DEFAULT_CHUNK_SIZE, bytes=None, callback=None):
|
|
try:
|
|
offset = 0
|
|
while bytes is None or offset < bytes:
|
|
if not bytes is None and bytes - offset < chunk_size:
|
|
chunk_size = bytes - offset
|
|
data = stream.read(chunk_size)
|
|
length = len(data)
|
|
if not length:
|
|
break
|
|
else:
|
|
yield data
|
|
if length < chunk_size:
|
|
break
|
|
offset += length
|
|
finally:
|
|
stream.close()
|
|
if callback:
|
|
callback()
|
|
|
|
def stream_file_or_304_or_206(
|
|
static_file,
|
|
chunk_size=DEFAULT_CHUNK_SIZE,
|
|
request=None,
|
|
headers={},
|
|
status=200,
|
|
error_message=None
|
|
):
|
|
# FIX THIS
|
|
# if error_message is None:
|
|
# error_message = rewrite.THREAD_LOCAL.routes.error_message % 'invalid request'
|
|
try:
|
|
if PY2:
|
|
open_f = file # this makes no sense but without it GAE cannot open files
|
|
else:
|
|
open_f = open
|
|
fp = open_f(static_file,'rb')
|
|
except IOError as e:
|
|
if e.errno == errno.EISDIR:
|
|
raise HTTP(403, error_message, web2py_error='file is a directory')
|
|
elif e.errno == errno.EACCES:
|
|
raise HTTP(403, error_message, web2py_error='inaccessible file')
|
|
else:
|
|
raise HTTP(404, error_message, web2py_error='invalid file')
|
|
else:
|
|
fp.close()
|
|
stat_file = os.stat(static_file)
|
|
fsize = stat_file[stat.ST_SIZE]
|
|
modified = stat_file[stat.ST_MTIME]
|
|
mtime = time.strftime('%a, %d %b %Y %H:%M:%S GMT', time.gmtime(modified))
|
|
headers.setdefault('Content-Type', contenttype(static_file))
|
|
headers.setdefault('Last-Modified', mtime)
|
|
headers.setdefault('Pragma', 'cache')
|
|
headers.setdefault('Cache-Control', 'private')
|
|
|
|
# if this is a normal response and not a respnse to an error page
|
|
if status == 200:
|
|
if request and request.env.http_if_modified_since == mtime:
|
|
raise HTTP(304, **{'Content-Type': headers['Content-Type']})
|
|
|
|
elif request and request.env.http_range:
|
|
start_items = regex_start_range.findall(request.env.http_range)
|
|
if not start_items:
|
|
start_items = [0]
|
|
stop_items = regex_stop_range.findall(request.env.http_range)
|
|
if not stop_items or int(stop_items[0]) > fsize - 1:
|
|
stop_items = [fsize - 1]
|
|
part = (int(start_items[0]), int(stop_items[0]), fsize)
|
|
bytes = part[1] - part[0] + 1
|
|
try:
|
|
stream = open(static_file, 'rb')
|
|
except IOError as e:
|
|
if e.errno in (errno.EISDIR, errno.EACCES):
|
|
raise HTTP(403)
|
|
else:
|
|
raise HTTP(404)
|
|
stream.seek(part[0])
|
|
headers['Content-Range'] = 'bytes %i-%i/%i' % part
|
|
headers['Content-Length'] = '%i' % bytes
|
|
status = 206
|
|
# in all the other cases (not 304, not 206, but 200 or error page)
|
|
if status != 206:
|
|
enc = request.env.http_accept_encoding
|
|
if enc and 'gzip' in enc and not 'Content-Encoding' in headers:
|
|
gzipped = static_file + '.gz'
|
|
if os.path.isfile(gzipped) and os.path.getmtime(gzipped) >= modified:
|
|
static_file = gzipped
|
|
fsize = os.path.getsize(gzipped)
|
|
headers['Content-Encoding'] = 'gzip'
|
|
headers['Vary'] = 'Accept-Encoding'
|
|
try:
|
|
stream = open(static_file, 'rb')
|
|
except IOError as e:
|
|
# this better not happen when returning an error page ;-)
|
|
if e.errno in (errno.EISDIR, errno.EACCES):
|
|
raise HTTP(403)
|
|
else:
|
|
raise HTTP(404)
|
|
headers['Content-Length'] = fsize
|
|
bytes = None
|
|
if request and request.env.web2py_use_wsgi_file_wrapper:
|
|
wrapped = request.env.wsgi_file_wrapper(stream, chunk_size)
|
|
else:
|
|
wrapped = streamer(stream, chunk_size=chunk_size, bytes=bytes)
|
|
raise HTTP(status, wrapped, **headers)
|