add optional timeout

freekode · Apr 28, 2024 · 75c5615 · 75c5615
1 parent db274dd
commit 75c5615
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyopenweathermap"
-version = "0.0.5"
+version = "0.0.6"
 description = "lib for OpenWeatherMap for Home Assistant"
 authors = ["Evgeny <[email protected]>"]
 license = "MIT"

diff --git a/src/pyopenweathermap/owm_client.py b/src/pyopenweathermap/owm_client.py
@@ -9,12 +9,13 @@
 
 class OWMClient:
     session: ClientSession | None = None
-    request_timeout: int = 10
+    request_timeout: int
 
-    def __init__(self, api_key, units="metric", lang='en'):
+    def __init__(self, api_key, units="metric", lang='en', request_timeout=20):
         self.api_key = api_key
         self.units = units
         self.lang = lang
+        self.request_timeout = request_timeout
 
     async def get_weather(self, lat, lon, weather_types=None) -> WeatherReport:
         if weather_types is None: