diff options
| author | Henrique Joaquim <h.joaquim@campus.fct.unl.pt> | 2023-05-19 19:27:40 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-19 18:27:40 +0000 |
| commit | 8924f880d1fc4b7c56c5671be22f68a24db4fa58 (patch) | |
| tree | 37788158feec962aab756a0dde5bea0b924890b2 | |
| parent | 9e610d449579206d6abd273f8b54e815fff31271 (diff) | |
adds primaryUsage to Posthog logs (#5037)
| -rw-r--r-- | openbb_terminal/core/models/profile_model.py | 3 | ||||
| -rw-r--r-- | openbb_terminal/loggers.py | 6 |
2 files changed, 8 insertions, 1 deletions
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) |