adds primaryUsage to Posthog logs (#5037)

This commit is contained in:
Henrique Joaquim
2023-05-19 19:27:40 +01:00
committed by GitHub
parent 9e610d4495
commit 8924f880d1
2 changed files with 8 additions and 1 deletions

View File

@@ -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:

View File

@@ -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)