Coverage for chalicelib/core/telemetry.py: 61%

1from chalicelib.utils import pg_client 

2import requests 

3from chalicelib.core import license 

4 

5 

6def process_data(data): 

7 return { 

8 'edition': license.EDITION, 

9 'tracking': data["opt_out"], 

10 'version': data["version_number"], 

11 'user_id': data["tenant_key"], 

12 'tenant_key': data["tenant_key"], 

13 'owner_email': None if data["opt_out"] else data["email"], 

14 'organization_name': None if data["opt_out"] else data["name"], 

15 'users_count': data["t_users"], 

16 'projects_count': data["t_projects"], 

17 'sessions_count': data["t_sessions"], 

18 'integrations_count': data["t_integrations"] 

19 } 

20 

21 

22def compute(): 

23 with pg_client.PostgresClient(long_query=True) as cur: 

24 cur.execute( 

25 f"""UPDATE public.tenants 

26 SET t_integrations = COALESCE((SELECT COUNT(DISTINCT provider) FROM public.integrations) + 

27 (SELECT COUNT(*) FROM public.webhooks WHERE type = 'slack') + 

28 (SELECT COUNT(*) FROM public.jira_cloud), 0), 

29 t_projects=COALESCE((SELECT COUNT(*) FROM public.projects WHERE deleted_at ISNULL), 0), 

30 t_sessions=t_sessions + COALESCE((SELECT COUNT(*) 

31 FROM public.sessions 

32 WHERE start_ts >= (SELECT last_telemetry FROM tenants) 

33 AND start_ts <=CAST(EXTRACT(epoch FROM date_trunc('day', now())) * 1000 AS BIGINT)), 0), 

34 t_users=COALESCE((SELECT COUNT(*) FROM public.users WHERE deleted_at ISNULL), 0), 

35 last_telemetry=CAST(EXTRACT(epoch FROM date_trunc('day', now())) * 1000 AS BIGINT) 

36 RETURNING name,t_integrations,t_projects,t_sessions,t_users,tenant_key,opt_out, 

37 (SELECT openreplay_version()) AS version_number,(SELECT email FROM public.users WHERE role = 'owner' LIMIT 1);""" 

38 ) 

39 data = cur.fetchone() 

40 if len(data) > 0: 

41 requests.post('https://api.openreplay.com/os/telemetry', json={"stats": [process_data(data)]}) 

42 

43 

44def new_client(): 

45 with pg_client.PostgresClient() as cur: 

46 cur.execute( 

47 f"""SELECT *, openreplay_version() AS version_number, 

48 (SELECT email FROM public.users WHERE role='owner' LIMIT 1) AS email 

49 FROM public.tenants 

50 LIMIT 1;""") 

51 data = cur.fetchone() 

52 requests.post('https://api.openreplay.com/os/signup', json=process_data(data))