paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-10-24 08:33:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

populate_billing_realms: Call sessions with billing_user.

Browse Source
This commit is contained in:
Aman Agrawal
2023-12-13 01:02:14 +00:00
committed by Tim Abbott
parent da74b9ade7
commit e9ce69aaf6
1 changed files with 21 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
zilencer/management/commands/populate_billing_realms.py
View File

@@ -27,7 +27,12 @@ from zerver.apps import flush_cache
from zerver.lib.remote_server import get_realms_info_for_push_bouncer from zerver.lib.remote_server import get_realms_info_for_push_bouncer
from zerver.lib.streams import create_stream_if_needed from zerver.lib.streams import create_stream_if_needed
from zerver.models import Realm, UserProfile, get_realm from zerver.models import Realm, UserProfile, get_realm
from zilencer.models import RemoteRealm, RemoteZulipServer from zilencer.models import (
RemoteRealm,
RemoteRealmBillingUser,
RemoteServerBillingUser,
RemoteZulipServer,
)
from zilencer.views import update_remote_realm_data_for_server from zilencer.views import update_remote_realm_data_for_server
from zproject.config import get_secret from zproject.config import get_secret
@@ -414,10 +419,14 @@ def populate_remote_server(customer_profile: CustomerProfile) -> Dict[str, str]:
last_audit_log_update=timezone_now(), last_audit_log_update=timezone_now(),
) )
billing_session = RemoteServerBillingSession(remote_server) billing_user = RemoteServerBillingUser.objects.create(
full_name="Server user",
remote_server=remote_server,
email=f"{unique_id}@example.com",
)
billing_session = RemoteServerBillingSession(remote_server, billing_user)
if customer_profile.tier == CustomerPlan.TIER_SELF_HOSTED_LEGACY: if customer_profile.tier == CustomerPlan.TIER_SELF_HOSTED_LEGACY:
# Create customer plan for these servers for temporary period. # Create customer plan for these servers for temporary period.
billing_session = RemoteServerBillingSession(remote_server)
renewal_date = datetime.strptime(customer_profile.renewal_date, TIMESTAMP_FORMAT).replace( renewal_date = datetime.strptime(customer_profile.renewal_date, TIMESTAMP_FORMAT).replace(
tzinfo=timezone.utc tzinfo=timezone.utc
) )
@@ -484,7 +493,15 @@ def populate_remote_realms(customer_profile: CustomerProfile) -> Dict[str, str]:
) )
remote_realm = RemoteRealm.objects.get(uuid=local_realm.uuid) remote_realm = RemoteRealm.objects.get(uuid=local_realm.uuid)
billing_session = RemoteRealmBillingSession(remote_realm) user = UserProfile.objects.filter(realm=local_realm).first()
assert user is not None
billing_user = RemoteRealmBillingUser.objects.create(
full_name=user.full_name,
remote_realm=remote_realm,
user_uuid=user.uuid,
email=user.email,
)
billing_session = RemoteRealmBillingSession(remote_realm, billing_user)
# TODO: Save property audit log data for server. # TODO: Save property audit log data for server.
remote_realm.server.last_audit_log_update = timezone_now() remote_realm.server.last_audit_log_update = timezone_now()
remote_realm.server.save(update_fields=["last_audit_log_update"]) remote_realm.server.save(update_fields=["last_audit_log_update"])