6b40f95033
Co-authored-by: Eve832 <eve.meelan@grafana.com>
Co-authored-by: Francisco Montes de Oca <nevermind89x@gmail.com>
Co-authored-by: Ildar Iskhakov <ildar.iskhakov@grafana.com>
Co-authored-by: Innokentii Konstantinov <innokenty.konstantinov@grafana.com>
Co-authored-by: Julia <ferril.darkdiver@gmail.com>
Co-authored-by: maskin25 <kengurek@gmail.com>
Co-authored-by: Matias Bordese <mbordese@gmail.com>
Co-authored-by: Matvey Kukuy <motakuk@gmail.com>
Co-authored-by: Michael Derynck <michael.derynck@grafana.com>
Co-authored-by: Richard Hartmann <richih@richih.org>
Co-authored-by: Robby Milo <robbymilo@fastmail.com>
Co-authored-by: Timur Olzhabayev <timur.olzhabayev@grafana.com>
Co-authored-by: Vadim Stepanov <vadimkerr@gmail.com>
Co-authored-by: Yulia Shanyrova <yulia.shanyrova@grafana.com>
95 lines
4.1 KiB
Python
95 lines
4.1 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 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")
|