44 lines
2 KiB
Python
44 lines
2 KiB
Python
from chalicelib.utils import pg_client
|
|
import requests
|
|
|
|
|
|
def process_data(data, edition='fos'):
|
|
return {
|
|
'edition': edition,
|
|
'tracking': data["opt_out"],
|
|
'version': data["version_number"],
|
|
'user_id': data["user_id"],
|
|
'owner_email': None if data["opt_out"] else data["email"],
|
|
'organization_name': None if data["opt_out"] else data["name"],
|
|
'users_count': data["t_users"],
|
|
'projects_count': data["t_projects"],
|
|
'sessions_count': data["t_sessions"],
|
|
'integrations_count': data["t_integrations"]
|
|
}
|
|
|
|
|
|
def compute():
|
|
with pg_client.PostgresClient() as cur:
|
|
cur.execute(
|
|
f"""UPDATE public.tenants
|
|
SET t_integrations = COALESCE((SELECT COUNT(DISTINCT provider) FROM public.integrations) +
|
|
(SELECT COUNT(*) FROM public.webhooks WHERE type = 'slack') +
|
|
(SELECT COUNT(*) FROM public.jira_cloud), 0),
|
|
t_projects=COALESCE((SELECT COUNT(*) FROM public.projects WHERE deleted_at ISNULL), 0),
|
|
t_sessions=COALESCE((SELECT COUNT(*) FROM public.sessions), 0),
|
|
t_users=COALESCE((SELECT COUNT(*) FROM public.users WHERE deleted_at ISNULL), 0)
|
|
RETURNING name,t_integrations,t_projects,t_sessions,t_users,user_id,opt_out,
|
|
(SELECT openreplay_version()) AS version_number,(SELECT email FROM public.users WHERE role = 'owner' LIMIT 1);"""
|
|
)
|
|
data = cur.fetchone()
|
|
requests.post('https://api.openreplay.com/os/telemetry', json={"stats": [process_data(data)]})
|
|
|
|
|
|
def new_client():
|
|
with pg_client.PostgresClient() as cur:
|
|
cur.execute(
|
|
f"""SELECT *,
|
|
(SELECT email FROM public.users WHERE role='owner' LIMIT 1) AS email
|
|
FROM public.tenants;""")
|
|
data = cur.fetchone()
|
|
requests.post('https://api.openreplay.com/os/signup', json=process_data(data))
|