| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- """xmlrpclib.Transport implementation
- """
- import logging
- import urllib.parse
- import xmlrpc.client
- from typing import TYPE_CHECKING, Tuple
- from pip._internal.exceptions import NetworkConnectionError
- from pip._internal.network.session import PipSession
- from pip._internal.network.utils import raise_for_status
- if TYPE_CHECKING:
- from xmlrpc.client import _HostType, _Marshallable
- logger = logging.getLogger(__name__)
- class PipXmlrpcTransport(xmlrpc.client.Transport):
- """Provide a `xmlrpclib.Transport` implementation via a `PipSession`
- object.
- """
- def __init__(self, index_url, session, use_datetime=False):
- # type: (str, PipSession, bool) -> None
- super().__init__(use_datetime)
- index_parts = urllib.parse.urlparse(index_url)
- self._scheme = index_parts.scheme
- self._session = session
- def request(self, host, handler, request_body, verbose=False):
- # type: (_HostType, str, bytes, bool) -> Tuple[_Marshallable, ...]
- assert isinstance(host, str)
- parts = (self._scheme, host, handler, None, None, None)
- url = urllib.parse.urlunparse(parts)
- try:
- headers = {'Content-Type': 'text/xml'}
- response = self._session.post(url, data=request_body,
- headers=headers, stream=True)
- raise_for_status(response)
- self.verbose = verbose
- return self.parse_response(response.raw)
- except NetworkConnectionError as exc:
- assert exc.response
- logger.critical(
- "HTTP error %s while getting %s",
- exc.response.status_code, url,
- )
- raise
|
