diff --git a/main.py b/main.py index 95433e1..1f68148 100644 --- a/main.py +++ b/main.py @@ -39,6 +39,19 @@ def format_date(dt): return dt.strftime("%a %d").lower() +def pollen_desc(index): + if index < 2.5: + return f"{index} l" + elif index < 4.9: + return f"{index} l-m" + elif index < 7.3: + return f"{index} m" + elif index < 9.7: + return f"{index} m-h" + else: + return f"{index} h" + + def relative_day_to_date(rel_dt): dt = datetime.now() day = timedelta(days=1) @@ -89,7 +102,7 @@ async def fetch_pollen(zipcode): datetime.fromisoformat(data["ForecastDate"]) ), "periods": [ - {"index": d["Index"], "period": relative_day_to_date(d["Type"])} + {"index": pollen_desc(d["Index"]), "period": relative_day_to_date(d["Type"])} for d in data["Location"]["periods"] ], }