laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

send template info in init telemetry event (#3153)

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
reflex/utils/prerequisites.py
View File

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

13
reflex/utils/telemetry.py
View File

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