add logging

Author: peterdudfield

nowcasting_datamodel/read/read_gsp.py

@@ -1,12 +1,16 @@
 """ Read pv functions """
 from typing import List, Union
+import logging
 
 from sqlalchemy import desc
 from sqlalchemy.orm import Session
 
 from nowcasting_datamodel.models import GSPYieldSQL, LocationSQL
 
 
+logger = logging.getLogger(__name__)
+
+
 def get_latest_gsp_yield(
     session: Session, gsps: List[LocationSQL], append_to_gsps: bool = False, regime: str = "in-day"
 ) -> Union[List[GSPYieldSQL], List[LocationSQL]]:
@@ -47,6 +51,8 @@ def get_latest_gsp_yield(
     # get all results
     gsp_yields: List[GSPYieldSQL] = query.all()
 
+    logger.debug(f'Found {len(gsp_yields)}  latest gsp yields')
+
     if not append_to_gsps:
         return gsp_yields
     else:
@@ -58,16 +64,20 @@ def get_latest_gsp_yield(
 
             gsp_systems_with_gsp_yields.append(gsp)
 
+        logger.debug(f'Found {len(gsp_systems_with_gsp_yields)} gsps with yields')
+
         # add pv systems that dont have any pv yields
         gsp_systems_with_gsp_yields_ids = [gsp.id for gsp in gsp_systems_with_gsp_yields]
 
         gsp_systems_with_no_gsp_yields = []
         for gsp in gsps:
-            if gsp.gsp_id not in gsp_systems_with_gsp_yields_ids:
+            if gsp.id not in gsp_systems_with_gsp_yields_ids:
                 gsp.last_gsp_yield = None
 
                 gsp_systems_with_no_gsp_yields.append(gsp)
 
+        logger.debug(f'Found {len(gsp_systems_with_gsp_yields)} gsps with no yields')
+
         all_gsp_systems = gsp_systems_with_gsp_yields + gsp_systems_with_no_gsp_yields
 
         return all_gsp_systems