Created
December 4, 2016 08:55
-
-
Save Garrett-R/9a107e057ff8864b40aff59f7d6f6a68 to your computer and use it in GitHub Desktop.
Python 3 version of http://flask.pocoo.org/snippets/122/
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from flask import after_this_request, request | |
from io import BytesIO as IO | |
import gzip | |
import functools | |
def gzipped(f): | |
@functools.wraps(f) | |
def view_func(*args, **kwargs): | |
@after_this_request | |
def zipper(response): | |
accept_encoding = request.headers.get('Accept-Encoding', '') | |
if 'gzip' not in accept_encoding.lower(): | |
return response | |
response.direct_passthrough = False | |
if (response.status_code < 200 or | |
response.status_code >= 300 or | |
'Content-Encoding' in response.headers): | |
return response | |
gzip_buffer = IO() | |
gzip_file = gzip.GzipFile(mode='wb', | |
fileobj=gzip_buffer) | |
gzip_file.write(response.data) | |
gzip_file.close() | |
response.data = gzip_buffer.getvalue() | |
response.headers['Content-Encoding'] = 'gzip' | |
response.headers['Vary'] = 'Accept-Encoding' | |
response.headers['Content-Length'] = len(response.data) | |
return response | |
return f(*args, **kwargs) | |
return view_func |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment