diff --git a/requirements.txt b/requirements.txt index cf4d31b..32aaeba 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,3 +2,5 @@ WebOb>=1.8.0 simplegeneric pytz netaddr>=0.7.12 +# TODO: Drop this when Python 3.10 is our minimum supported version +importlib_metadata>=4.4 # Apache-2.0 diff --git a/wsme/protocol.py b/wsme/protocol.py index f7c66da..0eda23f 100644 --- a/wsme/protocol.py +++ b/wsme/protocol.py @@ -1,6 +1,6 @@ import weakref -import pkg_resources +import importlib_metadata from wsme.exc import ClientSideError @@ -106,8 +106,9 @@ 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().select( + group='wsme.protocols', + ): if entry_point.name == name: protocol_class = entry_point.load() if protocol_class is None: