zulip/corporate/views/event_status.py

125 lines
3.8 KiB
Python

import logging
from typing import Optional
from django.http import HttpRequest, HttpResponse
from django.shortcuts import render
from corporate.lib.decorator import (
authenticated_remote_realm_management_endpoint,
authenticated_remote_server_management_endpoint,
self_hosting_management_endpoint,
)
from corporate.lib.stripe import (
EventStatusRequest,
RealmBillingSession,
RemoteRealmBillingSession,
RemoteServerBillingSession,
)
from zerver.decorator import require_organization_member, zulip_login_required
from zerver.lib.response import json_success
from zerver.lib.typed_endpoint import typed_endpoint
from zerver.models import UserProfile
billing_logger = logging.getLogger("corporate.stripe")
@require_organization_member
@typed_endpoint
def event_status(
request: HttpRequest,
user: UserProfile,
*,
stripe_session_id: Optional[str] = None,
stripe_payment_intent_id: Optional[str] = None,
) -> HttpResponse:
event_status_request = EventStatusRequest(
stripe_session_id=stripe_session_id, stripe_payment_intent_id=stripe_payment_intent_id
)
billing_session = RealmBillingSession(user)
data = billing_session.get_event_status(event_status_request)
return json_success(request, data)
@authenticated_remote_realm_management_endpoint
@typed_endpoint
def remote_realm_event_status(
request: HttpRequest,
billing_session: RemoteRealmBillingSession,
*,
stripe_session_id: Optional[str] = None,
stripe_payment_intent_id: Optional[str] = None,
) -> HttpResponse:
event_status_request = EventStatusRequest(
stripe_session_id=stripe_session_id, stripe_payment_intent_id=stripe_payment_intent_id
)
data = billing_session.get_event_status(event_status_request)
return json_success(request, data)
@authenticated_remote_server_management_endpoint
@typed_endpoint
def remote_server_event_status(
request: HttpRequest,
billing_session: RemoteServerBillingSession,
*,
stripe_session_id: Optional[str] = None,
stripe_payment_intent_id: Optional[str] = None,
) -> HttpResponse: # nocoverage
event_status_request = EventStatusRequest(
stripe_session_id=stripe_session_id, stripe_payment_intent_id=stripe_payment_intent_id
)
data = billing_session.get_event_status(event_status_request)
return json_success(request, data)
@zulip_login_required
@typed_endpoint
def event_status_page(
request: HttpRequest,
*,
stripe_session_id: str = "",
stripe_payment_intent_id: str = "",
) -> HttpResponse:
context = {
"stripe_session_id": stripe_session_id,
"stripe_payment_intent_id": stripe_payment_intent_id,
"billing_base_url": "",
}
return render(request, "corporate/event_status.html", context=context)
@self_hosting_management_endpoint
@typed_endpoint
def remote_realm_event_status_page(
request: HttpRequest,
*,
realm_uuid: str = "",
server_uuid: str = "",
stripe_session_id: str = "",
stripe_payment_intent_id: str = "",
) -> HttpResponse: # nocoverage
context = {
"stripe_session_id": stripe_session_id,
"stripe_payment_intent_id": stripe_payment_intent_id,
"billing_base_url": f"/realm/{realm_uuid}",
}
return render(request, "corporate/event_status.html", context=context)
@self_hosting_management_endpoint
@typed_endpoint
def remote_server_event_status_page(
request: HttpRequest,
*,
realm_uuid: str = "",
server_uuid: str = "",
stripe_session_id: str = "",
stripe_payment_intent_id: str = "",
) -> HttpResponse: # nocoverage
context = {
"stripe_session_id": stripe_session_id,
"stripe_payment_intent_id": stripe_payment_intent_id,
"billing_base_url": f"/server/{server_uuid}",
}
return render(request, "corporate/event_status.html", context=context)