mirror of
https://github.com/zulip/zulip.git
synced 2026-07-15 21:03:26 +08:00
b167eeff08
Moves and generalizes `switch_realm_from_standard_to_plus_plan` in stripe.py to be a more general function for changing a CustomerPlan to a new and valid tier, `do_change_plan_to_new_tier`. Adds a helper function with the previous function name to be used for the support view and management command for changing a realm from the Standard plan tier to the Plus plan tier.
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
from decimal import Decimal
|
|
from typing import Optional
|
|
from urllib.parse import urlencode, urljoin, urlunsplit
|
|
|
|
from django.conf import settings
|
|
from django.urls import reverse
|
|
|
|
from corporate.lib.stripe import RealmBillingSession
|
|
from corporate.models import CustomerPlan, get_customer_by_realm
|
|
from zerver.models import Realm, UserProfile, get_realm
|
|
|
|
|
|
def get_support_url(realm: Realm) -> str:
|
|
support_realm_uri = get_realm(settings.STAFF_SUBDOMAIN).uri
|
|
support_url = urljoin(
|
|
support_realm_uri,
|
|
urlunsplit(("", "", reverse("support"), urlencode({"q": realm.string_id}), "")),
|
|
)
|
|
return support_url
|
|
|
|
|
|
def get_discount_for_realm(realm: Realm) -> Optional[Decimal]:
|
|
customer = get_customer_by_realm(realm)
|
|
if customer is not None:
|
|
return customer.default_discount
|
|
return None
|
|
|
|
|
|
def attach_discount_to_realm(realm: Realm, discount: Decimal, *, acting_user: UserProfile) -> None:
|
|
billing_session = RealmBillingSession(acting_user, realm, support_session=True)
|
|
billing_session.attach_discount_to_customer(discount)
|
|
|
|
|
|
def approve_realm_sponsorship(realm: Realm, *, acting_user: UserProfile) -> None:
|
|
billing_session = RealmBillingSession(acting_user, realm, support_session=True)
|
|
billing_session.approve_sponsorship()
|
|
|
|
|
|
def update_realm_sponsorship_status(
|
|
realm: Realm, sponsorship_pending: bool, *, acting_user: UserProfile
|
|
) -> None:
|
|
billing_session = RealmBillingSession(acting_user, realm, support_session=True)
|
|
billing_session.update_customer_sponsorship_status(sponsorship_pending)
|
|
|
|
|
|
def update_realm_billing_method(
|
|
realm: Realm, charge_automatically: bool, *, acting_user: UserProfile
|
|
) -> None:
|
|
billing_session = RealmBillingSession(acting_user, realm, support_session=True)
|
|
billing_session.update_billing_method_of_current_plan(charge_automatically)
|
|
|
|
|
|
def switch_realm_from_standard_to_plus_plan(realm: Realm) -> None:
|
|
billing_session = RealmBillingSession(realm=realm)
|
|
billing_session.do_change_plan_to_new_tier(new_plan_tier=CustomerPlan.PLUS)
|