69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
from contextlib import suppress
|
|
from functools import lru_cache
|
|
|
|
import frappe
|
|
from frappe.utils.caching import site_cache
|
|
|
|
from posthog import Posthog # isort: skip
|
|
|
|
POSTHOG_PROJECT_FIELD = "posthog_project_id"
|
|
POSTHOG_HOST_FIELD = "posthog_host"
|
|
|
|
|
|
def is_enabled():
|
|
return bool(
|
|
frappe.conf.get(POSTHOG_HOST_FIELD)
|
|
and frappe.conf.get(POSTHOG_PROJECT_FIELD)
|
|
and frappe.get_system_settings("enable_telemetry")
|
|
)
|
|
|
|
|
|
def init_telemetry():
|
|
"""Init posthog for server side telemetry."""
|
|
if hasattr(frappe.local, "posthog"):
|
|
return
|
|
|
|
if not is_enabled():
|
|
return
|
|
|
|
posthog_host = frappe.conf.get(POSTHOG_HOST_FIELD)
|
|
posthog_project_id = frappe.conf.get(POSTHOG_PROJECT_FIELD)
|
|
|
|
with suppress(Exception):
|
|
frappe.local.posthog = _get_posthog_instance(posthog_project_id, posthog_host)
|
|
|
|
# Background jobs might exit before flushing telemetry, so explicitly flush queue
|
|
if frappe.job:
|
|
frappe.job.after_job.add(flush_telemetry)
|
|
|
|
|
|
@lru_cache
|
|
def _get_posthog_instance(project_id, host):
|
|
return Posthog(project_id, host=host, timeout=5, max_retries=3)
|
|
|
|
|
|
def flush_telemetry():
|
|
ph: Posthog = getattr(frappe.local, "posthog", None)
|
|
with suppress(Exception):
|
|
ph and ph.flush()
|
|
|
|
|
|
def capture(event, app, **kwargs):
|
|
init_telemetry()
|
|
ph: Posthog = getattr(frappe.local, "posthog", None)
|
|
with suppress(Exception):
|
|
ph and ph.capture(distinct_id=frappe.local.site, event=f"{app}_{event}", **kwargs)
|
|
|
|
|
|
def capture_doc(doc, action):
|
|
from frappe.utils.telemetry import site_age
|
|
|
|
with suppress(Exception):
|
|
age = site_age()
|
|
if not age or age > 15:
|
|
return
|
|
|
|
if doc.get("__islocal") or not doc.get("name"):
|
|
capture("document_created", "frappe", properties={"doctype": doc.doctype, "action": "Insert"})
|
|
else:
|
|
capture("document_modified", "frappe", properties={"doctype": doc.doctype, "action": action})
|