Improve support for fiscal years #140 (#149)

* Improve support for fiscal years #140 * Run pre-commit
Code4PuertoRico · Aug 13, 2020 · a3ee38a · a3ee38a
1 parent c447e5f
commit a3ee38a
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/contratospr/api/serializers.py b/contratospr/api/serializers.py
@@ -10,6 +10,12 @@
 )
 from ..contracts.utils import get_current_fiscal_year
 
+INITIAL_FISCAL_YEAR = 2016
+CURRENT_FISCAL_YEAR = get_current_fiscal_year()
+FISCAL_YEAR_CHOICES = [
+    (year, str(year)) for year in range(INITIAL_FISCAL_YEAR, CURRENT_FISCAL_YEAR)
+]
+
 
 class RecursiveSerializer(serializers.Serializer):
     def to_native(self, value):
@@ -18,9 +24,7 @@ def to_native(self, value):
 
 class HomeSerializer(serializers.Serializer):
     fiscal_year = serializers.ChoiceField(
-        choices=[(2016, "2016"), (2017, "2017"), (2018, "2018"), (2019, "2019")],
-        allow_null=False,
-        initial=get_current_fiscal_year() - 1,
+        choices=FISCAL_YEAR_CHOICES, allow_null=False, initial=CURRENT_FISCAL_YEAR - 1
     )