diff options
Diffstat (limited to 'python/decorators.py')
| -rw-r--r-- | python/decorators.py | 15 |
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 |