From 853985e1dca99e3f265c02f0f733f9c4b33106ca Mon Sep 17 00:00:00 2001 From: Ian Cordasco Date: Sun, 7 Feb 2016 09:40:18 -0600 Subject: [PATCH] Allow dictionary-like get behaviour with plugins --- flake8/plugins/manager.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/flake8/plugins/manager.py b/flake8/plugins/manager.py index 02a136c..3a20b91 100644 --- a/flake8/plugins/manager.py +++ b/flake8/plugins/manager.py @@ -203,6 +203,22 @@ class PluginTypeManager(object): LOG.debug('Retrieving plugin for "%s".', name) return self.plugins[name] + def get(self, name, default=None): + """Retrieve the plugin referred to by ``name`` or return the default. + + :param str name: + Name of the plugin to retrieve. + :param default: + Default value to return. + :returns: + Plugin object referred to by name, if it exists. + :rtype: + :class:`Plugin` + """ + if name in self: + return self[name] + return default + @property def names(self): """Proxy attribute to underlying manager."""