From 3c35ef1ea77acc45aa71b6ad08ebe032e40e2c22 Mon Sep 17 00:00:00 2001 From: Terry Chia 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 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src') 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: -- cgit v1.2.3