Fix holiday calculation URL and add ArgentinaDatos provider

Author: PPeitsch

app/config/config.py

@@ -12,7 +12,11 @@ class Config:
     HOLIDAY_PROVIDER = os.getenv("HOLIDAY_PROVIDER", "ARGENTINA_WEBSITE")
     HOLIDAYS_BASE_URL = os.getenv(
         "HOLIDAYS_BASE_URL",
-        "https://www.argentina.gob.ar/interior/feriados-nacionales-{year}",
+        "https://www.argentina.gob.ar/jefatura/feriados-nacionales-{year}",
+    )
+    HOLIDAY_API_URL = os.getenv(
+        "HOLIDAY_API_URL",
+        "https://api.argentinadatos.com/v1/feriados/{year}",
     )
 
     # Configuración horaria

app/services/holiday_providers/argentina_api_provider.py

@@ -0,0 +1,64 @@
+from datetime import datetime
+from typing import List
+
+import requests
+
+from app.models.models import Holiday
+
+
+class ArgentinaApiProvider:
+    """
+    Holiday provider that fetches holidays from the ArgentinaDatos API.
+    """
+
+    def __init__(self, api_url: str):
+        self.url_template = api_url
+
+    def get_holidays(self, year: int) -> List[Holiday]:
+        """
+        Fetches holidays for a given year from the API and returns them as Holiday objects.
+        """
+        url = self.url_template.format(year=year)
+        try:
+            response = requests.get(url, timeout=10)
+            response.raise_for_status()
+            data = response.json()
+        except requests.RequestException as e:
+            print(f"Error fetching holiday data from API for year {year}: {e}")
+            return []
+        except ValueError as e:
+            print(f"Error decoding JSON from API for year {year}: {e}")
+            return []
+
+        holidays: List[Holiday] = []
+        for entry in data:
+            try:
+                date_str = entry.get("fecha")
+                description = entry.get("nombre")
+                type_raw = entry.get("tipo")
+
+                if not date_str or not description:
+                    continue
+
+                parsed_date = datetime.strptime(date_str, "%Y-%m-%d").date()
+
+                # Filter by year just in case
+                if parsed_date.year != year:
+                    continue
+
+                type_map = {
+                    "inamovible": "Inamovible",
+                    "trasladable": "Trasladable",
+                    "puente": "Fines Turísticos",
+                    "nolaborable": "No Laborable",
+                }
+                type_ = type_map.get(type_raw, "Otro")
+
+                holiday = Holiday(date=parsed_date, description=description, type=type_)
+                holidays.append(holiday)
+
+            except (ValueError, AttributeError) as e:
+                print(f"Error parsing holiday entry: {entry}. Error: {e}")
+                continue
+
+        return holidays

app/services/holiday_service.py

@@ -1,13 +1,17 @@
 from flask import Config
 
+from app.services.holiday_providers.argentina_api_provider import ArgentinaApiProvider
 from app.services.holiday_providers.argentina_website_provider import (
     ArgentinaWebsiteProvider,
 )
 from app.services.holiday_providers.base import HolidayProvider
 
 # A mapping of provider names to their corresponding classes.
 # This makes it easy to add new providers in the future.
-PROVIDER_MAP = {"ARGENTINA_WEBSITE": ArgentinaWebsiteProvider}
+PROVIDER_MAP = {
+    "ARGENTINA_WEBSITE": ArgentinaWebsiteProvider,
+    "ARGENTINA_API": ArgentinaApiProvider,
+}
 
 
 def get_holiday_provider(config: Config) -> HolidayProvider:
@@ -27,7 +31,13 @@ def get_holiday_provider(config: Config) -> HolidayProvider:
         base_url = getattr(config, "HOLIDAYS_BASE_URL", None)
         if not base_url:
             raise ValueError("HOLIDAYS_BASE_URL is not configured.")
