"""Low level HTTP server.""" import asyncio from .web_protocol import RequestHandler from .web_request import BaseRequest __all__ = ('Server',) class Server: def __init__(self, handler, *, request_factory=None, loop=None, **kwargs): if loop is None: loop = asyncio.get_event_loop() self._loop = loop self._connections = {} self._kwargs = kwargs self.requests_count = 0 self.request_handler = handler self.request_factory = request_factory or self._make_request @property def connections(self): return list(self._connections.keys()) def connection_made(self, handler, transport): self._connections[handler] = transport def connection_lost(self, handler, exc=None): if handler in self._connections: del self._connections[handler] def _make_request(self, message, payload, protocol, writer, task): return BaseRequest( message, payload, protocol, writer, task, self._loop) async def shutdown(self, timeout=None): coros = [conn.shutdown(timeout) for conn in self._connections] await asyncio.gather(*coros, loop=self._loop) self._connections.clear() def __call__(self): return RequestHandler(self, loop=self._loop, **self._kwargs)