Add Switzerland fetcher

Author: thiagovmdon

docs/api.rst

@@ -23,6 +23,7 @@ API Reference
    fetchers/slovenia
    fetchers/southafrica
    fetchers/spain
+   fetchers/switzerland
    fetchers/uk_ea
    fetchers/uk_nrfa
-   fetchers/usa
+   fetchers/usa

docs/fetchers/switzerland.rst

@@ -0,0 +1,5 @@
+Switzerland Fetcher
+===================
+
+.. automodule:: rivretrieve.switzerland
+   :members:

examples/plot_switzerland_historical.py

@@ -0,0 +1,44 @@
+import logging
+import os
+
+import matplotlib.pyplot as plt
+
+from rivretrieve import SwitzerlandFetcher, constants
+
+logging.basicConfig(level=logging.INFO)
+
+influx_token = os.environ.get("RIVRETRIEVE_SWITZERLAND_INFLUX_TOKEN")
+if not influx_token:
+    raise SystemExit("Set RIVRETRIEVE_SWITZERLAND_INFLUX_TOKEN before running this example.")
+
+gauge_id = "2016"
+variable = constants.DISCHARGE_DAILY_MEAN
+start_date = "2020-01-01"
+end_date = "2020-01-31"
+
+fetcher = SwitzerlandFetcher(influx_token=influx_token)
+data = fetcher.get_data(
+    gauge_id=gauge_id,
+    variable=variable,
+    start_date=start_date,
+    end_date=end_date,
+)
+
+if data.empty:
+    print(f"No data found for gauge {gauge_id}, variable {variable}")
+else:
+    print(data.head())
+    print(f"Fetched {len(data)} rows from {data.index.min()} to {data.index.max()}")
+
+    plt.figure(figsize=(12, 6))
+    plt.plot(data.index, data[variable], label=f"{gauge_id} - {variable}", linewidth=1.8)
+    plt.xlabel(constants.TIME_INDEX)
+    plt.ylabel("Discharge (m3/s)")
+    plt.title(f"Switzerland historical series for gauge {gauge_id}")
+    plt.grid(True, alpha=0.3)
+    plt.legend()
+    plt.tight_layout()
+
+    output_path = "switzerland_historical_plot.png"
+    plt.savefig(output_path, dpi=150)
+    print(f"Saved plot to {output_path}")

examples/test_switzerland_fetcher.py

@@ -0,0 +1,41 @@
+import logging
+import os
+
+import matplotlib.pyplot as plt
+
+from rivretrieve import SwitzerlandFetcher, constants
+
+logging.basicConfig(level=logging.INFO)
+
+influx_token = os.environ.get("RIVRETRIEVE_SWITZERLAND_INFLUX_TOKEN")
+if not influx_token:
+    raise SystemExit("Set RIVRETRIEVE_SWITZERLAND_INFLUX_TOKEN before running this example.")
+
+gauge_id = "2016"
+variables = [
+    constants.DISCHARGE_DAILY_MEAN,
+    constants.WATER_TEMPERATURE_DAILY_MEAN,
+]
+start_date = "2026-02-01"
+end_date = "2026-02-07"
+
+fetcher = SwitzerlandFetcher(influx_token=influx_token)
+
+for variable in variables:
+    data = fetcher.get_data(gauge_id=gauge_id, variable=variable, start_date=start_date, end_date=end_date)
+    if data.empty:
+        print(f"No data found for {gauge_id} ({variable})")
+        continue
+
+    print(data.head())
+    plt.figure(figsize=(12, 6))
+    plt.plot(data.index, data[variable], label=f"{gauge_id} - {variable}")
+    plt.xlabel(constants.TIME_INDEX)
+    plt.ylabel(variable)
+    plt.title(f"Switzerland River Data ({gauge_id})")
+    plt.legend()
+    plt.grid(True)
+    plt.tight_layout()
+    plot_path = f"switzerland_{variable}_plot.png"
+    plt.savefig(plot_path)
+    print(f"Plot saved to {plot_path}")

rivretrieve/__init__.py

@@ -16,6 +16,7 @@
 from .slovenia import SloveniaFetcher
 from .southafrica import SouthAfricaFetcher
 from .spain import SpainFetcher
+from .switzerland import SwitzerlandFetcher
 from .uk_ea import UKEAFetcher
 from .uk_nrfa import UKNRFAFetcher
 from .usa import USAFetcher