From 537bad0812617635055ba04d62346a008f3f54e5 Mon Sep 17 00:00:00 2001
From: Matt Jia <mjia@redhat.com>
Date: Jun 12 2017 23:34:43 +0000
Subject: add errorhandlers to handle network problem


---

diff --git a/greenwave/app_factory.py b/greenwave/app_factory.py
index 6ed3b86..d2e9eec 100644
--- a/greenwave/app_factory.py
+++ b/greenwave/app_factory.py
@@ -14,6 +14,7 @@ import os
 from flask import Flask
 from greenwave.logger import init_logging
 from greenwave.api_v1 import api
+from requests import ConnectionError, Timeout
 
 
 def load_config(app):
@@ -41,6 +42,9 @@ def create_app(config_obj=None):
         load_config(app)
     if app.config['PRODUCTION'] and app.secret_key == 'replace-me-with-something-random':
         raise Warning("You need to change the app.secret_key value for production")
+    # register error handlers
+    app.register_error_handler(ConnectionError, lambda e: (str(e), 503))
+    app.register_error_handler(Timeout, lambda e: (str(e), 503))
     # initialize logging
     init_logging(app)
     # register blueprints