add timeout to get_psm3 (#741)

* add timeout

* update whatsnew

Author: cwhanse

docs/sphinx/source/whatsnew/v0.7.0.rst

@@ -132,6 +132,11 @@ Removal of prior version deprecations
 * Removed `ModelChain.prepare_inputs` clearsky assumption when no irradiance
   data was provided.
 
+Enhancements
+~~~~~~~~~~~~
+* Add `timeout` to :py:func:`pvlib.iotools.get_psm3`.
+
+
 Contributors
 ~~~~~~~~~~~~
 * Mark Campanellli (:ghuser:`markcampanelli`)

pvlib/iotools/psm3.py

@@ -19,7 +19,7 @@
 
 
 def get_psm3(latitude, longitude, api_key, email, names='tmy', interval=60,
-             full_name=PVLIB_PYTHON, affiliation=PVLIB_PYTHON):
+             full_name=PVLIB_PYTHON, affiliation=PVLIB_PYTHON, timeout=30):
     """
     Get PSM3 data
 
@@ -43,6 +43,8 @@ def get_psm3(latitude, longitude, api_key, email, names='tmy', interval=60,
         optional
     affiliation : str, default 'pvlib python'
         optional
+    timeout : int, default 30
+        time in seconds to wait for server response before timeout
 
     Returns
     -------
@@ -140,7 +142,7 @@ def get_psm3(latitude, longitude, api_key, email, names='tmy', interval=60,
         'interval': interval
     }
     # request CSV download from NREL PSM3
-    response = requests.get(URL, params=params)
+    response = requests.get(URL, params=params, timeout=timeout)
     if not response.ok:
         # if the API key is rejected, then the response status will be 403
         # Forbidden, and then the error is in the content and there is no JSON