added get_extension_for_class #2255
diff --git a/src/cryptography/x509/base.py b/src/cryptography/x509/base.py
index 2b4eeb5..f58b522 100644
--- a/src/cryptography/x509/base.py
+++ b/src/cryptography/x509/base.py
@@ -79,6 +79,13 @@
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)