Pull up DefaultProvider registration into a classmethod.

397d759e · Jason R. Coombs · e01792ec · 397d759e
Commit 397d759e authored Jan 26, 2016 by Jason R. Coombs
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 3 deletions

pkg_resources/__init__.py pkg_resources/__init__.py +7 -3

No files found.
--- a/pkg_resources/__init__.py
+++ b/pkg_resources/__init__.py
@@ -1725,10 +1725,14 @@ class DefaultProvider(EggProvider):
        with open(path, 'rb') as stream:
            return stream.read()

-register_loader_type(type(None), DefaultProvider)
+    @classmethod
+    def _register(cls):
+        register_loader_type(type(None), cls)

-if importlib_machinery is not None:
-    register_loader_type(importlib_machinery.SourceFileLoader, DefaultProvider)
+        if importlib_machinery is not None:
+            register_loader_type(importlib_machinery.SourceFileLoader, cls)
+
+DefaultProvider._register()


 class EmptyProvider(NullProvider):