Merge pull request #906 from uktrade/strip-territory-codes-for-hmrc

Strip territory code for HMRC.
uktrade · Jan 25, 2022 · 4bf08b9 · 4bf08b9
2 parents 1a95567 + 873e1b3
commit 4bf08b9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/api/licences/serializers/hmrc_integration.py b/api/licences/serializers/hmrc_integration.py
@@ -9,6 +9,16 @@ class HMRCIntegrationCountrySerializer(serializers.Serializer):
     id = serializers.CharField()
     name = serializers.CharField()
 
+    def strip_territory_code(self, country_code):
+        """Does nothing for "GB" but transforms "AE-DU" to "AE".
+        """
+        return country_code.split("-")[0]
+
+    def to_representation(self, data):
+        data = super().to_representation(data)
+        data["id"] = self.strip_territory_code(data["id"])
+        return data
+
 
 class HMRCIntegrationAddressSerializer(serializers.Serializer):
     line_1 = serializers.SerializerMethodField()