type: drop MiddlewareType in favour of Middleware builtin

This commit is contained in:
Evgenii Alekseev 2024-08-14 15:46:43 +03:00
parent 3353daec6d
commit 668be41c3e
3 changed files with 8 additions and 7 deletions

View File

@ -22,4 +22,3 @@ from collections.abc import Awaitable, Callable
HandlerType = Callable[[Request], Awaitable[StreamResponse]]
MiddlewareType = Callable[[Request, HandlerType], Awaitable[StreamResponse]]

View File

@ -21,6 +21,7 @@ import aiohttp_security
import socket
import types
from aiohttp.typedefs import Middleware
from aiohttp.web import Application, Request, StaticResource, StreamResponse, middleware
from aiohttp_session import setup as setup_session
from aiohttp_session.cookie_storage import EncryptedCookieStorage
@ -30,7 +31,7 @@ from enum import Enum
from ahriman.core.auth import Auth
from ahriman.core.configuration import Configuration
from ahriman.models.user_access import UserAccess
from ahriman.web.middlewares import HandlerType, MiddlewareType
from ahriman.web.middlewares import HandlerType
__all__ = ["setup_auth"]
@ -84,7 +85,7 @@ class _AuthorizationPolicy(aiohttp_security.AbstractAuthorizationPolicy):
return await self.validator.verify_access(identity, permission, context)
def _auth_handler(allow_read_only: bool) -> MiddlewareType:
def _auth_handler(allow_read_only: bool) -> Middleware:
"""
authorization and authentication middleware
@ -92,7 +93,7 @@ def _auth_handler(allow_read_only: bool) -> MiddlewareType:
allow_read_only: allow
Returns:
MiddlewareType: built middleware
Middleware: built middleware
"""
@middleware
async def handle(request: Request, handler: HandlerType) -> StreamResponse:

View File

@ -20,10 +20,11 @@
import aiohttp_jinja2
import logging
from aiohttp.typedefs import Middleware
from aiohttp.web import HTTPClientError, HTTPException, HTTPMethodNotAllowed, HTTPNoContent, HTTPServerError, \
HTTPUnauthorized, Request, StreamResponse, json_response, middleware
from ahriman.web.middlewares import HandlerType, MiddlewareType
from ahriman.web.middlewares import HandlerType
__all__ = ["exception_handler"]
@ -43,7 +44,7 @@ def _is_templated_unauthorized(request: Request) -> bool:
and "application/json" not in request.headers.getall("accept", [])
def exception_handler(logger: logging.Logger) -> MiddlewareType:
def exception_handler(logger: logging.Logger) -> Middleware:
"""
exception handler middleware. Just log any exception (except for client ones)
@ -51,7 +52,7 @@ def exception_handler(logger: logging.Logger) -> MiddlewareType:
logger(logging.Logger): class logger
Returns:
MiddlewareType: built middleware
Middleware: built middleware
Raises:
HTTPNoContent: OPTIONS method response