?HAMA

Navigation

Breadcrumbs

/ / opt / cloudlinux / venv / lib / python3.11 / site-packages / aiohttp

Create

Create File | Create Folder | Upload Files | Run Command

Edit File Size: 11.46 KB

//opt/cloudlinux/venv/lib/python3.11/site-packages/aiohttp/web_runner.py

Text Base64

import asyncio
import signal
import socket
import warnings
from abc import ABC, abstractmethod
from typing import Any, Awaitable, Callable, List, Optional, Set

from yarl import URL

from .typedefs import PathLike
from .web_app import Application
from .web_server import Server

try:
    from ssl import SSLContext
except ImportError:
    SSLContext = object  # type: ignore[misc,assignment]

__all__ = (
    "BaseSite",
    "TCPSite",
    "UnixSite",
    "NamedPipeSite",
    "SockSite",
    "BaseRunner",
    "AppRunner",
    "ServerRunner",
    "GracefulExit",
)

class GracefulExit(SystemExit):
    code = 1

def _raise_graceful_exit() -> None:
    raise GracefulExit()

class BaseSite(ABC):
    __slots__ = ("_runner", "_ssl_context", "_backlog", "_server")

def __init__(
        self,
        runner: "BaseRunner",
        *,
        shutdown_timeout: float = 60.0,
        ssl_context: Optional[SSLContext] = None,
        backlog: int = 128,
    ) -> None:
        if runner.server is None:
            raise RuntimeError("Call runner.setup() before making a site")
        if shutdown_timeout != 60.0:
            msg = "shutdown_timeout should be set on BaseRunner"
            warnings.warn(msg, DeprecationWarning, stacklevel=2)
            runner._shutdown_timeout = shutdown_timeout
        self._runner = runner
        self._ssl_context = ssl_context
        self._backlog = backlog
        self._server: Optional[asyncio.AbstractServer] = None

@property
    @abstractmethod
    def name(self) -> str:
        pass  # pragma: no cover

@abstractmethod
    async def start(self) -> None:
        self._runner._reg_site(self)

async def stop(self) -> None:
        self._runner._check_site(self)
        if self._server is not None:  # Maybe not started yet
            self._server.close()

self._runner._unreg_site(self)

class TCPSite(BaseSite):
    __slots__ = ("_host", "_port", "_reuse_address", "_reuse_port")

def __init__(
        self,
        runner: "BaseRunner",
        host: Optional[str] = None,
        port: Optional[int] = None,
        *,
        shutdown_timeout: float = 60.0,
        ssl_context: Optional[SSLContext] = None,
        backlog: int = 128,
        reuse_address: Optional[bool] = None,
        reuse_port: Optional[bool] = None,
    ) -> None:
        super().__init__(
            runner,
            shutdown_timeout=shutdown_timeout,
            ssl_context=ssl_context,
            backlog=backlog,
        )
        self._host = host
        if port is None:
            port = 8443 if self._ssl_context else 8080
        self._port = port
        self._reuse_address = reuse_address
        self._reuse_port = reuse_port

@property
    def name(self) -> str:
        scheme = "https" if self._ssl_context else "http"
        host = "0.0.0.0" if self._host is None else self._host
        return str(URL.build(scheme=scheme, host=host, port=self._port))

async def start(self) -> None:
        await super().start()
        loop = asyncio.get_event_loop()
        server = self._runner.server
        assert server is not None
        self._server = await loop.create_server(
            server,
            self._host,
            self._port,
            ssl=self._ssl_context,
            backlog=self._backlog,
            reuse_address=self._reuse_address,
            reuse_port=self._reuse_port,
        )

class UnixSite(BaseSite):
    __slots__ = ("_path",)

def __init__(
        self,
        runner: "BaseRunner",
        path: PathLike,
        *,
        shutdown_timeout: float = 60.0,
        ssl_context: Optional[SSLContext] = None,
        backlog: int = 128,
    ) -> None:
        super().__init__(
            runner,
            shutdown_timeout=shutdown_timeout,
            ssl_context=ssl_context,
            backlog=backlog,
        )
        self._path = path

@property
    def name(self) -> str:
        scheme = "https" if self._ssl_context else "http"
        return f"{scheme}://unix:{self._path}:"

async def start(self) -> None:
        await super().start()
        loop = asyncio.get_event_loop()
        server = self._runner.server
        assert server is not None
        self._server = await loop.create_unix_server(
            server,
            self._path,
            ssl=self._ssl_context,
            backlog=self._backlog,
        )

