Feat: sensor data source type filter

flexmeasures/api/common/schemas/sensor_data.py

@@ -174,6 +174,13 @@ class GetSensorDataFilterSchemaMixin:
             example=3,
         ),
     )
+    source_type = fields.Str(
+        required=False,
+        metadata=dict(
+            description="Filter by a specific data source type.",
+            example="demo script",
+        ),
+    )
 
 
 class GetSensorDataSchema(GetSensorDataFilterSchemaMixin, SensorDataDescriptionSchema):
@@ -228,6 +235,7 @@ def load_data_and_make_response(sensor_data_description: dict) -> dict:
         resolution = sensor_data_description.get("resolution")
         source = sensor_data_description.get("source")
         account = sensor_data_description.get("account")
+        source_type = sensor_data_description.get("source_type")
 
         # Post-load configuration of event frequency
         if resolution is None:
@@ -258,6 +266,7 @@ def load_data_and_make_response(sensor_data_description: dict) -> dict:
                 horizons_at_most=horizons_at_most,
                 source=source,
                 source_account_ids=account.id if account else None,
+                source_types=[source_type] if source_type else None,
                 beliefs_before=sensor_data_description.get("prior", None),
                 one_deterministic_belief_per_event=True,
                 resolution=resolution,

flexmeasures/api/v3_0/sensors.py

@@ -592,6 +592,7 @@ def get_data(self, id: int, **sensor_data_description: dict):
             - "prior" (the belief timing docs also apply here)
             - "source" (filter by data source ID, read [the docs about sources](https://flexmeasures.readthedocs.io/latest/api/notation.html#sources))
             - "account" (filter by the account ID linked to data sources)
+            - "source_type" (filter by data source type)
 
             An example query to fetch data for sensor with ID=1, for one hour starting June 7th 2021 at midnight, in 15 minute intervals, in m³/h:
 

flexmeasures/api/v3_0/tests/test_sensor_data.py

@@ -166,6 +166,59 @@ def test_get_sensor_data_filtered_by_source_account(
     assert all(a == b for a, b in zip(values, expected))
 
 
+@pytest.mark.parametrize(
+    "source_type, expected",
+    [
+        (
+            "user",
+            [
+                sum(GAS_MEASUREMENTS_10MIN[0:2]) / 2,  # 91.5
+                GAS_MEASUREMENTS_10MIN[2],  # 92.1
+                None,
+                None,
+            ],
+        ),
+        (
+            "demo script",
+            [
+                GAS_MEASUREMENTS_10MIN[0],  # 91.3
+                GAS_MEASUREMENTS_10MIN[2],  # 92.1
+                None,
+                None,
+            ],
+        ),
+        ("scheduler", [None, None, None, None]),
+    ],
+)
+@pytest.mark.parametrize(
+    "requesting_user", ["test_supplier_user_4@seita.nl"], indirect=True
+)
+def test_get_sensor_data_filtered_by_source_type(
+    client,
+    setup_api_test_data: dict[str, Sensor],
+    requesting_user,
+    source_type,
+    expected,
+):
+    """Check that GET /sensors/<id>/data can filter by source type."""
+    sensor = setup_api_test_data["some gas sensor"]
+    message = {
+        "start": "2021-05-02T00:00:00+02:00",
+        "duration": "PT1H20M",
+        "horizon": "PT0H",
+        "unit": "m³/h",
+        "source_type": source_type,
+        "resolution": "PT20M",
+    }
+    response = client.get(
+        url_for("SensorAPI:get_data", id=sensor.id),
+        query_string=message,
+    )
+    print("Server responded with:\n%s" % response.json)
+    assert response.status_code == 200
+    assert all(a == b for a, b in zip(response.json["values"], expected))
+
+
 @pytest.mark.parametrize(
     "requesting_user, status_code",
     [