From afa410dcce836b0dea503a68eae7467911201bc5 Mon Sep 17 00:00:00 2001 From: "Jason R. Coombs" Date: Wed, 25 Apr 2018 15:29:48 -0400 Subject: Refactor to support multiple loader names. Ref #1337. --- pkg_resources/__init__.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) (limited to 'pkg_resources') diff --git a/pkg_resources/__init__.py b/pkg_resources/__init__.py index 8d95bd29..2ccf85ad 100644 --- 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() -- cgit v1.2.3