From 707a3a82c3956b2cf935e88c5b6c7993a730c1b9 Mon Sep 17 00:00:00 2001 From: Joe Kaufeld Date: Tue, 29 Oct 2024 23:50:07 -0400 Subject: [PATCH] =?UTF-8?q?=F0=9F=8F=B7=EF=B8=8F=20fix=20types=20to=20be?= =?UTF-8?q?=20explicit=20about=20bytes=20being=20okay?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- spiderweb/middleware/__init__.py | 2 +- spiderweb/middleware/gzip.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/spiderweb/middleware/__init__.py b/spiderweb/middleware/__init__.py index 9118034..9ebff28 100644 --- a/spiderweb/middleware/__init__.py +++ b/spiderweb/middleware/__init__.py @@ -63,7 +63,7 @@ class MiddlewareMixin: def post_process_middleware( self, request: Request, response: HttpResponse, rendered_response: str - ) -> str: + ) -> str | bytes: # run them in reverse order, same as process_response. The top of the middleware # stack should be the first and last middleware to run. for middleware in reversed(self.middleware): diff --git a/spiderweb/middleware/gzip.py b/spiderweb/middleware/gzip.py index 00982b1..6610a6d 100644 --- a/spiderweb/middleware/gzip.py +++ b/spiderweb/middleware/gzip.py @@ -41,7 +41,7 @@ class GzipMiddleware(SpiderwebMiddleware): def post_process( self, request: Request, response: HttpResponse, rendered_response: str - ) -> str: + ) -> str | bytes: # Only actually compress the response if the following attributes are true: # # * The response status code is a 2xx success code