Update form fields and responses display for weight feature

Author: miriam-z

lung_cancer_screening/core/form_fields.py

@@ -441,3 +441,83 @@ def compress(self, data_list):
             total_inches = feet * 12 + inches
             return int(total_inches)
         return None
+
+
+class ImperialWeightWidget(widgets.MultiWidget):
+    """
+    A widget that splits weight into stone and pounds inputs.
+    """
+
+    def __init__(self, attrs=None):
+        weight_widgets = (
+            widgets.NumberInput(attrs=attrs),
+            widgets.NumberInput(attrs=attrs),
+        )
+        super().__init__(weight_widgets, attrs)
+
+    def decompress(self, value):
+        """
+        Convert total pounds back to stone and pounds for display.
+        """
+        if value:
+            stone = int(value // 14)
+            pounds = value % 14
+            return [stone, pounds]
+        return [None, None]
+
+    def subwidgets(self, name, value, attrs=None):
+        """
+        Expose data for each subwidget, so that we can render them separately in the template.
+        """
+        context = self.get_context(name, value, attrs)
+        for subwidget in context["widget"]["subwidgets"]:
+            yield subwidget
+
+
+class ImperialWeightField(forms.MultiValueField):
+    """
+    A field that combines stone and pounds into a single weight value in total pounds.
+    """
+
+    widget = ImperialWeightWidget
+
+    def __init__(self, *args, **kwargs):
+        error_messages = kwargs.get("error_messages", {})
+
+        stone_kwargs = {
+            "min_value": 0,
+            "max_value": 50,
+            "error_messages": {
+                'invalid': 'Stone must be in whole numbers.',
+                'min_value': 'Weight must be between 4 stone and 50 stone.',
+                'max_value': 'Weight must be between 4 stone and 50 stone.',
+                **error_messages,
+            },
+        }
+        pounds_kwargs = {
+            "min_value": 0,
+            "max_value": 13,
+            "error_messages": {
+                'invalid': 'Pounds must be in whole numbers.',
+                'min_value': 'Pounds must be between 0 and 13.',
+                'max_value': 'Pounds must be between 0 and 13.',
+                **error_messages,
+            },
+        }
+        fields = (
+            IntegerField(**stone_kwargs),
+            IntegerField(**pounds_kwargs),
+        )
+        kwargs["template_name"] = "forms/imperial-weight-input.jinja"
+
+        super().__init__(fields, *args, **kwargs)
+
+    def compress(self, data_list):
+        """
+        Convert stone and pounds to total pounds.
+        """
+        if data_list and all(item is not None for item in data_list):
+            stone, pounds = data_list
+            total_pounds = stone * 14 + pounds
+            return int(total_pounds)
+        return None

lung_cancer_screening/core/tests/acceptance/test_cannot_change_answers_after_submission.py

@@ -6,12 +6,12 @@
 
 from .helpers.user_interaction_helpers import (
     fill_in_and_submit_height_metric,
+    fill_in_and_submit_weight_metric,
     fill_in_and_submit_participant_id,
     fill_in_and_submit_smoking_eligibility,
     fill_in_and_submit_date_of_birth
 )
 
-
 class TestQuestionnaire(StaticLiveServerTestCase):
 
     @classmethod
@@ -39,6 +39,7 @@ def test_cannot_change_responses_once_checked_and_submitted(self):
         fill_in_and_submit_smoking_eligibility(page, smoking_status)
         fill_in_and_submit_date_of_birth(page, age)
         fill_in_and_submit_height_metric(page, "170")
+        fill_in_and_submit_weight_metric(page, "25.4")
 
         page.click("text=Submit")
 

lung_cancer_screening/questions/jinja2/responses.jinja

@@ -19,6 +19,7 @@
         <li>Have you ever smoked? {{ response_set.get_have_you_ever_smoked_display() }}</li>
         <li>What is your date of birth? {{ response_set.date_of_birth }}</li>
         <li>What is your height? {{ response_set.formatted_height }}</li>
+        <li>What is your weight? {{ response_set.formatted_weight }}</li>
       </ul>
 
       <form action="{{ request.path }}" method="post">