1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
|
from typing import Optional
from .._abc import AsyncSocketStream, AsyncResolver
from .abc import AsyncConnector
from .._protocols import http
class HttpAsyncConnector(AsyncConnector):
def __init__(
self,
username: Optional[str],
password: Optional[str],
resolver: AsyncResolver,
):
self._username = username
self._password = password
self._resolver = resolver
async def connect(
self,
stream: AsyncSocketStream,
host: str,
port: int,
) -> http.ConnectReply:
conn = http.Connection()
request = http.ConnectRequest(
host=host,
port=port,
username=self._username,
password=self._password,
)
data = conn.send(request)
await stream.write_all(data)
data = await stream.read()
reply: http.ConnectReply = conn.receive(data)
return reply
|