Mirror
/
zulip
mirror of https://github.com/zulip/zulip.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

stripe: Fill `get_type_of_plan_tier_change` for remote realm/server.

This commit is contained in:
Aman Agrawal 2023-12-04 13:17:28 +00:00 committed by Tim Abbott
parent f22ccd3125
commit 9935f002ec
1 changed files with 50 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
corporate/lib/stripe.py
View File

@ -2656,8 +2656,25 @@ class RemoteRealmBillingSession(BillingSession): # nocoverage
def get_type_of_plan_tier_change( def get_type_of_plan_tier_change(
self, current_plan_tier: int, new_plan_tier: int self, current_plan_tier: int, new_plan_tier: int
) -> PlanTierChangeType: ) -> PlanTierChangeType:
# TBD valid_plan_tiers = [
return PlanTierChangeType.INVALID CustomerPlan.TIER_SELF_HOSTED_BUSINESS,
CustomerPlan.TIER_SELF_HOSTED_PLUS,
]
if (
current_plan_tier not in valid_plan_tiers
or new_plan_tier not in valid_plan_tiers
or current_plan_tier == new_plan_tier
):
return PlanTierChangeType.INVALID
if (
current_plan_tier == CustomerPlan.TIER_SELF_HOSTED_BUSINESS
and new_plan_tier == CustomerPlan.TIER_SELF_HOSTED_PLUS
):
return PlanTierChangeType.UPGRADE
else:
assert current_plan_tier == CustomerPlan.TIER_SELF_HOSTED_PLUS
assert new_plan_tier == CustomerPlan.TIER_SELF_HOSTED_BUSINESS
return PlanTierChangeType.DOWNGRADE
@override @override
def has_billing_access(self) -> bool: def has_billing_access(self) -> bool:
@ -2918,8 +2935,37 @@ class RemoteServerBillingSession(BillingSession): # nocoverage
def get_type_of_plan_tier_change( def get_type_of_plan_tier_change(
self, current_plan_tier: int, new_plan_tier: int self, current_plan_tier: int, new_plan_tier: int
) -> PlanTierChangeType: ) -> PlanTierChangeType:
# TBD valid_plan_tiers = [
return PlanTierChangeType.INVALID CustomerPlan.TIER_SELF_HOSTED_LEGACY,
CustomerPlan.TIER_SELF_HOSTED_BUSINESS,
CustomerPlan.TIER_SELF_HOSTED_PLUS,
]
if (
current_plan_tier not in valid_plan_tiers
or new_plan_tier not in valid_plan_tiers
or current_plan_tier == new_plan_tier
):
return PlanTierChangeType.INVALID
if current_plan_tier == CustomerPlan.TIER_SELF_HOSTED_LEGACY and new_plan_tier in (
CustomerPlan.TIER_SELF_HOSTED_BUSINESS,
CustomerPlan.TIER_SELF_HOSTED_PLUS,
):
return PlanTierChangeType.UPGRADE
elif (
current_plan_tier == CustomerPlan.TIER_SELF_HOSTED_BUSINESS
and new_plan_tier == CustomerPlan.TIER_SELF_HOSTED_PLUS
):
return PlanTierChangeType.UPGRADE
elif (
current_plan_tier == CustomerPlan.TIER_SELF_HOSTED_BUSINESS
and new_plan_tier == CustomerPlan.TIER_SELF_HOSTED_LEGACY
):
return PlanTierChangeType.DOWNGRADE
else:
assert current_plan_tier == CustomerPlan.TIER_SELF_HOSTED_PLUS
assert new_plan_tier == CustomerPlan.TIER_SELF_HOSTED_BUSINESS
return PlanTierChangeType.DOWNGRADE
@override @override
def has_billing_access(self) -> bool: def has_billing_access(self) -> bool: