diff --git a/requirements.txt b/requirements.txt index cf4d31b..21561d6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,3 +2,4 @@ WebOb>=1.8.0 simplegeneric pytz netaddr>=0.7.12 +importlib_metadata>=1.7.0;python_version<'3.8' # Apache-2.0 diff --git a/wsme/protocol.py b/wsme/protocol.py index f7c66da..b132518 100644 --- a/wsme/protocol.py +++ b/wsme/protocol.py @@ -1,6 +1,9 @@ import weakref -import pkg_resources +try: + import importlib.metadata as importlib_metadata +except ImportError: # Python < 3.8 + import importlib_metadata from wsme.exc import ClientSideError @@ -106,8 +109,7 @@ def register_protocol(protocol): def getprotocol(name, **options): protocol_class = registered_protocols.get(name) if protocol_class is None: - for entry_point in pkg_resources.iter_entry_points( - 'wsme.protocols', name): + for entry_point in importlib_metadata.entry_points()['wsme.protocols']: if entry_point.name == name: protocol_class = entry_point.load() if protocol_class is None: