models: Refactor corporate/models.py into models package.

Fixes #34318.

Seperated models file into a package with component files.
This commit is contained in:
Ethan Mayer
2025-04-07 15:51:39 -05:00
committed by Tim Abbott
parent a8c513a95e
commit c12b94aea4
33 changed files with 690 additions and 666 deletions

View File

@@ -283,7 +283,7 @@ def get_realm_authentication_methods_for_page_params_api(
# The rest of the function is only for the mechanism of restricting
# certain backends based on the realm's plan type on Zulip Cloud.
from corporate.models import CustomerPlan
from corporate.models.plans import CustomerPlan
for backend_name, backend_result in result_dict.items():
available_for = AUTH_BACKEND_NAME_MAP[backend_name].available_for_cloud_plans

View File

@@ -136,7 +136,7 @@ def has_pending_sponsorship_request(
user_has_billing_access = user_profile is not None and user_profile.has_billing_access
if settings.CORPORATE_ENABLED and user_profile is not None and user_has_billing_access:
from corporate.models import get_customer_by_realm
from corporate.models.customers import get_customer_by_realm
customer = get_customer_by_realm(user_profile.realm)
if customer is not None:

View File

@@ -39,7 +39,9 @@ from requests import PreparedRequest
from two_factor.plugins.phonenumber.models import PhoneDevice
from typing_extensions import override
from corporate.models import Customer, CustomerPlan, LicenseLedger
from corporate.models.customers import Customer
from corporate.models.licenses import LicenseLedger
from corporate.models.plans import CustomerPlan
from zerver.actions.message_send import check_send_message, check_send_stream_message
from zerver.actions.realm_settings import do_change_realm_permission_group_setting
from zerver.actions.streams import bulk_add_subscriptions, bulk_remove_subscriptions

View File

@@ -37,7 +37,8 @@ realms used for testing; consider using deactivate_realm instead."""
# Deleting a Realm object also deletes associating billing
# metadata in an invariant-violating way, so we should
# never use this tool for a realm with billing set up.
from corporate.models import CustomerPlan, get_customer_by_realm
from corporate.models.customers import get_customer_by_realm
from corporate.models.plans import CustomerPlan
customer = get_customer_by_realm(realm)
if customer and (

View File

@@ -10,7 +10,8 @@ from django.conf import settings
from django.test import override_settings
from django.utils.timezone import now as timezone_now
from corporate.models import Customer, CustomerPlan
from corporate.models.customers import Customer
from corporate.models.plans import CustomerPlan
from zerver.context_processors import get_apps_page_url
from zerver.lib.integrations import CATEGORIES, INTEGRATIONS, META_CATEGORY
from zerver.lib.test_classes import ZulipTestCase

View File

@@ -10,7 +10,8 @@ from django.conf import settings
from django.test import override_settings
from django.utils.timezone import now as timezone_now
from corporate.models import Customer, CustomerPlan
from corporate.models.customers import Customer
from corporate.models.plans import CustomerPlan
from version import ZULIP_VERSION
from zerver.actions.create_user import do_create_user
from zerver.actions.realm_settings import do_change_realm_plan_type, do_set_realm_property

View File

@@ -29,7 +29,7 @@ from typing_extensions import override
from analytics.lib.counts import CountStat, LoggingCountStat
from analytics.models import InstallationCount, RealmCount, UserCount
from corporate.lib.stripe import RemoteRealmBillingSession
from corporate.models import CustomerPlan
from corporate.models.plans import CustomerPlan
from version import ZULIP_VERSION
from zerver.actions.create_realm import do_create_realm
from zerver.actions.message_delete import do_delete_messages