openreplay/api/chalicelib/core/log_tool_rollbar.py
2024-02-06 14:18:50 +01:00

39 lines
1.3 KiB
Python

from chalicelib.core import log_tools
from schemas import schemas
IN_TY = "rollbar"
async def get_all(tenant_id):
return await log_tools.get_all_by_tenant(tenant_id=tenant_id, integration=IN_TY)
async def get(project_id):
return await log_tools.get(project_id=project_id, integration=IN_TY)
async def update(tenant_id, project_id, changes):
options = {}
if "accessToken" in changes:
options["accessToken"] = changes["accessToken"]
return await log_tools.edit(project_id=project_id, integration=IN_TY, changes=options)
async def add(tenant_id, project_id, access_token):
options = {"accessToken": access_token}
return await log_tools.add(project_id=project_id, integration=IN_TY, options=options)
async def delete(tenant_id, project_id):
return await log_tools.delete(project_id=project_id, integration=IN_TY)
async def add_edit(tenant_id, project_id, data: schemas.IntegrationRollbarSchema):
s = await get(project_id)
if s is not None:
return await update(tenant_id=tenant_id, project_id=project_id,
changes={"accessToken": data.access_token})
else:
return await add(tenant_id=tenant_id,
project_id=project_id,
access_token=data.access_token)