Refactor to support multiple loader names. Ref #1337.

afa410dc · Jason R. Coombs · af4e065b · afa410dc
Commit afa410dc authored Apr 25, 2018 by Jason R. Coombs
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 6 deletions

pkg_resources/__init__.py pkg_resources/__init__.py +4 -6

No files found.
--- a/pkg_resources/__init__.py
+++ b/pkg_resources/__init__.py
@@ -1518,12 +1518,10 @@ class DefaultProvider(EggProvider):

    @classmethod
    def _register(cls):
-        loader_cls = getattr(
-            importlib_machinery,
-            'SourceFileLoader',
-            type(None),
-        )
-        register_loader_type(loader_cls, cls)
+        loader_names = 'SourceFileLoader',
+        for name in loader_names:
+            loader_cls = getattr(importlib_machinery, name, type(None))
+            register_loader_type(loader_cls, cls)


 DefaultProvider._register()