Improve performance on Python 3.11

Author: adamtheturtle

CHANGELOG.rst

@@ -2,6 +2,10 @@
 Release History
 ---------------
 
+2.4.3
+
+- Improves performance on Python 3.11.
+
 2.4.2
 
 - Added support for Python 3.11.

pip_check_reqs/find_extra_reqs.py

@@ -8,6 +8,7 @@
 import sys
 from pathlib import Path
 from typing import Callable, Iterable, List, Optional, Union
+from unittest import mock
 
 from packaging.utils import canonicalize_name
 from pip._internal.commands.show import search_packages_info
@@ -43,7 +44,13 @@ def find_extra_reqs(
         dist.metadata["Name"] for dist in importlib.metadata.distributions()
     ]
 
-    for package in search_packages_info(all_pkgs):
+    # On Python 3.11 (and maybe higher), setting this environment variable
+    # dramatically improves speeds.
+    # See https://github.com/r1chardj0n3s/pip-check-reqs/issues/123.
+    with mock.patch.dict(os.environ, {"_PIP_USE_IMPORTLIB_METADATA": "False"}):
+        packages_info = list(search_packages_info(all_pkgs))
+
+    for package in packages_info:
         package_name = package.name
         package_location = package.location
         package_files = []

pip_check_reqs/find_missing_reqs.py

@@ -8,6 +8,7 @@
 import sys
 from pathlib import Path
 from typing import Callable, Iterable, List, Optional, Tuple
+from unittest import mock
 
 from packaging.utils import NormalizedName, canonicalize_name
 from pip._internal.commands.show import search_packages_info
@@ -41,7 +42,13 @@ def find_missing_reqs(
         dist.metadata["Name"] for dist in importlib.metadata.distributions()
     ]
 
-    for package in search_packages_info(all_pkgs):
+    # On Python 3.11 (and maybe higher), setting this environment variable
+    # dramatically improves speeds.
+    # See https://github.com/r1chardj0n3s/pip-check-reqs/issues/123.
+    with mock.patch.dict(os.environ, {"_PIP_USE_IMPORTLIB_METADATA": "False"}):
+        packages_info = list(search_packages_info(all_pkgs))
+
+    for package in packages_info:
         package_name = package.name
         package_location = package.location
         package_files = []