From 3c35ef1ea77acc45aa71b6ad08ebe032e40e2c22 Mon Sep 17 00:00:00 2001
From: Terry Chia <terrycwk1994@gmail.com>
Date: Mon, 29 Aug 2016 20:21:28 +0800
Subject: Add a register_interface_if decorator. (#3120)

* Add a register_interface_if decorator.

* Add tests.

* PEP 8.
---
 src/cryptography/utils.py |  9 +++++++++
 tests/test_interfaces.py  | 28 +++++++++++++++++++++++++++-
 2 files changed, 36 insertions(+), 1 deletion(-)

diff --git a/src/cryptography/utils.py b/src/cryptography/utils.py
index d3e845ab..48ed449a 100644
--- a/src/cryptography/utils.py
+++ b/src/cryptography/utils.py
@@ -31,6 +31,15 @@ def register_interface(iface):
     return register_decorator
 
 
+def register_interface_if(predicate, iface):
+    def register_decorator(klass):
+        if predicate:
+            verify_interface(iface, klass)
+            iface.register(klass)
+        return klass
+    return register_decorator
+
+
 if hasattr(int, "from_bytes"):
     int_from_bytes = int.from_bytes
 else:
diff --git a/tests/test_interfaces.py b/tests/test_interfaces.py
index bdb4a94d..97df45a3 100644
--- a/tests/test_interfaces.py
+++ b/tests/test_interfaces.py
@@ -8,7 +8,33 @@ import pytest
 
 import six
 
-from cryptography.utils import InterfaceNotImplemented, verify_interface
+from cryptography.utils import (
+    InterfaceNotImplemented, register_interface_if, verify_interface
+)
+
+
+def test_register_interface_if_true():
+    @six.add_metaclass(abc.ABCMeta)
+    class SimpleInterface(object):
+        pass
+
+    @register_interface_if(1 == 1, SimpleInterface)
+    class SimpleClass(object):
+        pass
+
+    assert issubclass(SimpleClass, SimpleInterface) is True
+
+
+def test_register_interface_if_false():
+    @six.add_metaclass(abc.ABCMeta)
+    class SimpleInterface(object):
+        pass
+
+    @register_interface_if(1 == 2, SimpleInterface)
+    class SimpleClass(object):
+        pass
+
+    assert issubclass(SimpleClass, SimpleInterface) is False
 
 
 class TestVerifyInterface(object):
-- 
cgit v1.2.3