diff options
| -rw-r--r-- | nsploit/__main__.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/nsploit/__main__.py b/nsploit/__main__.py index 5f326ef..d4d2fa9 100644 --- a/nsploit/__main__.py +++ b/nsploit/__main__.py @@ -48,8 +48,9 @@ def target(script, target): def user_scope(comm): import nsploit as lib - scope = { name: getattr(lib, name) for name in dir(lib) } - scope['__version__'] = __version__ + import nsploit.types as types + scope = { name: getattr(lib, name) for name in dir(lib) } + scope |= { name: getattr(types, name) for name in dir(types) } scope['print'] = elog scope['io'] = comm return scope |