-        return provider_class(base_url=base_url)
+        return ArgentinaWebsiteProvider(base_url=base_url)
+
+    if provider_name.upper() == "ARGENTINA_API":
+        api_url = getattr(config, "HOLIDAY_API_URL", None)
+        if not api_url:
+            raise ValueError("HOLIDAY_API_URL is not configured.")
+        return ArgentinaApiProvider(api_url=api_url)
 
     # This part would be extended for other providers
     # For now, we raise an error if the provider is in the map but has no

tests/test_argentina_api_provider.py

@@ -0,0 +1,92 @@
+from unittest.mock import MagicMock, patch
+
+import pytest
+import requests
+
+from app.models.models import Holiday
+from app.services.holiday_providers.argentina_api_provider import ArgentinaApiProvider
+
+
+class TestArgentinaApiProvider:
+    """
+    Tests for the ArgentinaApiProvider.
+    """
+
+    @pytest.fixture
+    def mock_response(self):
+        """Creates a mock response object."""
+        mock = MagicMock(spec=requests.Response)
+        mock.raise_for_status.return_value = None
+        return mock
+
+    def test_get_holidays_success(self, mock_response):
+        """
+        Test successful holiday parsing from API JSON response.
+        """
+        year = 2025
+        mock_response.json.return_value = [
+            {"fecha": "2025-01-01", "nombre": "Año Nuevo", "tipo": "inamovible"},
+            {
+                "fecha": "2025-05-01",
+                "nombre": "Día del Trabajador",
+                "tipo": "inamovible",
+            },
+        ]
+        provider = ArgentinaApiProvider(api_url="http://fake-api.com/{year}")
+        with patch("requests.get", return_value=mock_response):
+            holidays = provider.get_holidays(year)
+            assert len(holidays) == 2
+            assert holidays[0].description == "Año Nuevo"
+            assert holidays[0].date.year == 2025
+
+    def test_get_holidays_filter_by_year(self, mock_response):
+        """
+        Test that holidays from other years are filtered out.
+        """
+        year = 2025
+        mock_response.json.return_value = [
+            {"fecha": "2025-01-01", "nombre": "Correct Year", "tipo": "inamovible"},
+            {"fecha": "2024-12-31", "nombre": "Wrong Year", "tipo": "inamovible"},
+        ]
+        provider = ArgentinaApiProvider(api_url="http://fake-api.com/{year}")
+        with patch("requests.get", return_value=mock_response):
+            holidays = provider.get_holidays(year)
+            assert len(holidays) == 1
+            assert holidays[0].description == "Correct Year"
+
+    def test_get_holidays_malformed_entry(self, mock_response):
+        """
+        Test that malformed entries are skipped.
+        """
+        year = 2025
+        mock_response.json.return_value = [
+            {"fecha": "2025-01-01", "nombre": "Good", "tipo": "inamovible"},
+            {"fecha": "", "nombre": "Bad Date", "tipo": "inamovible"},
+            {"fecha": "2025-01-02", "nombre": "", "tipo": "inamovible"},
+        ]
+        provider = ArgentinaApiProvider(api_url="http://fake-api.com/{year}")
+        with patch("requests.get", return_value=mock_response):
+            holidays = provider.get_holidays(year)
+            assert len(holidays) == 1
+            assert holidays[0].description == "Good"
+
+    def test_get_holidays_network_error(self):
+        """
+        Test that network errors are handled gracefully.
+        """
+        provider = ArgentinaApiProvider(api_url="http://fake-api.com/{year}")
+        with patch(
+            "requests.get", side_effect=requests.RequestException("Network Error")
+        ):
+            holidays = provider.get_holidays(2025)
+            assert holidays == []
+
+    def test_get_holidays_json_error(self, mock_response):
+        """
+        Test that JSON decoding errors are handled gracefully.
+        """
+        mock_response.json = MagicMock(side_effect=ValueError("Invalid JSON"))
+        provider = ArgentinaApiProvider(api_url="http://fake-api.com/{year}")
+        with patch("requests.get", return_value=mock_response):
+            holidays = provider.get_holidays(2025)
+            assert holidays == []