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

stripe: Raise 'MissingDataError' while fetching license count.

Browse Source 
If the RemoteRealmAuditLog has stale data, it means the server
stopped or never uploaded data. We raise MissingDataError in such
cases when a user action led to calculating licenses count from
stale data.
This commit is contained in:
Prakhar Pratyush
2023-12-06 23:55:49 +05:30
committed by Tim Abbott
parent 40621478cb
commit ed9b0d330d
6 changed files with 144 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
corporate/lib/stripe.py
View File

@@ -56,6 +56,7 @@ from zerver.models import (
get_realm,
get_system_bot,
)
from zilencer.lib.remote_counts import MissingDataError
from zilencer.models import (
RemoteRealm,
RemoteRealmAuditLog,
@@ -64,6 +65,7 @@ from zilencer.models import (
RemoteZulipServerAuditLog,
get_remote_realm_guest_and_non_guest_count,
get_remote_server_guest_and_non_guest_count,
has_stale_audit_log,
)
from zproject.config import get_secret
@@ -2802,6 +2804,8 @@ class RemoteRealmBillingSession(BillingSession): # nocoverage
@override
def current_count_for_billed_licenses(self) -> int:
if has_stale_audit_log(self.remote_realm.server):
raise MissingDataError
remote_realm_counts = get_remote_realm_guest_and_non_guest_count(self.remote_realm)
return remote_realm_counts.non_guest_user_count + remote_realm_counts.guest_user_count
@@ -3115,6 +3119,8 @@ class RemoteServerBillingSession(BillingSession): # nocoverage
@override
def current_count_for_billed_licenses(self) -> int:
if has_stale_audit_log(self.remote_server):
raise MissingDataError
remote_server_counts = get_remote_server_guest_and_non_guest_count(self.remote_server.id)
return remote_server_counts.non_guest_user_count + remote_server_counts.guest_user_count