Merge pull request #2199 from msabramo/create_Index_class_and_PyPI_instance

Reduce duplication of "pypi.python.org"

Author: pfmoore

pip/cmdoptions.py

@@ -11,6 +11,7 @@
 
 import copy
 from optparse import OptionGroup, SUPPRESS_HELP, Option
+from pip.index import PyPI
 from pip.locations import CA_BUNDLE_PATH, USER_CACHE_DIR, src_prefix
 
 
@@ -192,7 +193,7 @@ def make(self):
     '-i', '--index-url', '--pypi-url',
     dest='index_url',
     metavar='URL',
-    default='https://pypi.python.org/simple/',
+    default=PyPI.simple_url,
     help='Base URL of Python Package Index (default %default).')
 
 extra_index_url = OptionMaker(

pip/commands/search.py

@@ -6,6 +6,7 @@
 
 from pip.basecommand import Command, SUCCESS
 from pip.download import PipXmlrpcTransport
+from pip.index import PyPI
 from pip.utils import get_terminal_size
 from pip.utils.logging import indent_log
 from pip.exceptions import CommandError
@@ -30,7 +31,7 @@ def __init__(self, *args, **kw):
             '--index',
             dest='index',
             metavar='URL',
-            default='https://pypi.python.org/pypi',
+            default=PyPI.pypi_url,
             help='Base URL of Python Package Index (default %default)')
 
         self.parser.insert_option_group(0, self.cmd_opts)

pip/index.py

@@ -47,6 +47,21 @@
 logger = logging.getLogger(__name__)
 
 
+class Index(object):
+    def __init__(self, url):
+        self.url = url
+        self.netloc = urllib_parse.urlsplit(url).netloc
+        self.simple_url = self.url_to_path('simple')
+        self.pypi_url = self.url_to_path('pypi')
+        self.pip_json_url = self.url_to_path('pypi/pip/json')
+
+    def url_to_path(self, path):
+        return urllib_parse.urljoin(self.url, path)
+
+
+PyPI = Index('https://pypi.python.org/')
+
+
 class PackageFinder(object):
     """This finds packages.
 
@@ -302,7 +317,7 @@ def mkurl_pypi_url(url):
             )
 
             page = self._get_page(main_index_url, req)
-            if page is None and 'pypi.python.org' not in str(main_index_url):
+            if page is None and PyPI.netloc not in str(main_index_url):
                 warnings.warn(
                     "Failed to find %r at %s. It is suggested to upgrade "
                     "your index to support normalized names as the name in "
@@ -703,7 +718,7 @@ def _link_package_versions(self, link, search_name):
                         and comes_from is not None
                         and urllib_parse.urlparse(
                             comes_from.url
-                        ).netloc.endswith("pypi.python.org")):
+                        ).netloc.endswith(PyPI.netloc)):
                     if not wheel.supported(tags=supported_tags_noarch):
                         logger.debug(
                             "Skipping %s because it is a pypi-hosted binary "

pip/utils/outdated.py

@@ -10,6 +10,7 @@
 from pip._vendor import pkg_resources
 
 from pip.compat import total_seconds
+from pip.index import PyPI
 from pip.locations import USER_CACHE_DIR, running_under_virtualenv
 
 
@@ -104,7 +105,7 @@ def pip_version_check(session):
         # Refresh the version if we need to or just see if we need to warn
         if pypi_version is None:
             resp = session.get(
-                "https://pypi.python.org/pypi/pip/json",
+                PyPI.pip_json_url,
                 headers={"Accept": "application/json"},
             )
             resp.raise_for_status()