oncall-engine/engine/apps/telegram/models/verification/channel.py

from typing import Optional, Tuple
from uuid import uuid4

from django.core.exceptions import ValidationError
from django.db import models
from django.utils import timezone

from apps.telegram.models import TelegramToOrganizationConnector
from apps.user_management.organization_log_creator import OrganizationLogType, create_organization_log


class TelegramChannelVerificationCode(models.Model):
    uuid = models.UUIDField(primary_key=True, default=uuid4, editable=False)
    datetime = models.DateTimeField(auto_now_add=True)

    organization = models.OneToOneField(
        "user_management.Organization", on_delete=models.CASCADE, related_name="telegram_verification_code"
    )

    author = models.ForeignKey("user_management.User", on_delete=models.CASCADE, null=True, default=None)

    @property
    def is_active(self) -> bool:
        return self.datetime + timezone.timedelta(days=1) < timezone.now()

    @classmethod
    def verify_channel_and_discussion_group(
        cls,
        uuid_code: str,
        channel_chat_id: int,
        channel_name: str,
        discussion_group_chat_id: int,
        discussion_group_name: str,
    ) -> Tuple[Optional[TelegramToOrganizationConnector], bool]:
        try:
            verification_code = cls.objects.get(uuid=uuid_code)

            # see if a organization has other channels connected
            # if it is the first channel, make it default for the organization
            connector_exists = verification_code.organization.telegram_channel.exists()

            connector, created = TelegramToOrganizationConnector.objects.get_or_create(
                channel_chat_id=channel_chat_id,
                defaults={
                    "organization": verification_code.organization,
                    "channel_name": channel_name,
                    "discussion_group_chat_id": discussion_group_chat_id,
                    "discussion_group_name": discussion_group_name,
                    "is_default_channel": not connector_exists,
                },
            )

            description = f"Telegram channel @{channel_name} was connected to organization"
            create_organization_log(
                verification_code.organization,
                verification_code.author,
                OrganizationLogType.TYPE_TELEGRAM_CHANNEL_CONNECTED,
                description,
            )

            if not connector_exists:
                description = f"The default channel for incidents in Telegram was changed to @{channel_name}"
                create_organization_log(
                    verification_code.organization,
                    verification_code.author,
                    OrganizationLogType.TYPE_TELEGRAM_DEFAULT_CHANNEL_CHANGED,
                    description,
                )

            return connector, created

        except (ValidationError, cls.DoesNotExist):
            return None, False