diff --git a/api/onnx_web/logging.py b/api/onnx_web/logging.py index fa0e5227..fd85e410 100644 --- a/api/onnx_web/logging.py +++ b/api/onnx_web/logging.py @@ -7,7 +7,7 @@ from yaml import safe_load logging_path = environ.get("ONNX_WEB_LOGGING_PATH", "./logging.yaml") -def addLoggingLevel(levelName, levelNum, methodName=None): +def add_logging_level(level_name, level_num, method_name=None): """ From https://stackoverflow.com/a/35804945 @@ -34,30 +34,30 @@ def addLoggingLevel(levelName, levelNum, methodName=None): 5 """ - if not methodName: - methodName = levelName.lower() + if not method_name: + method_name = level_name.lower() - if hasattr(logging, levelName): - raise AttributeError("{} already defined in logging module".format(levelName)) - if hasattr(logging, methodName): - raise AttributeError("{} already defined in logging module".format(methodName)) - if hasattr(logging.getLoggerClass(), methodName): - raise AttributeError("{} already defined in logger class".format(methodName)) + if hasattr(logging, level_name): + raise AttributeError("{} already defined in logging module".format(level_name)) + if hasattr(logging, method_name): + raise AttributeError("{} already defined in logging module".format(method_name)) + if hasattr(logging.getLoggerClass(), method_name): + raise AttributeError("{} already defined in logger class".format(method_name)) # This method was inspired by the answers to Stack Overflow post # http://stackoverflow.com/q/2183233/2988730, especially # http://stackoverflow.com/a/13638084/2988730 def logForLevel(self, message, *args, **kwargs): - if self.isEnabledFor(levelNum): - self._log(levelNum, message, args, **kwargs) + if self.isEnabledFor(level_num): + self._log(level_num, message, args, **kwargs) def logToRoot(message, *args, **kwargs): - logging.log(levelNum, message, *args, **kwargs) + logging.log(level_num, message, *args, **kwargs) - logging.addLevelName(levelNum, levelName) - setattr(logging, levelName, levelNum) - setattr(logging.getLoggerClass(), methodName, logForLevel) - setattr(logging, methodName, logToRoot) + logging.addLevelName(level_num, level_name) + setattr(logging, level_name, level_num) + setattr(logging.getLoggerClass(), method_name, logForLevel) + setattr(logging, method_name, logToRoot) # setup logging config before anything else loads @@ -66,6 +66,6 @@ try: with open(logging_path, "r") as f: config_logging = safe_load(f) dictConfig(config_logging) - addLoggingLevel("TRACE", logging.DEBUG - 5) + add_logging_level("TRACE", logging.DEBUG - 5) except Exception as err: print("error loading logging config: %s" % (err))