move analytics code from scripts to skyvern module (#32)
This commit is contained in:
@@ -1,71 +0,0 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import importlib.metadata
|
||||
import platform
|
||||
from typing import Any, Dict
|
||||
|
||||
import typer
|
||||
from posthog import Posthog
|
||||
|
||||
from skyvern.forge.sdk.settings_manager import SettingsManager
|
||||
|
||||
posthog = Posthog(
|
||||
"phc_bVT2ugnZhMHRWqMvSRHPdeTjaPxQqT3QSsI3r5FlQR5",
|
||||
host="https://app.posthog.com",
|
||||
disable_geoip=False,
|
||||
)
|
||||
|
||||
DISTINCT_ID = "oss"
|
||||
|
||||
|
||||
def get_oss_version() -> str:
|
||||
try:
|
||||
return importlib.metadata.version("skyvern")
|
||||
except Exception:
|
||||
return "unknown"
|
||||
|
||||
|
||||
def analytics_metadata() -> Dict[str, Any]:
|
||||
return {
|
||||
"os": platform.system().lower(),
|
||||
"oss_version": get_oss_version(),
|
||||
"machine": platform.machine(),
|
||||
"platform": platform.platform(),
|
||||
"python_version": platform.python_version(),
|
||||
"environment": SettingsManager.get_settings().ENV,
|
||||
}
|
||||
|
||||
|
||||
def capture(
|
||||
event: str,
|
||||
data: dict[str, Any] | None = None,
|
||||
) -> None:
|
||||
# If telemetry is disabled, don't send any data
|
||||
if not SettingsManager.get_settings().SKYVERN_TELEMETRY:
|
||||
return
|
||||
|
||||
distinct_id = SettingsManager.get_settings().ANALYTICS_ID
|
||||
|
||||
payload: dict[str, Any] = data or {}
|
||||
try:
|
||||
posthog.capture(distinct_id=distinct_id, event=event, properties=payload)
|
||||
except Exception as e:
|
||||
payload.update(
|
||||
{
|
||||
"capture_error": str(e),
|
||||
}
|
||||
)
|
||||
posthog.capture(
|
||||
distinct_id=distinct_id,
|
||||
event="failure",
|
||||
properties=payload,
|
||||
)
|
||||
|
||||
|
||||
# This is the main function that will be called by the typer CLI. This is separate from capture because typer
|
||||
# doesn't support dict type input arguments.
|
||||
def capture_simple(event: str) -> None:
|
||||
capture(event)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
typer.run(capture_simple)
|
||||
Reference in New Issue
Block a user