cb8d87e367
* refactor(chalice): upgraded dependencies refactor(crons): upgraded dependencies refactor(alerts): upgraded dependencies * fix(chalice): fixed boarding * fix(chalice): fixed assign session * refactor(assist-stats): upgraded dependencies * fixed(assist-stats): fixed import issue * fix(chalice): changed env vars * fix(chalice): fixed search sessions for EE
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
from chalicelib.core.log_tools import log_tools
|
|
from schemas import schemas
|
|
|
|
IN_TY = "newrelic"
|
|
|
|
|
|
def get_all(tenant_id):
|
|
return log_tools.get_all_by_tenant(tenant_id=tenant_id, integration=IN_TY)
|
|
|
|
|
|
def get(project_id):
|
|
return log_tools.get(project_id=project_id, integration=IN_TY)
|
|
|
|
|
|
def update(tenant_id, project_id, changes):
|
|
options = {}
|
|
if "region" in changes:
|
|
options["region"] = changes["region"]
|
|
if "applicationId" in changes:
|
|
options["applicationId"] = changes["applicationId"]
|
|
if "xQueryKey" in changes:
|
|
options["xQueryKey"] = changes["xQueryKey"]
|
|
|
|
return log_tools.edit(project_id=project_id, integration=IN_TY, changes=options)
|
|
|
|
|
|
def add(tenant_id, project_id, application_id, x_query_key, region):
|
|
# region=False => US; region=True => EU
|
|
options = {"applicationId": application_id, "xQueryKey": x_query_key, "region": region}
|
|
return log_tools.add(project_id=project_id, integration=IN_TY, options=options)
|
|
|
|
|
|
def delete(tenant_id, project_id):
|
|
return log_tools.delete(project_id=project_id, integration=IN_TY)
|
|
|
|
|
|
def add_edit(tenant_id, project_id, data: schemas.IntegrationNewrelicSchema):
|
|
s = get(project_id)
|
|
if s is not None:
|
|
return update(tenant_id=tenant_id, project_id=project_id,
|
|
changes={"applicationId": data.application_id,
|
|
"xQueryKey": data.x_query_key,
|
|
"region": data.region})
|
|
else:
|
|
return add(tenant_id=tenant_id,
|
|
project_id=project_id,
|
|
application_id=data.application_id,
|
|
x_query_key=data.x_query_key,
|
|
region=data.region)
|