Decorator to gzip Respone payload

gistfile1.txt

"""Decorator to gzip Respone payload

	Obtained from http://kb.sites.apiit.edu.my/knowledge-base/how-to-gzip-response-in-flask/

	Example use case:

	@gzipped
	def get_data():
		return response
"""

import logging
import gzip
import functools
from flask import after_this_request, request

LOGGER = logging.getLogger(__name__)

def gzipped(func):
	@functools.wraps(func)
	def view_func(*args, **kwargs):
		@after_this_request
		def _(resp):
			if cannot_zip(resp):
				return resp

			accept_encoding = request.headers.get('Accept-Encoding', '')

			if 'gzip' not in accept_encoding.lower():
				return resp

			resp.direct_passthrough = False
			resp.data = gzip.compress(resp.data)
			resp.headers['Content-Encoding'] = 'gzip'
			resp.headers['Content-Length'] = len(resp.data)
			return resp

		return func(*args, **kwargs)

	return view_func


def cannot_zip(resp):
	return (resp.status_code < 200 or
			resp.status_code >= 300 or
			resp.status_code == 204 or  # no content
			'Content-Encoding' in resp.headers)