2cb786acc8
* fix(chalice): fixed experimental sessions search for table of X * refactor(chalice): upgraded dependencies * refactor(alerts): upgraded dependencies * refactor(crons): upgraded dependencies * refactor(chalice): enhanced telemetry
75 lines
4.1 KiB
Python
75 lines
4.1 KiB
Python
from chalicelib.utils import pg_client
|
|
from chalicelib.core import license
|
|
import requests
|
|
|
|
|
|
def process_data(data):
|
|
return {
|
|
'edition': license.EDITION,
|
|
'tracking': data["opt_out"],
|
|
'version': data["version_number"],
|
|
'user_id': data["tenant_key"],
|
|
'tenant_key': data["tenant_key"],
|
|
'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(long_query=True) as cur:
|
|
cur.execute(
|
|
f"""UPDATE public.tenants
|
|
SET t_integrations = COALESCE((SELECT COUNT(DISTINCT provider)
|
|
FROM public.integrations
|
|
INNER JOIN public.projects USING (project_id)
|
|
WHERE projects.tenant_id = all_tenants.tenant_id) +
|
|
(SELECT COUNT(*)
|
|
FROM public.webhooks
|
|
WHERE webhooks.tenant_id = all_tenants.tenant_id
|
|
AND type = 'slack') +
|
|
(SELECT COUNT(*)
|
|
FROM public.jira_cloud
|
|
INNER JOIN public.users USING (user_id)
|
|
WHERE users.tenant_id = all_tenants.tenant_id), 0),
|
|
t_projects=COALESCE((SELECT COUNT(*)
|
|
FROM public.projects
|
|
WHERE deleted_at ISNULL
|
|
AND projects.tenant_id = all_tenants.tenant_id), 0),
|
|
t_sessions=t_sessions + COALESCE((SELECT COUNT(*)
|
|
FROM public.sessions INNER JOIN public.projects USING (project_id)
|
|
WHERE projects.tenant_id = all_tenants.tenant_id
|
|
AND start_ts >= (SELECT last_telemetry FROM tenants)
|
|
AND start_ts <=CAST(EXTRACT(epoch FROM date_trunc('day', now())) * 1000 AS BIGINT)), 0),
|
|
t_users=COALESCE((SELECT COUNT(*)
|
|
FROM public.users
|
|
WHERE deleted_at ISNULL
|
|
AND users.tenant_id = all_tenants.tenant_id), 0),
|
|
last_telemetry=CAST(EXTRACT(epoch FROM date_trunc('day', now())) * 1000 AS BIGINT)
|
|
FROM (SELECT tenant_id
|
|
FROM public.tenants
|
|
) AS all_tenants
|
|
WHERE tenants.tenant_id = all_tenants.tenant_id
|
|
RETURNING name,t_integrations,t_projects,t_sessions,t_users,tenant_key,opt_out,
|
|
(SELECT openreplay_version()) AS version_number,
|
|
(SELECT email FROM public.users WHERE role = 'owner' AND users.tenant_id=tenants.tenant_id LIMIT 1);"""
|
|
)
|
|
data = cur.fetchall()
|
|
if len(data) > 0:
|
|
requests.post('https://api.openreplay.com/os/telemetry',
|
|
json={"stats": [process_data(d) for d in data]})
|
|
|
|
|
|
def new_client(tenant_id):
|
|
with pg_client.PostgresClient() as cur:
|
|
cur.execute(
|
|
cur.mogrify(f"""SELECT *, openreplay_version() AS version_number,
|
|
(SELECT email FROM public.users WHERE tenant_id=%(tenant_id)s AND role='owner' LIMIT 1) AS email
|
|
FROM public.tenants
|
|
WHERE tenant_id=%(tenant_id)s
|
|
LIMIT 1;""", {"tenant_id": tenant_id}))
|
|
data = cur.fetchone()
|
|
requests.post('https://api.openreplay.com/os/signup', json=process_data(data))
|