Add timeout - Fixes #1

martin-majlis · Dec 16, 2017 · 2d304fe · 2d304fe
1 parent dbf50ed
commit 2d304fe
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/wikipediaapi/wikipedia.py b/wikipediaapi/wikipedia.py
@@ -39,6 +39,7 @@ def __init__(
             user_agent=(
             'Wikipedia-API (https://github.com/martin-majlis/Wikipedia-API)'
             ),
+            timeout=10.0
     ):
         '''
         Language of the API being requested.
@@ -48,6 +49,7 @@ def __init__(
         self.language = language.strip().lower()
         self.user_agent = user_agent
         self.extract_format = extract_format
+        self.timeout = timeout
 
     def page(
             self,
@@ -250,7 +252,8 @@ def _query(
         r = requests.get(
             base_url,
             params=params,
-            headers=headers
+            headers=headers,
+            timeout=self.timeout,
         )
         return r.json()