From e683ca778ba7a3bf3b3c640771f1d2b7be53a9db Mon Sep 17 00:00:00 2001 From: hjoaquim Date: Fri, 27 Jan 2023 11:51:48 +0000 Subject: [PATCH] log user on the sdk --- openbb_terminal/core/library/operation.py | 18 ++++++++++++++++-- .../core/log/generation/user_logger.py | 4 ++-- 2 files changed, 18 insertions(+), 4 deletions(-) diff --git a/openbb_terminal/core/library/operation.py b/openbb_terminal/core/library/operation.py index 89e80903810..93cba207bd2 100644 --- a/openbb_terminal/core/library/operation.py +++ b/openbb_terminal/core/library/operation.py @@ -6,9 +6,9 @@ from typing import Any, Callable, Dict, List, Optional from inspect import signature import openbb_terminal.config_terminal as cfg - from openbb_terminal.core.library.metadata import Metadata from openbb_terminal.core.library.trail_map import TrailMap +from openbb_terminal.session.sdk_session import login # pylint: disable=import-outside-toplevel @@ -232,7 +232,7 @@ class OperationLogger: if func_module == "openbb_terminal.keys_model": - from openbb_terminal.core.log.generation.settings_logger import ( # pylint: disable=C0415 + from openbb_terminal.core.log.generation.settings_logger import ( # pylint: disable=import-outside-toplevel log_keys, ) @@ -252,6 +252,9 @@ class OperationLogger: method_result=method_result, method_chosen=self.__method_chosen, ) + self.__log_if_login( + method_chosen=self.__method_chosen, + ) self.__log_end( logger=logger, method_chosen=self.__method_chosen, @@ -275,6 +278,17 @@ class OperationLogger: extra={"func_name_override": method_chosen.__name__}, ) + @staticmethod + def __log_if_login( + method_chosen: Callable, + ): + if method_chosen.__name__ == login.__name__: + from openbb_terminal.core.log.generation.user_logger import ( # pylint: disable=import-outside-toplevel + log_user, + ) + + log_user(with_rollover=False) + @staticmethod def __log_end(logger: Logger, method_chosen: Callable): logger.info( diff --git a/openbb_terminal/core/log/generation/user_logger.py b/openbb_terminal/core/log/generation/user_logger.py index 59a07275afd..acda5a0e8b1 100644 --- a/openbb_terminal/core/log/generation/user_logger.py +++ b/openbb_terminal/core/log/generation/user_logger.py @@ -15,13 +15,13 @@ logger = logging.getLogger(__name__) def log_user(with_rollover: bool = True): """Log user""" if not User.is_guest(): - log_user_info() + _log_user_info() if with_rollover: do_rollover() -def log_user_info(): +def _log_user_info(): """Log user info""" user_info = {"user_uuid": User.get_uuid()} logger.info("USER: %s ", json.dumps(user_info))