add logging handler to Session

Author: jlwalke2

src/sasctl/core.py

@@ -327,27 +327,52 @@ def __init__(self, hostname,
         if current_session() is None:
             current_session(self)
 
-    def add_stderr_logger(self, level=None):
-        """Log session requests and responses to stderr.
+    def add_logger(self, handler, level=None):
+        """Log session requests and responses.
 
         Parameters
         ----------
+        handler : logging.Handler
+            A Handler instance to use for logging the requests and responses.
         level : int, optional
-            The logging level of the handler.  Defaults to logging.DEBUG
+            The logging level to assign to the handler.  Ignored if handler's
+            logging level is already set.  Defaults to DEBUG.
 
         Returns
         -------
-        logging.StreamHandler
+        handler
+
+
+        .. versionadded:: 1.2.0
 
         """
         level = level or logging.DEBUG
 
-        handler = logging.StreamHandler()
-        handler.setLevel(level=level)
+        if handler.level == logging.NOTSET:
+            handler.setLevel(level)
+
         self.message_log.addHandler(handler)
-        self.message_log.setLevel(level)
+
+        if self.message_log.level == logging.NOTSET:
+            self.message_log.setLevel(handler.level)
+
         return handler
 
+    def add_stderr_logger(self, level=None):
+        """Log session requests and responses to stderr.
+
+        Parameters
+        ----------
+        level : int, optional
+            The logging level of the handler.  Defaults to logging.DEBUG
+
+        Returns
+        -------
+        logging.StreamHandler
+
+        """
+        return self.add_logger(logging.StreamHandler(), level)
+
     @property
     def username(self):
         return self._settings.get('username')