feat: add season_start/season_end to JobRequest with wrap support

Generated with [Claude Code](https://claude.ai/code)
via [Happy](https://happy.engineering)

Co-Authored-By: Claude <noreply@anthropic.com>
Co-Authored-By: Happy <yesreply@happy.engineering>

app/models.py

@@ -93,6 +93,19 @@ class JobRequest(BaseModel):
     time_range: TimeRange = Field(default_factory=TimeRange)
     indicator_ids: list[str]
     email: str
+    season_start: int = Field(default=1, ge=1, le=12)
+    season_end: int = Field(default=12, ge=1, le=12)
+
+    def season_months(self) -> list[int]:
+        """Return ordered list of month numbers in the analysis season.
+
+        Supports year-boundary wrapping: season_start=10, season_end=3
+        yields [10, 11, 12, 1, 2, 3].
+        """
+        if self.season_start <= self.season_end:
+            return list(range(self.season_start, self.season_end + 1))
+        else:
+            return list(range(self.season_start, 13)) + list(range(1, self.season_end + 1))
 
     @field_validator("indicator_ids")
     @classmethod

tests/test_models.py

@@ -86,3 +86,66 @@ def test_indicator_result_fields():
     )
     assert result.status == "amber"
     assert result.trend == "deteriorating"
+
+
+def test_job_request_season_defaults():
+    """Default season is full year (1-12)."""
+    req = JobRequest(
+        aoi=AOI(name="Test", bbox=[36.75, -1.35, 36.95, -1.20]),
+        indicator_ids=["fires"],
+        email="t@t.com",
+    )
+    assert req.season_start == 1
+    assert req.season_end == 12
+
+
+def test_job_request_season_months_normal():
+    """Non-wrapping season: Apr-Sep."""
+    req = JobRequest(
+        aoi=AOI(name="Test", bbox=[36.75, -1.35, 36.95, -1.20]),
+        indicator_ids=["fires"],
+        email="t@t.com",
+        season_start=4,
+        season_end=9,
+    )
+    assert req.season_months() == [4, 5, 6, 7, 8, 9]
+
+
+def test_job_request_season_months_wrapping():
+    """Wrapping season: Oct-Mar (Southern Hemisphere)."""
+    req = JobRequest(
+        aoi=AOI(name="Test", bbox=[36.75, -1.35, 36.95, -1.20]),
+        indicator_ids=["fires"],
+        email="t@t.com",
+        season_start=10,
+        season_end=3,
+    )
+    assert req.season_months() == [10, 11, 12, 1, 2, 3]
+
+
+def test_job_request_season_months_full_year():
+    """Full year default."""
+    req = JobRequest(
+        aoi=AOI(name="Test", bbox=[36.75, -1.35, 36.95, -1.20]),
+        indicator_ids=["fires"],
+        email="t@t.com",
+    )
+    assert req.season_months() == [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
+
+
+def test_job_request_season_validation():
+    """Season months must be 1-12."""
+    with pytest.raises(Exception):
+        JobRequest(
+            aoi=AOI(name="Test", bbox=[36.75, -1.35, 36.95, -1.20]),
+            indicator_ids=["fires"],
+            email="t@t.com",
+            season_start=0,
+        )
+    with pytest.raises(Exception):
+        JobRequest(
+            aoi=AOI(name="Test", bbox=[36.75, -1.35, 36.95, -1.20]),
+            indicator_ids=["fires"],
+            email="t@t.com",
+            season_end=13,
+        )