paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-04 05:53:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

stripe: Add 'is_sponsored' method to 'BillingSession' class.

Browse Source 
This prep commit adds a 'is_sponsored' method as we need
to explicitly check this in 'event_status' view.
This commit is contained in:
Prakhar Pratyush
2023-11-27 17:55:11 +05:30
committed by Tim Abbott
parent 29f77bfd31
commit 1b17626327
1 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
corporate/lib/stripe.py
View File

@@ -612,6 +612,10 @@ class BillingSession(ABC):
def approve_sponsorship(self) -> None:
pass
@abstractmethod
def is_sponsored(self) -> bool:
pass
@abstractmethod
def is_sponsored_or_pending(self, customer: Optional[Customer]) -> bool:
pass
@@ -1774,11 +1778,13 @@ class RealmBillingSession(BillingSession):
)
internal_send_private_message(notification_bot, user, message)
@override
def is_sponsored(self) -> bool:
return self.realm.plan_type == self.realm.PLAN_TYPE_STANDARD_FREE
@override
def is_sponsored_or_pending(self, customer: Optional[Customer]) -> bool:
if (
customer is not None and customer.sponsorship_pending
) or self.realm.plan_type == self.realm.PLAN_TYPE_STANDARD_FREE:
if (customer is not None and customer.sponsorship_pending) or self.is_sponsored():
return True
return False
@@ -1967,11 +1973,13 @@ class RemoteRealmBillingSession(BillingSession): # nocoverage
# TBD
pass
@override
def is_sponsored(self) -> bool:
return self.remote_realm.plan_type == self.remote_realm.PLAN_TYPE_COMMUNITY
@override
def is_sponsored_or_pending(self, customer: Optional[Customer]) -> bool:
if (
customer is not None and customer.sponsorship_pending
) or self.remote_realm.plan_type == self.remote_realm.PLAN_TYPE_COMMUNITY:
if (customer is not None and customer.sponsorship_pending) or self.is_sponsored():
return True
return False
@@ -2171,6 +2179,10 @@ class RemoteServerBillingSession(BillingSession): # nocoverage
plan.status = CustomerPlan.ENDED
plan.save(update_fields=["status"])
@override
def is_sponsored(self) -> bool:
return self.remote_server.plan_type == self.remote_server.PLAN_TYPE_COMMUNITY
@override
def is_sponsored_or_pending(self, customer: Optional[Customer]) -> bool:
# TBD