add fiscal year versioning

Author: damonmcc

dcpy/test/utils/test_versions.py

@@ -24,6 +24,13 @@ def test_parsing_valid_versions(self):
                     format=versions.DateVersionFormat.quarter,
                 ),
             ],
+            [
+                "FY2020",
+                versions.Date(
+                    date=date(2020, 1, 1),
+                    format=versions.DateVersionFormat.fiscal_year,
+                ),
+            ],
             [
                 "26prelim",
                 versions.CapitalBudget(
@@ -62,6 +69,8 @@ def test_parsing_invalid_version(self):
             versions.parse("23v")
         with self.assertRaises(Exception):
             versions.parse("2v12")
+        with self.assertRaises(Exception):
+            versions.parse("FY26")
         with self.assertRaises(Exception):
             versions.parse("20231212")
         with self.assertRaises(Exception):
@@ -158,6 +167,7 @@ def test_sort_invalid_versions(self):
     def test_is_newer_valid_versions(self):
         for version_1, version_2, bool_expected in [
             ["23v2", "22v3.4", True],
+            ["FY2019", "FY2020", False],
             ["23Q1", "23Q2", False],
             ["23v2.0.1", "23v2", True],
             ["23Q1.1", "23Q1", True],
@@ -182,6 +192,7 @@ def test_bumping_versions(self):
             ["major", None, "23v2.1", "23v3"],
             ["minor", None, "23v2", "23v2.1"],
             ["minor", None, "23v2.1", "23v2.2"],
+            [None, 1, "FY2020", "FY2021"],
             [None, 1, "23Q1", "23Q2"],
             [None, 2, "23Q4", "24Q2"],
             [None, 7, "23Q2", "25Q1"],

dcpy/utils/versions.py

@@ -10,6 +10,7 @@
 
 
 class DateVersionFormat(StrEnum):
+    fiscal_year = "Fiscal Year"
     quarter = "Quarter"
     month = "Month"
     date = "Date"
@@ -159,6 +160,13 @@ class Date(Version):
     @property
     def label(self) -> str:
         match self.format:
+            case DateVersionFormat.fiscal_year:
+                if self.patch == 0:
+                    return (
+                        f"FY{self.date.strftime('%Y')}"
+                    )
+                else:
+                    return f"FY{self.date.strftime('%Y')}.{self.patch}"
             case DateVersionFormat.quarter:
                 if self.patch == 0:
                     return (
@@ -176,6 +184,8 @@ def label(self) -> str:
                     return self.date.strftime("%Y-%m-%d")
                 else:
                     return f"{self.date.strftime('%Y-%m-%d')}.{self.patch}"
+            case _:
+                raise ValueError(f"Unsupported DateVersionFormat '{self.format}'")
 
     def __lt__(self, other) -> bool:
         match other:
@@ -253,6 +263,11 @@ def parse(v: str) -> Version:
             return MajorMinor(
                 year=int(m[1]), major=int(m[2]), minor=int(m[3]), patch=int(m[4])
             )
+        case r"^FY(\d{4})$" as m:
+            return Date(
+                date(int(m[1]), 1, 1),
+                format=DateVersionFormat.fiscal_year,
+            )
         case r"^(\d{2})Q(\d)$" as m:
             return Date(
                 date(2000 + int(m[1]), int(m[2]) * 3 - 2, 1),
@@ -403,6 +418,11 @@ def bump(
                 minor=previous_version.minor,
                 patch=previous_version.patch + bump_by,
             )
+        case Date(format=DateVersionFormat.fiscal_year), None:
+            return Date(
+                date=previous_version.date + relativedelta(years=bump_by),
+                format=previous_version.format,
+            )
         case Date(format=DateVersionFormat.quarter), None:
             return Date(
                 date=previous_version.date + relativedelta(months=bump_by * 3),