Add documentaiton decorator for Enterprise.

1 files changed, 14 insertions, 1 deletions

diff --git a/python/decorators.py b/python/decorators.py
index 7c7025e3..c6664652 100644
--- a/python/decorators.py
+++ b/python/decorators.py
@@ -23,4 +23,17 @@ def deprecated(cls):
 	else:
 		cls.__doc__ = deprecated_note
 
-	return cls
\ No newline at end of file
+	return cls
+
+
+def enterprise(cls):
+	enterprise_note = '''
+	.. note: This object is only available in the Enterprise edition of Binary Ninja.
+'''
+
+	if hasattr(cls, "__doc__") and cls.__doc__:
+		cls.__doc__ = enterprise_note + cls.__doc__
+	else:
+		cls.__doc__ = enterprise_note
+
+	return cls