9c550af721
* Draft support of oncall-gw * Clean up * Create oncall connector on org create in gcom * Naming fixes * Rework oncall-gateway package. \nMove it from apps. * Fix typo
96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
import requests
|
|
from celery.utils.log import get_task_logger
|
|
from django.conf import settings
|
|
|
|
from common.custom_celery_tasks import shared_dedicated_queue_retry_task
|
|
|
|
from .oncall_gateway_client import OnCallGatewayAPIClient
|
|
|
|
task_logger = get_task_logger(__name__)
|
|
|
|
|
|
@shared_dedicated_queue_retry_task(
|
|
autoretry_for=(Exception,),
|
|
retry_backoff=True,
|
|
max_retries=None,
|
|
)
|
|
def create_oncall_connector_async(oncall_org_id, backend):
|
|
client = OnCallGatewayAPIClient(settings.ONCALL_GATEWAY_URL, settings.ONCALL_GATEWAY_API_TOKEN)
|
|
try:
|
|
client.post_oncall_connector(oncall_org_id, backend)
|
|
except requests.exceptions.HTTPError as http_exc:
|
|
# TODO: decide which http codes to retry
|
|
print(http_exc.response)
|
|
if http_exc.response.status_code == 409:
|
|
task_logger.error(
|
|
f"Failed to create OnCallConnector oncall_org_id={oncall_org_id} backend={backend} exc={http_exc}"
|
|
)
|
|
else:
|
|
raise http_exc
|
|
except Exception as e:
|
|
task_logger.error(f"Failed to create OnCallConnector oncall_org_id={oncall_org_id} backend={backend} exc={e}")
|
|
raise e
|
|
|
|
|
|
@shared_dedicated_queue_retry_task(
|
|
autoretry_for=(Exception,),
|
|
retry_backoff=True,
|
|
max_retries=None,
|
|
)
|
|
def delete_oncall_connector_async(oncall_org_id):
|
|
client = OnCallGatewayAPIClient(settings.ONCALL_GATEWAY_URL, settings.ONCALL_GATEWAY_API_TOKEN)
|
|
try:
|
|
client.delete_slack_connector(oncall_org_id)
|
|
except requests.exceptions.HTTPError as http_exc:
|
|
if http_exc.response.status_code == 404:
|
|
# 404 indicates than resourse was deleted already
|
|
return
|
|
else:
|
|
task_logger.error(f"Failed to delete OnCallConnector oncall_org_id={oncall_org_id} exc={http_exc}")
|
|
raise http_exc
|
|
except Exception as e:
|
|
task_logger.error(f"Failed to delete OnCallConnector oncall_org_id={oncall_org_id} exc={e}")
|
|
raise e
|
|
|
|
|
|
@shared_dedicated_queue_retry_task(
|
|
autoretry_for=(Exception,),
|
|
retry_backoff=True,
|
|
max_retries=None,
|
|
)
|
|
def create_slack_connector_async(slack_id, backend):
|
|
client = OnCallGatewayAPIClient(settings.ONCALL_GATEWAY_URL, settings.ONCALL_GATEWAY_API_TOKEN)
|
|
try:
|
|
client.post_slack_connector(slack_id, backend)
|
|
except requests.exceptions.HTTPError as http_exc:
|
|
# TODO: decide which http codes to retry
|
|
if http_exc.response.status_code == 409:
|
|
task_logger.error(
|
|
f"Failed to create SlackConnector oncall_org_id={slack_id} backend={backend} exc={http_exc}"
|
|
)
|
|
else:
|
|
raise http_exc
|
|
except Exception as e:
|
|
task_logger.error(f"Failed to create SlackConnector slack_id={slack_id} backend={backend} exc={e}")
|
|
raise e
|
|
|
|
|
|
@shared_dedicated_queue_retry_task(
|
|
autoretry_for=(Exception,),
|
|
retry_backoff=True,
|
|
max_retries=None,
|
|
)
|
|
def delete_slack_connector_async(slack_id):
|
|
client = OnCallGatewayAPIClient(settings.ONCALL_GATEWAY_URL, settings.ONCALL_GATEWAY_API_TOKEN)
|
|
try:
|
|
client.delete_slack_connector(slack_id)
|
|
except requests.exceptions.HTTPError as http_exc:
|
|
if http_exc.response.status_code == 404:
|
|
# 404 indicates than resourse was deleted already
|
|
return
|
|
else:
|
|
task_logger.error(f"Failed to delete OnCallConnector slack_id={slack_id} exc={http_exc}")
|
|
raise http_exc
|
|
except Exception as e:
|
|
task_logger.error(f"Failed to delete OnCallConnector slack_id={slack_id} exc={e}")
|
|
raise e
|