mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 18:32:48 +00:00
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: MaxKB
|
||
@Author：虎
|
||
@file： gzip.py
|
||
@date：2025/2/27 10:03
|
||
@desc:
|
||
"""
|
||
from django.utils.cache import patch_vary_headers
|
||
from django.utils.deprecation import MiddlewareMixin
|
||
from django.utils.regex_helper import _lazy_re_compile
|
||
from django.utils.text import compress_sequence, compress_string
|
||
|
||
re_accepts_gzip = _lazy_re_compile(r"\bgzip\b")
|
||
|
||
|
||
class GZipMiddleware(MiddlewareMixin):
|
||
"""
|
||
Compress content if the browser allows gzip compression.
|
||
Set the Vary header accordingly, so that caches will base their storage
|
||
on the Accept-Encoding header.
|
||
"""
|
||
|
||
max_random_bytes = 100
|
||
|
||
def process_response(self, request, response):
|
||
if request.method != 'GET' or request.path.startswith('/api'):
|
||
return response
|
||
# It's not worth attempting to compress really short responses.
|
||
if not response.streaming and len(response.content) < 200:
|
||
return response
|
||
|
||
# Avoid gzipping if we've already got a content-encoding.
|
||
if response.has_header("Content-Encoding"):
|
||
return response
|
||
|
||
patch_vary_headers(response, ("Accept-Encoding",))
|
||
|
||
ae = request.META.get("HTTP_ACCEPT_ENCODING", "")
|
||
if not re_accepts_gzip.search(ae):
|
||
return response
|
||
|
||
if response.streaming:
|
||
if response.is_async:
|
||
# pull to lexical scope to capture fixed reference in case
|
||
# streaming_content is set again later.
|
||
original_iterator = response.streaming_content
|
||
|
||
async def gzip_wrapper():
|
||
async for chunk in original_iterator:
|
||
yield compress_string(
|
||
chunk,
|
||
max_random_bytes=self.max_random_bytes,
|
||
)
|
||
|
||
response.streaming_content = gzip_wrapper()
|
||
else:
|
||
response.streaming_content = compress_sequence(
|
||
response.streaming_content,
|
||
max_random_bytes=self.max_random_bytes,
|
||
)
|
||
# Delete the `Content-Length` header for streaming content, because
|
||
# we won't know the compressed size until we stream it.
|
||
del response.headers["Content-Length"]
|
||
else:
|
||
# Return the compressed content only if it's actually shorter.
|
||
compressed_content = compress_string(
|
||
response.content,
|
||
max_random_bytes=self.max_random_bytes,
|
||
)
|
||
if len(compressed_content) >= len(response.content):
|
||
return response
|
||
response.content = compressed_content
|
||
response.headers["Content-Length"] = str(len(response.content))
|
||
|
||
# If there is a strong ETag, make it weak to fulfill the requirements
|
||
# of RFC 9110 Section 8.8.1 while also allowing conditional request
|
||
# matches on ETags.
|
||
etag = response.get("ETag")
|
||
if etag and etag.startswith('"'):
|
||
response.headers["ETag"] = "W/" + etag
|
||
response.headers["Content-Encoding"] = "gzip"
|
||
|
||
return response
|