Limits for go-live form step 2

app/__init__.py

@@ -429,6 +429,14 @@ def format_thousands(value):
     return value
 
 
+def format_thousands_localized(value):
+    if isinstance(value, Number):
+        return "{:,}".format(int(value)).replace(",", "\u2009" if get_current_locale(current_app) == "fr" else ",")
+    if value is None:
+        return ""
+    return value
+
+
 def valid_phone_number(phone_number):
     try:
         validate_phone_number(phone_number)

app/main/forms.py

@@ -3,7 +3,7 @@
 from itertools import chain
 
 import pytz
-from flask import request
+from flask import current_app, request
 from flask_babel import lazy_gettext as _l
 from flask_wtf import FlaskForm as Form
 from flask_wtf.file import FileAllowed
@@ -42,7 +42,7 @@
 )
 from wtforms.widgets import CheckboxInput, ListWidget
 
-from app import format_thousands
+from app import format_thousands, format_thousands_localized
 from app.main.validators import (
     Blocklist,
     CsvFileValidator,
@@ -1800,47 +1800,191 @@ class GoLiveAboutServiceFormNoOrg(StripWhitespaceForm):
     )
 
 
+class OptionalIntegerRange:
+    def __init__(self, trigger_field, trigger_value, min=None, max=None, message=None):
+        self.trigger_field = trigger_field
+        self.trigger_value = trigger_value
+        self.min = min
+        self.max = max
+        self.message = message
+
+    def __call__(self, form, field):
+        trigger_data = getattr(form, self.trigger_field).data
+
+        # Only validate if the trigger condition is met
+        if trigger_data == self.trigger_value:
+            # First check if empty
+            if not field.data:
+                raise ValidationError(self.message or _l("This cannot be empty"))
+
+            # Then check range if value is provided
+            if self.min is not None and field.data < self.min:
+                raise ValidationError(_l("Number must be more than {min}").format(min=format_thousands_localized(self.min)))
+            if self.max is not None and field.data > self.max:
+                raise ValidationError(_l("Number must be less than {max}").format(max=format_thousands_localized(self.max)))
+
+
 class GoLiveAboutNotificationsForm(GoLiveAboutServiceForm):
