mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 05:21:48 +00:00
log user on the sdk
This commit is contained in:
@@ -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(
|
||||
|
||||
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user