diff --git a/nanomsg/__init__.py b/nanomsg/__init__.py index ab584dc..e58acaa 100644 --- a/nanomsg/__init__.py +++ b/nanomsg/__init__.py @@ -6,6 +6,8 @@ from . import wrapper +__all__ = ['wrapper', 'NanoMsgError', 'NanoMsgAPIError', 'Device', 'Socket'] + try: buffer except NameError: @@ -18,6 +20,7 @@ if name.startswith('NN_'): name = name[3:] globals()[name] = value + __all__.append(name) if hasattr(wrapper, 'create_writable_buffer'):