GlitchTip/apps/organizations_ext/email.py

from django.conf import settings

from glitchtip.email import DetailEmail

from .models import Organization, OrganizationUser


class MetQuotaEmail(DetailEmail):
    html_template_name = "organizations/met-quota-drip.html"
    text_template_name = "organizations/met-quota-drip.txt"
    subject_template_name = "organizations/met-quota-drip-subject.txt"
    model = Organization

    def get_email(self):
        return self.object.email

    def get_context_data(self, **kwargs):
        from djstripe.models import Product

        context = super().get_context_data(**kwargs)
        base_url = settings.GLITCHTIP_URL.geturl()
        organization = self.object
        subscription_link = f"{base_url}/{organization.slug}/settings/subscription"
        product = Product.objects.filter(
            plan__subscriptions__customer__subscriber=organization,
            plan__subscriptions__status="active",
        ).first()
        context.update(
            {
                "organization": organization,
                "product": product,
                "event_limit": product.metadata.get("events") if product else None,
                "subscription_link": subscription_link,
            }
        )
        return context


class InvitationEmail(DetailEmail):
    html_template_name = "organizations/invite-user-drip.html"
    text_template_name = "organizations/invite-user-drip.txt"
    subject_template_name = "organizations/invite-user-drip-subject.txt"
    model = OrganizationUser

    def get_email(self):
        return self.object.email

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        org_user = context["object"]
        context["token"] = self.kwargs["token"]
        context["user"] = org_user
        context["organization"] = org_user.organization
        return context