From 5bb160f081068db46c7eff29c057f7cafbad3b05 Mon Sep 17 00:00:00 2001 From: l <56237933+lcdunne@users.noreply.github.com> Date: Sat, 23 Sep 2023 09:36:25 +0100 Subject: [PATCH 1/2] Add remove_client method to base_client.registry.BaseOAuth --- authlib/integrations/base_client/registry.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/authlib/integrations/base_client/registry.py b/authlib/integrations/base_client/registry.py index 68d1be5d..024ddc6d 100644 --- a/authlib/integrations/base_client/registry.py +++ b/authlib/integrations/base_client/registry.py @@ -86,6 +86,16 @@ def register(self, name, overwrite=False, **kwargs): self._registry[name] = (overwrite, kwargs) return self.create_client(name) + def remove_client(self, name): + if name not in self._registry: + raise KeyError(f"Client {name} not found in registry.") + + if name not in self._clients: + raise KeyError(f"Client {name} not found in clients.") + + del self._registry[name] + del self._clients[name] + def generate_client_kwargs(self, name, overwrite, **kwargs): fetch_token = kwargs.pop('fetch_token', None) update_token = kwargs.pop('update_token', None) From 60dc9d04b384ffae56a3ffab15227a401e7b18b8 Mon Sep 17 00:00:00 2001 From: l <56237933+lcdunne@users.noreply.github.com> Date: Sat, 23 Sep 2023 09:46:32 +0100 Subject: [PATCH 2/2] Change from double to single quotes to match style --- authlib/integrations/base_client/registry.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/authlib/integrations/base_client/registry.py b/authlib/integrations/base_client/registry.py index 024ddc6d..8bff6cbd 100644 --- a/authlib/integrations/base_client/registry.py +++ b/authlib/integrations/base_client/registry.py @@ -88,11 +88,11 @@ def register(self, name, overwrite=False, **kwargs): def remove_client(self, name): if name not in self._registry: - raise KeyError(f"Client {name} not found in registry.") + raise KeyError(f'Client {name} not found in registry.') if name not in self._clients: - raise KeyError(f"Client {name} not found in clients.") - + raise KeyError(f'Client {name} not found in clients.') + del self._registry[name] del self._clients[name]