oncall-engine/engine/apps/grafana_plugin/views/sync.py
Michael Derynck 6b40f95033 World, meet OnCall!
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>
2022-06-03 08:09:47 -06:00

73 lines
2.8 KiB
Python

import logging
from django.apps import apps
from django.conf import settings
from rest_framework import status
from rest_framework.request import Request
from rest_framework.response import Response
from rest_framework.views import APIView
from apps.grafana_plugin.permissions import PluginTokenVerified
from apps.grafana_plugin.tasks.sync import plugin_sync_organization_async
from apps.user_management.models import Organization
from common.api_helpers.mixins import GrafanaHeadersMixin
logger = logging.getLogger(__name__)
class PluginSyncView(GrafanaHeadersMixin, APIView):
permission_classes = (PluginTokenVerified,)
def post(self, request: Request) -> Response:
stack_id = self.instance_context["stack_id"]
org_id = self.instance_context["org_id"]
is_installed = False
try:
organization = Organization.objects.get(stack_id=stack_id, org_id=org_id)
if organization.api_token_status == Organization.API_TOKEN_STATUS_OK:
is_installed = True
organization.api_token_status = Organization.API_TOKEN_STATUS_PENDING
organization.save(update_fields=["api_token_status"])
plugin_sync_organization_async.apply_async((organization.pk,))
except Organization.DoesNotExist:
logger.info(f"Organization for stack {stack_id} org {org_id} was not found")
allow_signup = True
if not organization:
DynamicSetting = apps.get_model("base", "DynamicSetting")
allow_signup = DynamicSetting.objects.get_or_create(
name="allow_plugin_organization_signup", defaults={"boolean_value": True}
)[0].boolean_value
return Response(
status=status.HTTP_202_ACCEPTED,
data={
"is_installed": is_installed,
"is_user_anonymous": self.grafana_context["IsAnonymous"],
"allow_signup": allow_signup,
},
)
def get(self, request: Request) -> Response:
stack_id = self.instance_context["stack_id"]
org_id = self.instance_context["org_id"]
token_ok = False
try:
organization = Organization.objects.get(stack_id=stack_id, org_id=org_id)
if organization.api_token_status == Organization.API_TOKEN_STATUS_PENDING:
return Response(status=status.HTTP_202_ACCEPTED)
elif organization.api_token_status == Organization.API_TOKEN_STATUS_OK:
token_ok = True
except Organization.DoesNotExist:
logger.info(f"Organization for stack {stack_id} org {org_id} was not found")
return Response(
status=status.HTTP_200_OK,
data={
"token_ok": token_ok,
"license": settings.LICENSE,
"version": settings.VERSION,
},
)