voussoir/ycdl
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use voussoirkit.flasktools for gzip_response.

Browse source
This commit is contained in:
voussoir 2021-04-02 18:25:55 -07:00
parent e00bfa368b
commit 9b6af96b95
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 3 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
frontends/ycdl_flask/backend/common.py
View file

@ -3,13 +3,10 @@ Do not execute this file directly.
Use ycdl_flask_dev.py or ycdl_flask_prod.py. Use ycdl_flask_dev.py or ycdl_flask_prod.py.
''' '''
import flask; from flask import request import flask; from flask import request
import gzip
import io
import mimetypes
import os
import threading import threading
import time import time
from voussoirkit import flasktools
from voussoirkit import pathclass from voussoirkit import pathclass
import ycdl import ycdl
@ -40,7 +37,7 @@ jinja_filters.register_all(site)
site.debug = True site.debug = True
site.localhost_only = False site.localhost_only = False
#################################################################################################### # Request decorators ###############################################################################
@site.before_request @site.before_request
def before_request(): def before_request():
@ -49,37 +46,9 @@ def before_request():
if site.localhost_only and not request.is_localhost: if site.localhost_only and not request.is_localhost:
flask.abort(403) flask.abort(403)
gzip_minimum_size = 500
gzip_maximum_size = 5 * 2**20
gzip_level = 3
@site.after_request @site.after_request
def after_request(response): def after_request(response):
''' response = flasktools.gzip_response(request, response)
Thank you close.io.
https://github.com/closeio/Flask-gzip
'''
accept_encoding = request.headers.get('Accept-Encoding', '')
bail = False
bail = bail or response.status_code < 200
bail = bail or response.status_code >= 300
bail = bail or response.direct_passthrough
bail = bail or int(response.headers.get('Content-Length', 0)) > gzip_maximum_size
bail = bail or len(response.get_data()) < gzip_minimum_size
bail = bail or 'gzip' not in accept_encoding.lower()
bail = bail or 'Content-Encoding' in response.headers
if bail:
return response
gzip_buffer = io.BytesIO()
gzip_file = gzip.GzipFile(mode='wb', compresslevel=gzip_level, fileobj=gzip_buffer)
gzip_file.write(response.get_data())
gzip_file.close()
response.set_data(gzip_buffer.getvalue())
response.headers['Content-Encoding'] = 'gzip'
response.headers['Content-Length'] = len(response.get_data())
return response return response
#################################################################################################### ####################################################################################################