Skip to content

Commit

Permalink
Enable importlib.metadata backend on Python 3.11
Browse files Browse the repository at this point in the history
  • Loading branch information
uranusjr committed Apr 19, 2022
1 parent c247ddc commit c07ae74
Show file tree
Hide file tree
Showing 3 changed files with 35 additions and 5 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/ci.yml
Expand Up @@ -219,11 +219,12 @@ jobs:
env:
TEMP: "R:\\Temp"

# TODO: Remove this when we add Python 3.11 to CI.
tests-importlib-metadata:
name: tests for importlib.metadata backend
runs-on: ubuntu-latest
env:
_PIP_METADATA_BACKEND_IMPORTLIB: egg-compat
_PIP_USE_IMPORTLIB_METADATA: 'true'

needs: [pre-commit, packaging, determine-changes]
if: >-
Expand All @@ -241,7 +242,6 @@ jobs:

- run: pip install nox 'virtualenv<20'

# Main check
- name: Run unit tests
run: >-
nox -s test-3.10 --
Expand Down
5 changes: 3 additions & 2 deletions news/10709.process.rst
@@ -1,3 +1,4 @@
Start migration of distribution metadata implementation from ``pkg_resources``
to ``importlib.metadata``. The new implementation is currently not exposed in
any user-facing way, but included in the code base for easier development.
to ``importlib.metadata``. The new implementation is enabled by default on
Python 3.11 (or later), but can be disabled with a private environment variable
``_PIP_USE_IMPORTLIB_METADATA``.
31 changes: 30 additions & 1 deletion src/pip/_internal/metadata/__init__.py
@@ -1,7 +1,11 @@
import contextlib
import functools
import os
import sys
from typing import TYPE_CHECKING, List, Optional, Type, cast

from pip._internal.utils.misc import strtobool

from .base import BaseDistribution, BaseEnvironment, FilesystemWheel, MemoryWheel, Wheel

if TYPE_CHECKING:
Expand All @@ -22,14 +26,39 @@
]


def _should_use_importlib_metadata() -> bool:
"""Whether to use the ``importlib.metadata`` or ``pkg_resources`` backend.
By default, pip uses ``importlib.metadata`` on Python 3.11+, and
``pkg_resourcess`` otherwise. This can be overriden by a couple of ways:
* If environment variable ``_PIP_USE_IMPORTLIB_METADATA`` is set, it
dictates whether ``importlib.metadata`` is used, regardless of Python
version.
* On Python 3.11+, Python distributors can patch ``importlib.metadata``
to add a global constant ``_PIP_USE_IMPORTLIB_METADATA = False``. This
makes pip use ``pkg_resources`` (unless the user set the aforementioned
environment variable to *True*).
"""
with contextlib.suppress(KeyError, ValueError):
return bool(strtobool(os.environ["_PIP_USE_IMPORTLIB_METADATA"]))
if sys.version_info < (3, 11):
return False
try:
import importlib.metadata
except ImportError:
return True
return bool(getattr(importlib.metadata, "_PIP_USE_IMPORTLIB_METADATA", True))


class Backend(Protocol):
Distribution: Type[BaseDistribution]
Environment: Type[BaseEnvironment]


@functools.lru_cache(maxsize=None)
def select_backend() -> Backend:
if os.environ.get("_PIP_METADATA_BACKEND_IMPORTLIB"):
if _should_use_importlib_metadata():
from . import importlib

return cast(Backend, importlib)
Expand Down

0 comments on commit c07ae74

Please sign in to comment.