send template info in init telemetry event (#3153)

This commit is contained in:
Martin Xu 2024-04-24 06:47:33 -07:00 committed by GitHub
parent b084c805d4
commit a5f9001cf9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 4 deletions

View File

@ -1410,4 +1410,4 @@ def initialize_app(app_name: str, template: str | None = None):
template_url=template_url,
)
telemetry.send("init")
telemetry.send("init", template=template)

View File

@ -87,11 +87,12 @@ def _raise_on_missing_project_hash() -> bool:
return True
def _prepare_event(event: str) -> dict:
def _prepare_event(event: str, **kwargs) -> dict:
"""Prepare the event to be sent to the PostHog server.
Args:
event: The event name.
kwargs: Additional data to send with the event.
Returns:
The event data.
@ -122,6 +123,11 @@ def _prepare_event(event: str) -> dict:
"python_version": get_python_version(),
"cpu_count": get_cpu_count(),
"memory": get_memory(),
**(
{"template": template}
if (template := kwargs.get("template")) is not None
else {}
),
},
"timestamp": stamp,
}
@ -135,12 +141,13 @@ def _send_event(event_data: dict) -> bool:
return False
def send(event: str, telemetry_enabled: bool | None = None) -> bool:
def send(event: str, telemetry_enabled: bool | None = None, **kwargs) -> bool:
"""Send anonymous telemetry for Reflex.
Args:
event: The event name.
telemetry_enabled: Whether to send the telemetry (If None, get from config).
kwargs: Additional data to send with the event.
Returns:
Whether the telemetry was sent successfully.
@ -155,7 +162,7 @@ def send(event: str, telemetry_enabled: bool | None = None) -> bool:
if not telemetry_enabled:
return False
event_data = _prepare_event(event)
event_data = _prepare_event(event, **kwargs)
if not event_data:
return False