feat: add cache and logging

gismeteo/api.py

@@ -1,7 +1,9 @@
 import datetime
+import logging
 from typing import Any, Dict, List
 
 import aiohttp
+from aiocache import cached
 from bs4 import BeautifulSoup
 
 from weather.api import WeatherApi
@@ -11,15 +13,27 @@ from . import datehelp
 from .location import LOCATION_BUNDLE
 from .parser import DAYS_PARSER, LOCATION_PARSER, ONE_DAY_PARSER, ROW_PARSERS
 
+logger = logging.getLogger("gismeteo")
+
 
 class GismeteoApi(WeatherApi):
     BASE_URL = "https://www.gismeteo.ru"
+    CACHE_TTL = 10 * 60
 
-    USER_AGENT = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36"
-    COOKIE = "cf_clearance=U28mYVC0ENu88vorlL_CWmWOoevvXp0vb4xCqfqYC9s-1722273367-1.0.1.1-IDV73azTHY0V.NAnmEvok3zf5HHEkvF098pmya7IiqRRB5nk3FhbLCb0AeWm_kpTFqi1niFk2mYN_ramGTSl0A"
+    USER_AGENT = (
+        "Mozilla/5.0 (X11; Linux x86_64) "
+        "AppleWebKit/537.36 (KHTML, like Gecko) "
+        "Chrome/126.0.0.0 Safari/537.36"
+    )
+    COOKIE = (
+        "cf_clearance=U28mYVC0ENu88vorlL_CWmWOoevvXp0vb4xCqfqYC9s-"
+        "1722273367-1.0.1.1-"
+        "IDV73azTHY0V.NAnmEvok3zf5HHEkvF098pmya7IiqRRB5nk3FhbLCb0AeWm_kpTFqi1niFk2mYN_ramGTSl0A"
+    )
 
     async def _request(self, endpoint: str) -> str:
         url = f"{self.BASE_URL}/{endpoint}"
+        logger.info(url)
         async with aiohttp.ClientSession(
             headers={
                 "User-Agent": self.USER_AGENT,
@@ -58,7 +72,6 @@ class GismeteoApi(WeatherApi):
                 while len(result) < index + 1:
                     result.append({})
                 result[index][parser.KEY] = value
-        print(">", result)
         values = [WeatherValue(**item) for item in result]
         return WeatherResponse(
             location=location or "n/a",
@@ -67,11 +80,13 @@ class GismeteoApi(WeatherApi):
             values=values,
         )
 
+    @cached(ttl=CACHE_TTL)
     async def get_day(self, location_id: str, date: datetime.date) -> WeatherResponse:
         location = LOCATION_BUNDLE.parse(location_id)
         data = await self._request(f"weather-{location}/{datehelp.dump(date)}")
         return self._parse_oneday(date, data)
 
+    @cached(ttl=CACHE_TTL)
     async def get_days(self, location_id: str, days: int) -> WeatherResponse:
         location = LOCATION_BUNDLE.parse(location_id)
         data = await self._request(f"weather-{location}/{days}-days")