class NamedPipeSite(BaseSite):
    __slots__ = ("_path",)

def __init__(
        self, runner: "BaseRunner", path: str, *, shutdown_timeout: float = 60.0
    ) -> None:
        loop = asyncio.get_event_loop()
        if not isinstance(
            loop, asyncio.ProactorEventLoop  # type: ignore[attr-defined]
        ):
            raise RuntimeError(
                "Named Pipes only available in proactor" "loop under windows"
            )
        super().__init__(runner, shutdown_timeout=shutdown_timeout)
        self._path = path

@property
    def name(self) -> str:
        return self._path

async def start(self) -> None:
        await super().start()
        loop = asyncio.get_event_loop()
        server = self._runner.server
        assert server is not None
        _server = await loop.start_serving_pipe(  # type: ignore[attr-defined]
            server, self._path
        )
        self._server = _server[0]

class SockSite(BaseSite):
    __slots__ = ("_sock", "_name")

def __init__(
        self,
        runner: "BaseRunner",
        sock: socket.socket,
        *,
        shutdown_timeout: float = 60.0,
        ssl_context: Optional[SSLContext] = None,
        backlog: int = 128,
    ) -> None:
        super().__init__(
            runner,
            shutdown_timeout=shutdown_timeout,
            ssl_context=ssl_context,
            backlog=backlog,
        )
        self._sock = sock
        scheme = "https" if self._ssl_context else "http"
        if hasattr(socket, "AF_UNIX") and sock.family == socket.AF_UNIX:
            name = f"{scheme}://unix:{sock.getsockname()}:"
        else:
            host, port = sock.getsockname()[:2]
            name = str(URL.build(scheme=scheme, host=host, port=port))
        self._name = name

@property
    def name(self) -> str:
        return self._name

class BaseRunner(ABC):
    __slots__ = (
        "shutdown_callback",
        "_handle_signals",
        "_kwargs",
        "_server",
        "_sites",
        "_shutdown_timeout",
    )

def __init__(
        self,
        *,
        handle_signals: bool = False,
        shutdown_timeout: float = 60.0,
        **kwargs: Any,
    ) -> None:
        self.shutdown_callback: Optional[Callable[[], Awaitable[None]]] = None
        self._handle_signals = handle_signals
        self._kwargs = kwargs
        self._server: Optional[Server] = None
        self._sites: List[BaseSite] = []
        self._shutdown_timeout = shutdown_timeout

@property
    def server(self) -> Optional[Server]:
        return self._server

@property
    def addresses(self) -> List[Any]:
        ret: List[Any] = []
        for site in self._sites:
            server = site._server
            if server is not None:
                sockets = server.sockets  # type: ignore[attr-defined]
                if sockets is not None:
                    for sock in sockets:
                        ret.append(sock.getsockname())
        return ret

@property
    def sites(self) -> Set[BaseSite]:
        return set(self._sites)

async def setup(self) -> None:
        loop = asyncio.get_event_loop()

if self._handle_signals:
            try:
                loop.add_signal_handler(signal.SIGINT, _raise_graceful_exit)
                loop.add_signal_handler(signal.SIGTERM, _raise_graceful_exit)
            except NotImplementedError:  # pragma: no cover
                # add_signal_handler is not implemented on Windows
                pass

self._server = await self._make_server()

@abstractmethod
    async def shutdown(self) -> None:
        """Call any shutdown hooks to help server close gracefully."""

async def cleanup(self) -> None:
        # The loop over sites is intentional, an exception on gather()
        # leaves self._sites in unpredictable state.
        # The loop guaranties that a site is either deleted on success or
        # still present on failure
        for site in list(self._sites):
            await site.stop()

if self._server:  # If setup succeeded
            self._server.pre_shutdown()
            await self.shutdown()

if self.shutdown_callback:
                await self.shutdown_callback()

await self._server.shutdown(self._shutdown_timeout)
        await self._cleanup_server()

self._server = None
        if self._handle_signals:
            loop = asyncio.get_running_loop()
            try:
                loop.remove_signal_handler(signal.SIGINT)
                loop.remove_signal_handler(signal.SIGTERM)
            except NotImplementedError:  # pragma: no cover
                # remove_signal_handler is not implemented on Windows
                pass

@abstractmethod
    async def _make_server(self) -> Server:
        pass  # pragma: no cover

@abstractmethod
    async def _cleanup_server(self) -> None:
        pass  # pragma: no cover

def _reg_site(self, site: BaseSite) -> None:
        if site in self._sites:
            raise RuntimeError(f"Site {site} is already registered in runner {self}")
        self._sites.append(site)

