From 545fef9216559d08f58f7e20082b36352203a8cf Mon Sep 17 00:00:00 2001 From: Alexander Taylor Date: Wed, 11 May 2022 14:12:15 -0400 Subject: Add documentaiton decorator for Enterprise. --- python/decorators.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'python') 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 -- cgit v1.3.1