Feature/eco counter serialize data from year field

eco_counter/api/serializers.py

@@ -1,3 +1,4 @@
+from django.db.models import Q
 from rest_framework import serializers
 
 from ..models import (
@@ -35,6 +36,7 @@ class StationSerializer(serializers.ModelSerializer):
     lon = serializers.SerializerMethodField()
     lat = serializers.SerializerMethodField()
     sensor_types = serializers.SerializerMethodField()
+    data_from_year = serializers.SerializerMethodField()
 
     class Meta:
         model = Station
@@ -52,6 +54,7 @@ class Meta:
             "lon",
             "lat",
             "sensor_types",
+            "data_from_year",
         ]
 
     def get_y(self, obj):
@@ -79,6 +82,19 @@ def get_sensor_types(self, obj):
                 result.append(type)
         return result
 
+    def get_data_from_year(self, obj):
+        q_exp = (
+            Q(value_at__gt=0)
+            | Q(value_pt__gt=0)
+            | Q(value_jt__gt=0)
+            | Q(value_bt__gt=0)
+        )
+        qs = YearData.objects.filter(q_exp, station=obj).order_by("year__year_number")
+        if qs.count() > 0:
+            return qs[0].year.year_number
+        else:
+            return None
+
 
 class YearSerializer(serializers.ModelSerializer):
     station_name = serializers.PrimaryKeyRelatedField(

eco_counter/tests/test_api.py

@@ -3,6 +3,7 @@
 import pytest
 from rest_framework.reverse import reverse
 
+from .conftest import TEST_TIMESTAMP
 from .constants import TEST_EC_STATION_NAME
 
 
@@ -271,6 +272,7 @@ def test__station(api_client, stations, year_datas):
     assert response.status_code == 200
     assert response.json()["results"][0]["name"] == TEST_EC_STATION_NAME
     assert response.json()["results"][0]["sensor_types"] == ["at"]
+    assert response.json()["results"][0]["data_from_year"] == TEST_TIMESTAMP.year
     # Test retrieving station by data type
     url = reverse("eco_counter:stations-list") + "?data_type=a"
     response = api_client.get(url)