diff --git a/src/pythonjsonlogger/jsonlogger.py b/src/pythonjsonlogger/jsonlogger.py index 519a64d..8bbffb0 100644 --- a/src/pythonjsonlogger/jsonlogger.py +++ b/src/pythonjsonlogger/jsonlogger.py @@ -162,7 +162,7 @@ def __init__( self._skip_fields = dict(zip(self._required_fields, self._required_fields)) self._skip_fields.update(self.reserved_attrs) - def _str_to_fn(self, fn_as_str): + def _str_to_fn(self, fn_as_str: str) -> Union[Any, Callable]: """ If the argument is not a string, return whatever was passed in. Parses a string such as package.module.function, imports the module @@ -172,10 +172,9 @@ def _str_to_fn(self, fn_as_str): """ if not isinstance(fn_as_str, str): return fn_as_str - - path, _, function = fn_as_str.rpartition(".") + path, _, _function = fn_as_str.rpartition(".") module = importlib.import_module(path) - return getattr(module, function) + return getattr(module, _function) def parse(self) -> List[str]: """ @@ -197,8 +196,7 @@ def parse(self) -> List[str]: if self._fmt: return formatter_style_pattern.findall(self._fmt) - else: - return [] + return [] def add_fields( self, @@ -227,19 +225,19 @@ def add_fields( self._perform_rename_log_fields(log_record) - def _perform_rename_log_fields(self, log_record): + def _perform_rename_log_fields(self, log_record: Dict[str, Any]) -> None: for old_field_name, new_field_name in self.rename_fields.items(): log_record[new_field_name] = log_record[old_field_name] del log_record[old_field_name] - def process_log_record(self, log_record): + def process_log_record(self, log_record: Dict[str, Any]): """ Override this method to implement custom logic on the possibly ordered dictionary. """ return log_record - def jsonify_log_record(self, log_record): + def jsonify_log_record(self, log_record: Dict[str, Any]): """Returns a json string of the log record.""" return self.json_serializer( log_record,