109 lines
4.6 KiB
Python
109 lines
4.6 KiB
Python
from urllib.parse import urljoin
|
|
|
|
import requests
|
|
from celery.utils.log import get_task_logger
|
|
from django.conf import settings
|
|
from django.utils import timezone
|
|
from rest_framework import status
|
|
|
|
from apps.base.utils import live_settings
|
|
from apps.oss_installation.models import CloudConnector, CloudHeartbeat, OssInstallation
|
|
from apps.oss_installation.usage_stats import UsageStatsService
|
|
from common.custom_celery_tasks import shared_dedicated_queue_retry_task
|
|
|
|
logger = get_task_logger(__name__)
|
|
|
|
|
|
@shared_dedicated_queue_retry_task()
|
|
def send_usage_stats_report():
|
|
logger.info("Start send_usage_stats_report")
|
|
installation = OssInstallation.objects.get_or_create()[0]
|
|
enabled = live_settings.SEND_ANONYMOUS_USAGE_STATS
|
|
if enabled:
|
|
logger.info("send_usage_stats_report is enabled")
|
|
service = UsageStatsService()
|
|
service.send_usage_stats_report()
|
|
else:
|
|
logger.info("send_usage_stats_report is disabled")
|
|
installation.report_sent_at = timezone.now()
|
|
installation.save()
|
|
logger.info("Finish send_usage_stats_report")
|
|
|
|
|
|
def _setup_heartbeat_integration():
|
|
"""Setup Grafana Cloud OnCall heartbeat integration."""
|
|
cloud_heartbeat = None
|
|
api_token = live_settings.GRAFANA_CLOUD_ONCALL_TOKEN
|
|
# don't specify a team in the data, so heartbeat integration will be created in the General.
|
|
data = {"type": "formatted_webhook", "name": f"OnCall {settings.BASE_URL}"}
|
|
url = urljoin(settings.GRAFANA_CLOUD_ONCALL_API_URL, "/api/v1/integrations/")
|
|
try:
|
|
headers = {"Authorization": api_token}
|
|
r = requests.post(url=url, data=data, headers=headers, timeout=5)
|
|
if r.status_code == status.HTTP_201_CREATED:
|
|
response_data = r.json()
|
|
cloud_heartbeat, _ = CloudHeartbeat.objects.update_or_create(
|
|
defaults={"integration_id": response_data["id"], "integration_url": response_data["heartbeat"]["link"]}
|
|
)
|
|
except requests.Timeout:
|
|
logger.warning("Unable to create cloud heartbeat integration. Request timeout.")
|
|
except requests.exceptions.RequestException as e:
|
|
logger.warning(f"Unable to create cloud heartbeat integration. Request exception {str(e)}.")
|
|
return cloud_heartbeat
|
|
|
|
|
|
@shared_dedicated_queue_retry_task()
|
|
def send_cloud_heartbeat():
|
|
"""Send heartbeat to Grafana Cloud OnCall integration."""
|
|
if not live_settings.GRAFANA_CLOUD_ONCALL_HEARTBEAT_ENABLED or not live_settings.GRAFANA_CLOUD_ONCALL_TOKEN:
|
|
logger.info(
|
|
"Unable to send cloud heartbeat. Check values for GRAFANA_CLOUD_ONCALL_HEARTBEAT_ENABLED and GRAFANA_CLOUD_ONCALL_TOKEN."
|
|
)
|
|
return
|
|
|
|
logger.info("Start send cloud heartbeat")
|
|
try:
|
|
cloud_heartbeat = CloudHeartbeat.objects.get()
|
|
except CloudHeartbeat.DoesNotExist:
|
|
cloud_heartbeat = _setup_heartbeat_integration()
|
|
|
|
if cloud_heartbeat is None:
|
|
logger.warning("Unable to setup cloud heartbeat integration.")
|
|
return
|
|
cloud_heartbeat.success = False
|
|
try:
|
|
response = requests.get(cloud_heartbeat.integration_url, timeout=5)
|
|
logger.info(f"Send cloud heartbeat with response {response.status_code}")
|
|
except requests.Timeout:
|
|
logger.warning("Unable to send cloud heartbeat. Request timeout.")
|
|
except requests.exceptions.RequestException as e:
|
|
logger.warning(f"Unable to send cloud heartbeat. Request exception {str(e)}.")
|
|
else:
|
|
if response.status_code == status.HTTP_200_OK:
|
|
cloud_heartbeat.success = True
|
|
logger.info("Successfully send cloud heartbeat")
|
|
elif response.status_code == status.HTTP_403_FORBIDDEN:
|
|
# check for 403 because AlertChannelDefiningMixin returns 403 if no integration was found.
|
|
logger.info("Failed to send cloud heartbeat. Integration was not created yet")
|
|
# force re-creation on next run
|
|
cloud_heartbeat.delete()
|
|
else:
|
|
logger.info(f"Failed to send cloud heartbeat. response {response.status_code}")
|
|
# save result of cloud heartbeat if it wasn't deleted
|
|
if cloud_heartbeat.pk is not None:
|
|
cloud_heartbeat.save()
|
|
logger.info("Finish send cloud heartbeat")
|
|
|
|
|
|
@shared_dedicated_queue_retry_task()
|
|
def sync_users_with_cloud():
|
|
logger.info("Start sync_users_with_cloud")
|
|
connector = CloudConnector.objects.first()
|
|
if connector is not None:
|
|
status, error = connector.sync_users_with_cloud()
|
|
log_message = "Users synced. Status {status}."
|
|
if error:
|
|
log_message += f" Error {error}"
|
|
logger.info(log_message)
|
|
else:
|
|
logger.info("Grafana Cloud is not connected")
|