Added timeout for data providers #18

pskowronek · Oct 7, 2023 · a9309b5 · a9309b5
1 parent f2515db
commit a9309b5
Show file tree

Hide file tree

Showing 7 changed files with 14 additions and 7 deletions.
diff --git a/providers/airly.py b/providers/airly.py
@@ -45,7 +45,8 @@ def acquire(self):
                     "apikey" : self.key,
                     "Accept-Language" : "en",
                     "Accept" : "application/json"
-                }
+                },
+                timeout=(2, 4)
             )
             return r.status_code, r.text
         except Exception as e:

diff --git a/providers/aqicn.py b/providers/aqicn.py
@@ -40,7 +40,8 @@ def acquire(self):
                 "https://api.waqi.info/feed/{}/?token={}".format(
                     self.city_or_id if self.city_or_id else "geo:{};{}".format(self.lat, self.lon),
                     self.key
-                )
+                ),
+                timeout=(2, 4)
             )
             return r.status_code, r.text
         except Exception as e:

diff --git a/providers/darksky.py b/providers/darksky.py
@@ -48,7 +48,8 @@ def acquire(self):
                 params = {
                     "units" : self.units,
                     "exclude" : "minutely,hourly,flags"
-                }
+                },
+                timeout=(2, 4)
             )
             return r.status_code, r.text
 

diff --git a/providers/gmaps.py b/providers/gmaps.py
@@ -66,6 +66,7 @@ def acquire(self):
                     self.dest_lon,
                     self.key
                 ),
+                timeout=(2, 4)
             )
             return r.status_code, r.text
         except Exception as e:

diff --git a/providers/openweather.py b/providers/openweather.py
@@ -43,7 +43,8 @@ def acquire(self):
                     "lat" : self.lat,
                     "lon" : self.lon,
                     "units" : self.units
-                }
+                },
+                timeout=(2, 4)
             )
             return r.status_code, r.text
 

diff --git a/providers/weatherbit.py b/providers/weatherbit.py
@@ -44,7 +44,8 @@ def acquire(self):
                     "lat" : self.lat,
                     "lon" : self.lon,
                     "units" : self.units
-                }
+                },
+                timeout=(2, 4)
             )
             current = r.json()
 
@@ -55,7 +56,8 @@ def acquire(self):
                     "lat" : self.lat,
                     "lon" : self.lon,
                     "units" : self.units
-                }
+                },
+                timeout=(2, 4)
             )
             forecast = r.json()
 

diff --git a/requirements.txt b/requirements.txt
@@ -34,4 +34,4 @@ psutil == 5.4.7
 uptime == 3.0.1
 
 # providers/meteoalarm.py: 6
-meteoalertapi == 0.2.0 
+meteoalertapi == 0.3.0