added missing auth failure exception handeling

StorjOld · Sep 1, 2015 · 36eac2e · 36eac2e
1 parent d219671
commit 36eac2e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dataserv/app.py b/dataserv/app.py
@@ -85,6 +85,9 @@ def ping(btc_addr):
     except LookupError:
         msg = "Farmer not found."
         return make_response(error_msg.format(msg), 404)
+    except AuthError:
+        msg = "Invalid authentication headers."
+        return make_response(error_msg.format(msg), 401)
 
 
 @app.route('/api/address', methods=["GET"])
@@ -155,6 +158,9 @@ def set_height(btc_addr, height):
     except LookupError:
         msg = "Farmer not found."
         return make_response(msg, 404)
+    except AuthError:
+        msg = "Invalid authentication headers."
+        return make_response(error_msg.format(msg), 401)
 
 
 if __name__ == '__main__':  # pragma: no cover