corporate: Create VALID_BILLING_MODALITY_VALUES in views.py.

This commit is contained in:
Vishnu KS 2021-04-09 16:01:07 +05:30 committed by Tim Abbott
parent 6db6fab5ff
commit ef1b3a4067
1 changed files with 2 additions and 1 deletions

View File

@ -51,6 +51,7 @@ from zerver.models import UserProfile, get_realm
billing_logger = logging.getLogger("corporate.stripe") billing_logger = logging.getLogger("corporate.stripe")
VALID_BILLING_MODALITY_VALUES = ["send_invoice", "charge_automatically"]
VALID_LICENSE_MANAGEMENT_VALUES = ["automatic", "manual"] VALID_LICENSE_MANAGEMENT_VALUES = ["automatic", "manual"]
@ -69,7 +70,7 @@ def check_upgrade_parameters(
has_stripe_token: bool, has_stripe_token: bool,
seat_count: int, seat_count: int,
) -> None: ) -> None:
if billing_modality not in ["send_invoice", "charge_automatically"]: if billing_modality not in VALID_BILLING_MODALITY_VALUES:
raise BillingError("unknown billing_modality") raise BillingError("unknown billing_modality")
if schedule not in ["annual", "monthly"]: if schedule not in ["annual", "monthly"]:
raise BillingError("unknown schedule") raise BillingError("unknown schedule")