Display log messages from imported modules in Custom Script?

[llamafilm]

When I import modules in a custom script, is there some way to include their logging messages in the Netbox custom script results output? For example the requests module logs many helpful debugging messages.

[llamafilm]

I found a solution, by adding a custom logging handler. Here's a working example.

import logging
import requests

from extras.scripts import Script

class NetboxHandler(logging.Handler):
    """Send log messages from imported modules to Netbox custom script log"""

    def __init__(self, script: Script)-> None:
        logging.Handler.__init__(self)
        self.script = script
        self.setFormatter(logging.Formatter('[%(name)s] %(message)s'))

    def emit(self, record: logging.LogRecord) -> None:
        msg = self.format(record)

        if record.levelno == 60:
            self.script.log_success(msg)
        elif record.levelno == logging.ERROR:
            self.script.log_failure(msg)
        elif record.levelno == logging.WARNING:
            self.script.log_warning(msg)
        elif record.levelno == logging.INFO:
            self.script.log_info(msg)
        else:
            self.script.log_debug(msg)

class Debug(Script):
    class Meta:
        name = "debug"
        commit_default = True
        scheduling_enabled = False

    def run(self, data, commit):
        # set log level for imported modules
        logging.basicConfig(level=logging.DEBUG, handlers=[NetboxHandler(self)])

        self.log_info('beginning debug script')

        req = requests.post('https://google.com')

I also added a custom log level 60 which can be used in imported modules to log a Netbox success message. My motivation for this is I have a rather large chunk of code that needs to be reused by multiple Netbox scripts. So this lets me put it in a separate module.

# define custom log level for Netbox success
def log_success(self, message, *args, **kwargs):
    self._log(60, message, args, **kwargs)
logging.Logger.success = log_success
logger = logging.getLogger(__name__)
logger.success("it worked!")