From db34d7e3d9e6b13e549866324323e1df8a6e7777 Mon Sep 17 00:00:00 2001 From: Phoebe Queen Date: Wed, 12 Aug 2015 02:15:08 +0100 Subject: added get_extension_for_class #2255 --- src/cryptography/x509/base.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/cryptography/x509/base.py b/src/cryptography/x509/base.py index 2b4eeb56..f58b5221 100644 --- a/src/cryptography/x509/base.py +++ b/src/cryptography/x509/base.py @@ -79,6 +79,13 @@ class Extensions(object): raise ExtensionNotFound("No {0} extension was found".format(oid), oid) + def get_extension_for_class(self, extclass): + for ext in self: + if type(ext.value) == extclass: + return ext + + raise ExtensionNotFound("No {0} extension was found".format(extclass), extclass) + def __iter__(self): return iter(self._extensions) -- cgit v1.2.3