48bfe86d62
* Add cloud connection statuses on user page * Add fixes for oncall hobby docker-compose installation * Fix for links and for cloud user status at User settings * Delete cloud api token on cloud disconnect * Merge branch 'dev' into cloud_connection_statuses_on_user_page * Fix cloud statuses for users * Fix usagestats service * Fix phone verification message in users table * added request after syncing user * Add endpoint to create CloudHeartbeat and polish code * Fix imports * Check token and heartbeat setting in setup_hertbeat_integration * Add macthed_users_count in cloud users * Sync users on token change * Fix query param * Fix tests * Heartbit button logic, tab width fix, coount users fix * Solve problem of existent cloud heartbeat integration * Solve problem of existent cloud heartbeat integration 2 * Solve problem of existent cloud heartbeat integration 3 * fix build * build fix, styles for env variables description Co-authored-by: Ildar Iskhakov <ildar.iskhakov@grafana.com> Co-authored-by: Yulia Shanyrova <yulia.shanyrova@grafana.com>
106 lines
4.2 KiB
Python
106 lines
4.2 KiB
Python
from contextlib import suppress
|
|
|
|
from django.conf import settings
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from django.http import HttpResponse
|
|
from rest_framework import status, viewsets
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from telegram import error
|
|
|
|
from apps.api.permissions import IsAdmin
|
|
from apps.api.serializers.live_setting import LiveSettingSerializer
|
|
from apps.auth_token.auth import PluginAuthentication
|
|
from apps.base.models import LiveSetting
|
|
from apps.base.utils import live_settings
|
|
from apps.oss_installation.tasks import sync_users_with_cloud
|
|
from apps.slack.tasks import unpopulate_slack_user_identities
|
|
from apps.telegram.client import TelegramClient
|
|
from apps.telegram.tasks import register_telegram_webhook
|
|
from apps.user_management.models import User
|
|
from common.api_helpers.mixins import PublicPrimaryKeyMixin
|
|
|
|
|
|
class LiveSettingViewSet(PublicPrimaryKeyMixin, viewsets.ModelViewSet):
|
|
serializer_class = LiveSettingSerializer
|
|
authentication_classes = (PluginAuthentication,)
|
|
permission_classes = (IsAuthenticated, IsAdmin)
|
|
|
|
def dispatch(self, request, *args, **kwargs):
|
|
if not settings.FEATURE_LIVE_SETTINGS_ENABLED:
|
|
return HttpResponse(status=status.HTTP_404_NOT_FOUND)
|
|
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
def get_queryset(self):
|
|
LiveSetting.populate_settings_if_needed()
|
|
queryset = LiveSetting.objects.filter(name__in=LiveSetting.AVAILABLE_NAMES).order_by("name")
|
|
search = self.request.query_params.get("search", None)
|
|
if search:
|
|
queryset = queryset.filter(name=search)
|
|
return queryset
|
|
|
|
def perform_update(self, serializer):
|
|
new_value = serializer.validated_data["value"]
|
|
self._update_hook(new_value)
|
|
instance = serializer.save()
|
|
sync_users = self.request.query_params.get("sync_users", "true") == "true"
|
|
if instance.name == "GRAFANA_CLOUD_ONCALL_TOKEN" and sync_users:
|
|
sync_users_with_cloud.apply_async()
|
|
|
|
def perform_destroy(self, instance):
|
|
new_value = instance.default_value
|
|
self._update_hook(new_value)
|
|
|
|
super().perform_destroy(instance)
|
|
|
|
def _update_hook(self, new_value):
|
|
instance = self.get_object()
|
|
|
|
if instance.name == "TELEGRAM_TOKEN":
|
|
try:
|
|
old_token = live_settings.TELEGRAM_TOKEN
|
|
except ImproperlyConfigured:
|
|
old_token = None
|
|
|
|
if old_token != new_value:
|
|
self._reset_telegram_integration(new_token=new_value)
|
|
|
|
for setting_name in ["SLACK_CLIENT_OAUTH_ID", "SLACK_CLIENT_OAUTH_SECRET"]:
|
|
if instance.name == setting_name:
|
|
if getattr(live_settings, setting_name) != new_value:
|
|
organization = self.request.auth.organization
|
|
sti = organization.slack_team_identity
|
|
if sti is not None:
|
|
unpopulate_slack_user_identities.apply_async((sti.pk, True), countdown=0)
|
|
|
|
if instance.name == "GRAFANA_CLOUD_ONCALL_TOKEN":
|
|
from apps.oss_installation.models import CloudConnector
|
|
|
|
try:
|
|
old_token = live_settings.GRAFANA_CLOUD_ONCALL_TOKEN
|
|
except ImproperlyConfigured:
|
|
old_token = None
|
|
|
|
if old_token != new_value:
|
|
CloudConnector.remove_sync()
|
|
|
|
def _reset_telegram_integration(self, new_token):
|
|
# tell Telegram to cancel sending events from old bot
|
|
with suppress(ImproperlyConfigured, error.InvalidToken, error.Unauthorized):
|
|
old_client = TelegramClient()
|
|
old_client.api_client.delete_webhook()
|
|
|
|
# delete telegram channels for current team
|
|
organization = self.request.auth.organization
|
|
organization.telegram_channel.all().delete()
|
|
|
|
# delete telegram connectors for users in team
|
|
users_with_telegram_connector = User.objects.filter(
|
|
organization=organization, telegram_connection__isnull=False
|
|
).distinct()
|
|
|
|
for user in users_with_telegram_connector:
|
|
user.telegram_connection.delete()
|
|
|
|
# tell Telegram to send updates to new bot
|
|
register_telegram_webhook.delay(token=new_token)
|