feat: allow to pass SlotSearchType to find-appointment command (#14)

* feat: allow to pass slotSearchType to find-appointment command

fix cs

fix cs

fix cs

* update README.md

Author: dawidgora

README.md

@@ -130,6 +130,14 @@ All commands are run from the web terminal.
     python medichaser.py find-appointment -r 204 -s 112 -l 60
     ```
 
+- **Search diagnostic procedures** (set slot search type):
+
+    ```bash
+    python medichaser.py find-appointment -r 204 -s 132 -S DiagnosticProcedure
+    ```
+
+    Use `-S`/`--slot-search-type` to override the default slot search type.
+
 - **Continuous monitoring and notifications**:
 
     ```bash

medichaser.py

@@ -77,6 +77,8 @@
 MEDICOVER_MAIN_URL = "https://online24.medicover.pl"
 MEDICOVER_API_URL = "https://api-gateway-online24.medicover.pl"
 
+DEFAULT_SLOT_SEARCH_TYPE: int = 0
+
 token_lock = FileLock(TOKEN_LOCK_PATH, timeout=60)
 login_lock = FileLock(LOGIN_LOCK_PATH, timeout=60)
 
@@ -801,6 +803,7 @@ def find_appointments(
         end_date: datetime.date | None,
         language: int | None,
         doctor: int | None = None,
+        slot_search_type: int | str = DEFAULT_SLOT_SEARCH_TYPE,
     ) -> list[dict[str, Any]]:
         appointment_url = (
             f"{MEDICOVER_API_URL}/appointments/api/search-appointments/slots"
@@ -812,7 +815,7 @@ def find_appointments(
             "Page": 1,
             "PageSize": 5000,
             "StartTime": start_date.isoformat(),
-            "SlotSearchType": 0,
+            "SlotSearchType": slot_search_type,
             "VisitType": "Center",
         }
 
@@ -837,13 +840,16 @@ def find_appointments(
         return items
 
     def find_filters(
-        self, region: int | None = None, specialty: list[int] | None = None
+        self,
+        region: int | None = None,
+        specialty: list[int] | None = None,
+        slot_search_type: int | str = DEFAULT_SLOT_SEARCH_TYPE,
     ) -> dict[str, Any]:
         filters_url = (
             f"{MEDICOVER_API_URL}/appointments/api/search-appointments/filters"
         )
 
-        params: dict[str, Any] = {"SlotSearchType": 0}
+        params: dict[str, Any] = {"SlotSearchType": slot_search_type}
         if region:
             params["RegionIds"] = region
         if specialty:
@@ -950,6 +956,13 @@ def json_date_serializer(obj: Any) -> str:
     raise TypeError(f"Object of type {type(obj).__name__} is not JSON serializable")
 
 
+def parse_slot_search_type(value: str) -> int | str:
+    try:
+        return int(value)
+    except ValueError:
+        return value
+
+
 class NextRun:
     def __init__(self, interval_minutes: int | None = 60) -> None:
         self.next_run = datetime.datetime.now(tz=datetime.UTC)
@@ -1033,6 +1046,13 @@ def main() -> None:
         default=None,
         help="Repeat interval in minutes",
     )
+    find_appointment.add_argument(
+        "-S",
+        "--slot-search-type",
+        default=DEFAULT_SLOT_SEARCH_TYPE,
+        type=parse_slot_search_type,
+        help='Slot search type (e.g. "0", "DiagnosticProcedure"). Default: 0',
+    )
 
     list_filters = subparsers.add_parser("list-filters", help="List filters")
     list_filters_subparsers = list_filters.add_subparsers(
@@ -1120,6 +1140,7 @@ def main() -> None:
                         args.enddate,
                         args.language,
                         args.doctor,
+                        args.slot_search_type,
                     )
                 except ExpiredToken as e:
                     log.warning("Expired token error: %s", e)

tests.py

@@ -26,6 +26,7 @@
 from notifiers.exceptions import BadArguments
 
 from medichaser import (
+    DEFAULT_SLOT_SEARCH_TYPE,
     AppointmentFinder,
     Authenticator,
     InvalidGrantError,
@@ -460,6 +461,10 @@ def test_find_appointments_basic(self, monkeypatch: pytest.MonkeyPatch) -> None:
 
         assert result == [{"id": 1}, {"id": 2}]
         mock_http_get.assert_called_once()
+        call_args = mock_http_get.call_args
+        assert call_args is not None
+        params: dict[str, Any] = call_args.args[1]
+        assert params["SlotSearchType"] == DEFAULT_SLOT_SEARCH_TYPE
 
     def test_find_appointments_with_filters(
         self, monkeypatch: pytest.MonkeyPatch
@@ -486,6 +491,7 @@ def test_find_appointments_with_filters(
         call_args = mock_http_get.call_args
         assert call_args is not None
         params: dict[str, Any] = call_args.args[1]
+        assert params["SlotSearchType"] == DEFAULT_SLOT_SEARCH_TYPE
         assert "DoctorLanguageIds" in params
         assert "DoctorIds" in params
 
@@ -534,6 +540,10 @@ def test_find_filters(self, monkeypatch: pytest.MonkeyPatch) -> None:
 
         assert result == {"regions": [{"id": 1}]}
         mock_http_get.assert_called_once()
+        call_args = mock_http_get.call_args
+        assert call_args is not None
+        params: dict[str, Any] = call_args.args[1]
+        assert params["SlotSearchType"] == DEFAULT_SLOT_SEARCH_TYPE
 
 
 class TestNotifier:
@@ -1099,6 +1109,7 @@ def test_main_find_appointment_single_run(monkeypatch: pytest.MonkeyPatch) -> No
         interval=None,
         notification="pushbullet",
         title="Test",
+        slot_search_type="DiagnosticProcedure",
     )
 
     mock_parser = MagicMock()
@@ -1133,7 +1144,14 @@ def test_main_find_appointment_single_run(monkeypatch: pytest.MonkeyPatch) -> No
     mock_auth_instance.login.assert_called_once()
     mock_auth_instance.refresh_token.assert_called_once()
     mock_finder_instance.find_appointments.assert_called_once_with(
-        1, [2], 3, datetime.date(2025, 1, 1), datetime.date(2025, 1, 31), 6, 4
+        1,
+        [2],
+        3,
+        datetime.date(2025, 1, 1),
+        datetime.date(2025, 1, 31),
+        6,
+        4,
+        "DiagnosticProcedure",
     )
     mock_display.assert_called_once_with([{"id": 1, "name": "Appointment"}])
     mock_notifier.assert_called_once_with(
@@ -1248,6 +1266,7 @@ def test_main_find_appointment_interval_run(monkeypatch: pytest.MonkeyPatch) ->
         interval=10,
         notification="pushover",
         title="Interval Test",
+        slot_search_type=DEFAULT_SLOT_SEARCH_TYPE,
     )
 
     mock_parser = MagicMock()