Connected logger to other classes

smbclientng/__main__.py

@@ -9,6 +9,7 @@
 from smbclientng.core.Config import Config
 from smbclientng.core.Credentials import Credentials
 from smbclientng.core.InteractiveShell import InteractiveShell
+from smbclientng.core.Logger import Logger
 from smbclientng.core.SessionsManager import SessionsManager
 
 
@@ -106,7 +107,9 @@ def main():
     config.not_interactive = options.not_interactive
     config.startup_script = options.startup_script
 
-    sessionsManager = SessionsManager(config=config)
+    logger = Logger(config=config, logfile=None)
+
+    sessionsManager = SessionsManager(config=config, logger=logger)
 
     if any([(options.auth_domain != '.'), (options.auth_username is not None), (options.auth_password is not None),(options.auth_hashes is not None)]):
         credentials = Credentials(
@@ -127,7 +130,8 @@ def main():
     # Start the main interactive command line
     shell = InteractiveShell(
         sessionsManager=sessionsManager, 
-        config=config
+        config=config,
+        logger=logger
     )
     shell.run()
 

smbclientng/core/InteractiveShell.py

@@ -81,10 +81,12 @@ class InteractiveShell(object):
     running = True
     modules = {}
 
-    def __init__(self, sessionsManager, config):
+    def __init__(self, sessionsManager, config, logger):
         # Objects
         self.sessionsManager = sessionsManager
         self.config = config
+        self.logger = logger
+        # Internals
         self.commandCompleterObject = CommandCompleter(smbSession=self.sessionsManager.current_session, config=self.config)
         readline.set_completer(self.commandCompleterObject.complete)
         readline.parse_and_bind("tab: complete")

smbclientng/core/Module.py

@@ -20,9 +20,10 @@ class Module(object):
     smbSession = None
     options = None
 
-    def __init__(self, smbSession, config):
+    def __init__(self, smbSession, config, logger):
         self.smbSession = smbSession
         self.config = config
+        self.logger = logger
 
     def parseArgs(self):
         raise NotImplementedError("Subclasses must implement this method")

smbclientng/core/SMBSession.py

@@ -53,10 +53,11 @@ class SMBSession(object):
         test_rights(sharename): Tests read and write access rights on a share.
     """
 
-    def __init__(self, host, port, credentials, config=None):
+    def __init__(self, host, port, credentials, config=None, logger=None):
         super(SMBSession, self).__init__()
         # Objects
         self.config = config
+        self.logger = logger
 
         # Target server
         self.host = host

smbclientng/core/SessionsManager.py

@@ -17,13 +17,14 @@ class SessionsManager(object):
     current_session_id = None
     sessions = {}
 
-    def __init__(self, config):
+    def __init__(self, config, logger):
         self.sessions = {}
         self.next_session_id = 1
         self.current_session = None
         self.current_session_id = None
 
-        self.config = config
+        self.config = config,
+        self.logger = logger
 
     def create_new_session(self, credentials, host, port=445):
         """