| import urllib.parse |
| |
| |
| class PackageIndex: |
| """Represents a Package Index and provides easier access to endpoints""" |
| |
| __slots__ = ["url", "netloc", "simple_url", "pypi_url", "file_storage_domain"] |
| |
| def __init__(self, url: str, file_storage_domain: str) -> None: |
| super().__init__() |
| self.url = url |
| self.netloc = urllib.parse.urlsplit(url).netloc |
| self.simple_url = self._url_for_path("simple") |
| self.pypi_url = self._url_for_path("pypi") |
| |
| # This is part of a temporary hack used to block installs of PyPI |
| # packages which depend on external urls only necessary until PyPI can |
| # block such packages themselves |
| self.file_storage_domain = file_storage_domain |
| |
| def _url_for_path(self, path: str) -> str: |
| return urllib.parse.urljoin(self.url, path) |
| |
| |
| PyPI = PackageIndex("https://pypi.org/", file_storage_domain="files.pythonhosted.org") |
| TestPyPI = PackageIndex( |
| "https://test.pypi.org/", file_storage_domain="test-files.pythonhosted.org" |
| ) |
