9a5fc4bac7
* feat(api): remove stage name from email subject * change(api): refactored code & SAML2 SSO SLO SLS * change(api): SAML2 extracted & custom configuration * change(api): SAML2 migrate user after signup * feat(api): return project_key with session's details * change(api): SAML2 * feat(db): tenants & users table changes for SAML2
68 lines
No EOL
3.5 KiB
Python
68 lines
No EOL
3.5 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
|
|
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=COALESCE((SELECT COUNT(*)
|
|
FROM public.sessions
|
|
INNER JOIN public.projects USING (project_id)
|
|
WHERE projects.tenant_id = all_tenants.tenant_id), 0),
|
|
t_users=COALESCE((SELECT COUNT(*)
|
|
FROM public.users
|
|
WHERE deleted_at ISNULL
|
|
AND users.tenant_id = all_tenants.tenant_id), 0)
|
|
FROM (
|
|
SELECT tenant_id
|
|
FROM public.tenants
|
|
) AS all_tenants
|
|
WHERE tenants.tenant_id = all_tenants.tenant_id
|
|
RETURNING *,(SELECT email FROM users_ee WHERE role = 'owner' AND users_ee.tenant_id = tenants.tenant_id LIMIT 1);"""
|
|
)
|
|
data = cur.fetchall()
|
|
requests.post('https://parrot.asayer.io/os/telemetry',
|
|
json={"stats": [process_data(d, edition='ee') for d in data]})
|
|
|
|
|
|
def new_client(tenant_id):
|
|
with pg_client.PostgresClient() as cur:
|
|
cur.execute(
|
|
cur.mogrify(f"""SELECT *,
|
|
(SELECT email FROM public.users WHERE tenant_id=%(tenant_id)s) AS email
|
|
FROM public.tenants
|
|
WHERE tenant_id=%(tenant_id)s;""", {"tenant_id": tenant_id}))
|
|
data = cur.fetchone()
|
|
requests.post('https://parrot.asayer.io/os/signup', json=process_data(data, edition='ee')) |