plot_stock_market example retry on failed download of data from google (scikit-learn#9172)

hakaa1 · hakaa1 · commit 91e79908028e · 2017-07-22T16:28:44.000+02:00
diff --git a/examples/applications/plot_stock_market.py b/examples/applications/plot_stock_market.py
@@ -77,7 +77,34 @@
 # #############################################################################
 # Retrieve the data from Internet
 
+def retry(f, n_retry=3):
+    """Define wrapper function to retry function calls in case of Exceptions
 
+    Parameters
+    -----------
+    f: function
+        Function that will be wrapped
+    n_retry: int
+        Number of retries of function call
+
+    Returns
+    -------
+    f : function
+        Wrapped input function
+
+    """
+    def wrapper(*args, **kwargs):
+        exception = None
+        for i in range(n_retry):
+            try:
+                return f(*args, **kwargs)
+            except Exception as e:
+                exception = e
+        raise exception
+    return wrapper
+
+
+@retry
 def quotes_historical_google(symbol, date1, date2):
     """Get the historical data from Google finance.
 