def _check_site(self, site: BaseSite) -> None:
        if site not in self._sites:
            raise RuntimeError(f"Site {site} is not registered in runner {self}")

def _unreg_site(self, site: BaseSite) -> None:
        if site not in self._sites:
            raise RuntimeError(f"Site {site} is not registered in runner {self}")
        self._sites.remove(site)

class ServerRunner(BaseRunner):
    """Low-level web server runner"""

__slots__ = ("_web_server",)

def __init__(
        self, web_server: Server, *, handle_signals: bool = False, **kwargs: Any
    ) -> None:
        super().__init__(handle_signals=handle_signals, **kwargs)
        self._web_server = web_server

async def shutdown(self) -> None:
        pass

async def _make_server(self) -> Server:
        return self._web_server

async def _cleanup_server(self) -> None:
        pass

class AppRunner(BaseRunner):
    """Web Application runner"""

__slots__ = ("_app",)

def __init__(
        self, app: Application, *, handle_signals: bool = False, **kwargs: Any
    ) -> None:
        super().__init__(handle_signals=handle_signals, **kwargs)
        if not isinstance(app, Application):
            raise TypeError(
                "The first argument should be web.Application "
                "instance, got {!r}".format(app)
            )
        self._app = app

@property
    def app(self) -> Application:
        return self._app

async def shutdown(self) -> None:
        await self._app.shutdown()

async def _make_server(self) -> Server:
        loop = asyncio.get_event_loop()
        self._app._set_loop(loop)
        self._app.on_startup.freeze()
        await self._app.startup()
        self._app.freeze()

return self._app._make_handler(loop=loop, **self._kwargs)

async def _cleanup_server(self) -> None:
        await self._app.cleanup()

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Directory Contents

Dirs: 2 × Files: 59

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
.hash DIR	-	drwxr-xr-x	2026-02-06 08:01:08	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2026-02-06 08:07:29	Edit Download recursive recursive
abc.py	5.37 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
base_protocol.py	2.68 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
client.py	46.17 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
client_exceptions.py	9.19 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
client_proto.py	8.45 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
client_reqrep.py	38.75 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
client_ws.py	10.75 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
compression_utils.py	4.90 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
connector.py	51.56 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
cookiejar.py	13.69 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
formdata.py	5.96 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
hdrs.py	4.50 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
helpers.py	29.55 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
http.py	1.80 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
http_exceptions.py	2.65 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
http_parser.py	34.66 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
http_websocket.py	26.09 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
http_writer.py	5.79 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
locks.py	1.11 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
log.py	325 B	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
multipart.py	31.71 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
payload.py	13.22 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
payload_streamer.py	2.04 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
py.typed	7 B	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
pytest_plugin.py	11.33 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
resolver.py	4.95 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
streams.py	20.35 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
tcp_helpers.py	961 B	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
test_utils.py	19.71 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
tracing.py	14.78 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
typedefs.py	1.44 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web.py	18.81 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_app.py	17.88 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_exceptions.py	10.12 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_fileresponse.py	11.15 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_log.py	7.62 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_middlewares.py	3.94 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_protocol.py	22.50 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_request.py	28.08 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_response.py	27.08 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_routedef.py	5.99 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_runner.py	11.46 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_server.py	2.53 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_urldispatcher.py	39.12 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
web_ws.py	18.21 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
worker.py	7.78 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
_cparser.pxd	4.22 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
_find_header.pxd	68 B	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
_headers.pxi	1.96 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
_helpers.cpython-311-x86_64-linux-gnu.so	86.01 KB	lrwxr-xr-x	2026-01-20 13:18:50	Edit Download recursive recursive
_helpers.pyi	202 B	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
_helpers.pyx	1.02 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
_http_parser.cpython-311-x86_64-linux-gnu.so	569.59 KB	lrwxr-xr-x	2026-01-20 13:18:50	Edit Download recursive recursive
_http_parser.pyx	27.40 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
_http_writer.cpython-311-x86_64-linux-gnu.so	75.01 KB	lrwxr-xr-x	2026-01-20 13:18:50	Edit Download recursive recursive
_http_writer.pyx	4.47 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
_websocket.cpython-311-x86_64-linux-gnu.so	53.37 KB	lrwxr-xr-x	2026-01-20 13:18:50	Edit Download recursive recursive
_websocket.pyx	1.52 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive
__init__.py	7.58 KB	lrw-r--r--	2026-01-20 13:01:48	Edit Download recursive recursive