fix: meteo test

VITObelgium · Mar 11, 2024 · db9d73e · db9d73e
1 parent d0f7161
commit db9d73e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pywig/__init__.py b/pywig/__init__.py
@@ -58,14 +58,15 @@ def get_meteo(self, field_id: str, key: str, start_date: datetime.date = None, e
         :rtype: List
         """
         field = self.get_field_details(field_id)
+        print(field)
         if field.meteo and key.lower() in field.meteo:
             meteo = list(map(lambda x: MeteoStat(date=x['date'], value=x['value']), field.meteo[key.lower()]))
             if start_date:
                 meteo = [x for x in meteo if start_date <= datetime.datetime.strptime(x.date, "%Y-%m-%d").date()]
             if end_date:
                 meteo = [x for x in meteo if datetime.datetime.strptime(x.date, "%Y-%m-%d").date() <= end_date]
             return meteo
-        elif field.metadata.country == 'Belgium':
+        elif field.metadata and field.metadata['country'] == 'Belgium':
             start_date = start_date.isoformat() if start_date else field.metadata['startDate']
             end_date = end_date.isoformat() if end_date else field.metadata['endDate']