From 8924f880d1fc4b7c56c5671be22f68a24db4fa58 Mon Sep 17 00:00:00 2001 From: Henrique Joaquim Date: Fri, 19 May 2023 19:27:40 +0100 Subject: adds primaryUsage to Posthog logs (#5037) --- openbb_terminal/core/models/profile_model.py | 3 +++ openbb_terminal/loggers.py | 6 +++++- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/openbb_terminal/core/models/profile_model.py b/openbb_terminal/core/models/profile_model.py index 9cbe215ddd2..56bebae2d11 100644 --- a/openbb_terminal/core/models/profile_model.py +++ b/openbb_terminal/core/models/profile_model.py @@ -14,6 +14,7 @@ class ProfileModel(BaseModel): uuid: str = "" email: str = "" username: str = "" + primary_usage: str = "" remember: bool = True def load_user_info(self, session: dict, email: str, remember: bool): @@ -28,11 +29,13 @@ class ProfileModel(BaseModel): remember : bool Remember the session. """ + self.token_type = session.get("token_type", "") self.token = session.get("access_token", "") self.uuid = session.get("uuid", "") self.email = email self.username = self.email[: self.email.find("@")] + self.primary_usage = session.get("primary_usage", "") self.remember = remember def get_uuid(self) -> str: diff --git a/openbb_terminal/loggers.py b/openbb_terminal/loggers.py index 74dcda9e213..e4a8c37596c 100644 --- a/openbb_terminal/loggers.py +++ b/openbb_terminal/loggers.py @@ -184,7 +184,11 @@ class PosthogHandler(logging.Handler): ): self.logged_in = True openbb_posthog.identify( - get_user_uuid(), {"email": get_current_user().profile.email} + get_user_uuid(), + { + "email": get_current_user().profile.email, + "primaryUsage": get_current_user().profile.primary_usage, + }, ) openbb_posthog.alias(get_user_uuid(), app_settings.identifier) -- cgit v1.2.3