-    notification_types = MultiCheckboxField(
-        _l("Specify the type of notifications you plan on sending."),
-        choices=[
-            ("email", _l("Email")),
-            ("sms", _l("Text message")),
-        ],
+    # todo: How can we program a pre-approved stretched daily limit? *10? *X? Another config constant?
+    def volume_choices(self, limit, notification_type):
+        return [
+            ("0", _l("None")),
+            (f"{limit}", _l("1 to {}").format(format_thousands_localized(limit))),
+            (
+                f"{limit*10}",
+                _l("{min} to {max}").format(
+                    min=format_thousands_localized(limit + 1), max=format_thousands_localized(limit * 10)
+                ),
+            ),
+            (f"more_{notification_type}", _l("More than {}").format(format_thousands_localized(limit * 10))),
+        ]
+
+    def volume_choices_restricted(self, limit):
+        return [
+            ("0", _l("None")),
+            (f"{limit}", _l("1 to {}").format(format_thousands_localized(limit))),
+            (f"{limit*10}", _l("More than {}").format(format_thousands_localized(limit * 10))),
+        ]
+
+    def more_validators(self, limit, notification_type):
+        return [
+            OptionalIntegerRange(
+                trigger_field=f"daily_{notification_type}_volume",
+                trigger_value=f"more_{notification_type}",
+                min=limit * 10,
+            )
+        ]
+
+    daily_email_volume = RadioField(
+        _l("How many emails do you expect to send on a busy day?"),
         validators=[DataRequired()],
     )
-    expected_volume = RadioField(
-        _l("How many notifications do you plan on sending per month?"),
-        choices=[
-            ("1-1k", _l("1 to 1,000 notifications")),
-            ("1k-10k", _l("1,000 to 10,000 notifications")),
-            ("10k-100k", _l("10,000 to 100,000 notifications")),
-            ("100k+", _l("More than 100,000 notifications")),
-        ],
+    annual_email_volume = RadioField(
+        _l("How many emails do you expect to  send in a year?"),
+        validators=[DataRequired()],
+    )
+    daily_sms_volume = RadioField(
+        _l("How many text messages do you expect to send on a busy day?"),
+        validators=[DataRequired()],
+    )
+    annual_sms_volume = RadioField(
+        _l("How many text messages do you expect to  send in a year?"),
         validators=[DataRequired()],
     )
 
+    # bug: how_many_more persists if user changes their input to one of the preset radios.
+    how_many_more_email = IntegerField(
+        label=_l("How many?"),
+        default="",
+    )
+    how_many_more_sms = IntegerField(
+        label=_l("How many?"),
+        default="",
+    )
+
+    # todo: update with new annual limits app config when available
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.daily_email_volume.choices = self.volume_choices(
+            limit=current_app.config["DEFAULT_LIVE_SERVICE_LIMIT"], notification_type="email"
+        )
+        self.annual_email_volume.choices = self.volume_choices_restricted(limit=current_app.config["DEFAULT_LIVE_SERVICE_LIMIT"])
+        self.daily_sms_volume.choices = self.volume_choices(
+            limit=current_app.config["DEFAULT_LIVE_SMS_DAILY_LIMIT"], notification_type="sms"
+        )
+        self.annual_sms_volume.choices = self.volume_choices_restricted(limit=current_app.config["DEFAULT_LIVE_SMS_DAILY_LIMIT"])
+        self.how_many_more_email.validators = self.more_validators(
+            limit=current_app.config["DEFAULT_LIVE_SERVICE_LIMIT"], notification_type="email"
+        )
+        self.how_many_more_sms.validators = self.more_validators(
+            limit=current_app.config["DEFAULT_LIVE_SMS_DAILY_LIMIT"], notification_type="sms"
+        )
+
+    @property
+    def volume_conditionals(self):
+        return {"more_email": self.how_many_more_email, "more_sms": self.how_many_more_sms}
+
 
 class GoLiveAboutNotificationsFormNoOrg(GoLiveAboutServiceFormNoOrg):
-    notification_types = MultiCheckboxField(
-        _l("Specify the type of notifications you plan on sending."),
-        choices=[
-            ("email", _l("Email")),
-            ("sms", _l("Text message")),
-        ],
+    # todo: How can we program a pre-approved stretched daily limit? *10? *X? Another config constant?
+    def volume_choices(self, limit, notification_type):
+        return [
+            ("0", _l("None")),
+            (f"{limit}", _l("1 to {}").format(format_thousands_localized(limit))),
+            (
+                f"{limit*10}",
+                _l("{min} to {max}").format(
+                    min=format_thousands_localized(limit + 1), max=format_thousands_localized(limit * 10)
+                ),
+            ),
+            (f"more_{notification_type}", _l("More than {}").format(format_thousands_localized(limit * 10))),
+        ]
+
+    def volume_choices_restricted(self, limit):
+        return [
+            ("0", _l("None")),
+            (f"{limit}", _l("1 to {}").format(format_thousands_localized(limit))),
+            (f"{limit*10}", _l("More than {}").format(format_thousands_localized(limit * 10))),
+        ]
+
+    def more_validators(self, limit, notification_type):
+        return [
+            OptionalIntegerRange(
+                trigger_field=f"daily_{notification_type}_volume",
+                trigger_value=f"more_{notification_type}",
+                min=limit * 10,
+            )
+        ]
+
+    daily_email_volume = RadioField(
+        _l("How many emails do you expect to send on a busy day?"),
         validators=[DataRequired()],
     )
-    expected_volume = RadioField(
-        _l("How many notifications do you plan on sending per month?"),
-        choices=[
-            ("1-1k", _l("1 to 1,000 notifications")),
-            ("1k-10k", _l("1,000 to 10,000 notifications")),
-            ("10k-100k", _l("10,000 to 100,000 notifications")),
-            ("100k+", _l("More than 100,000 notifications")),
-        ],
+    annual_email_volume = RadioField(
+        _l("How many emails do you expect to  send in a year?"),
+        validators=[DataRequired()],
+    )
+    daily_sms_volume = RadioField(
+        _l("How many text messages do you expect to send on a busy day?"),
+        validators=[DataRequired()],
+    )
+    annual_sms_volume = RadioField(
+        _l("How many text messages do you expect to  send in a year?"),
         validators=[DataRequired()],
     )
 
+    # bug: how_many_more persists if user changes their input to one of the preset radios.
+    how_many_more_email = IntegerField(
+        label=_l("How many?"),
+        default="",
+    )
+    how_many_more_sms = IntegerField(
+        label=_l("How many?"),
+        default="",
+    )
+
+    # todo: update with new annual limits app config when available
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.daily_email_volume.choices = self.volume_choices(
+            limit=current_app.config["DEFAULT_LIVE_SERVICE_LIMIT"], notification_type="email"
+        )
+        self.annual_email_volume.choices = self.volume_choices_restricted(limit=current_app.config["DEFAULT_LIVE_SERVICE_LIMIT"])
+        self.daily_sms_volume.choices = self.volume_choices(
+            limit=current_app.config["DEFAULT_LIVE_SMS_DAILY_LIMIT"], notification_type="sms"
+        )
+        self.annual_sms_volume.choices = self.volume_choices_restricted(limit=current_app.config["DEFAULT_LIVE_SMS_DAILY_LIMIT"])
+        self.how_many_more_email.validators = self.more_validators(
+            limit=current_app.config["DEFAULT_LIVE_SERVICE_LIMIT"], notification_type="email"
+        )
+        self.how_many_more_sms.validators = self.more_validators(
+            limit=current_app.config["DEFAULT_LIVE_SMS_DAILY_LIMIT"], notification_type="sms"
+        )
+
+    @property
+    def volume_conditionals(self):
+        return {"more_email": self.how_many_more_email, "more_sms": self.how_many_more_sms}
+
 
 class BrandingGOCForm(StripWhitespaceForm):
     """

app/templates/views/service-settings/use-case.html

@@ -2,7 +2,7 @@
 {% from "components/form.html" import form_wrapper %}
 {% from "components/page-header.html" import page_header %}
 {% from "components/page-footer.html" import page_footer %}
-{% from "components/radios.html" import radios %}
+{% from "components/select-input.html" import select %}
 {% from "components/textbox.html" import textbox %}
 {% from "components/checkbox.html" import checkboxes %}
 
@@ -38,8 +38,14 @@
         <p>
           {{ _("This information helps us set sending limits and budget accordingly.") }}
         </p>
-        {{ checkboxes(form.notification_types) }}
-        {{ radios(form.expected_volume) }}
+        <h2 class="heading-medium">{{ _("About emails you want to send") }}</h2>
+        {{ select(form.daily_email_volume, option_conditionals=form.volume_conditionals) }}
+        {{ select(form.annual_email_volume, option_conditionals=form.volume_conditionals) }}
+
+        <h2 class="heading-medium">{{ _("About text messages you want to send") }}</h2>
+        {{ select(form.daily_sms_volume, option_conditionals=form.volume_conditionals) }}
+        {{ select(form.annual_sms_volume, option_conditionals=form.volume_conditionals) }}
+
 
         <!-- Coming from the previous step. If it's not included, checkboxes will be reset -->
         <div class="hidden">

app/translations/csv/fr.csv

@@ -1455,6 +1455,19 @@
 "1,000 to 10,000 notifications","1 000 à 10 000 notifications"
 "10,000 to 100,000 notifications","10 000 à 100 000 notifications"
 "More than 100,000 notifications","Plus de 100 000 notifications"
+"None","Aucun"
+"1 to {}","1 à {}"
+"{min} to {max}","{min} à {max}"
+"More than {}","Plus de {}"
+"How many?","Combien?"
+"About emails you want to send","Volume de courriels prévu"
+"How many emails do you expect to send on a busy day?","Combien de courriels enverrez-vous lors d’une journée occupée?"
+"How many emails do you expect to  send in a year?","Combien de courriels enverrez-vous par année?"
+"About text messages you want to send","Volume de messages texte prévu"
+"How many text messages do you expect to send on a busy day?","Combien de messages texte enverrez-vous lors d’une journée occupée?"
+"How many text messages do you expect to  send in a year?","Combien de messages texte enverrez-vous par année?"
+"Number must be more than {min}","Le nombre doit être supérieur à {min}"
+"Number must be less than {max}","Le nombre doit être inférieur à {max}"
 "Specify the type of notifications you plan on sending.","Précisez le type de notification que vous prévoyez envoyer."
 "How many notifications do you plan on sending per month?","Combien de notifications prévoyez-vous envoyer par mois?"
 "Once you have completed all the steps, submit your request to the GC Notify team. We’ll be in touch within 2 business days.","Après avoir complété toutes les étapes, soumettez votre demande à l’équipe Notification GC. Nous communiquerons avec vous dans un délai de 2 jours ouvrables."