From f3ac4cfbeb0db171bf55440c7f0d2dce8d8d6e8f Mon Sep 17 00:00:00 2001 From: Joseph Date: Tue, 8 May 2018 22:16:08 +0100 Subject: Switch from using abort to using werkzeug exception --- pysite/database.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'pysite/database.py') diff --git a/pysite/database.py b/pysite/database.py index 41fbd253..b9ed5b89 100644 --- a/pysite/database.py +++ b/pysite/database.py @@ -7,9 +7,9 @@ from typing import Any, Callable, Dict, Iterator, List, Optional, Union import re import rethinkdb -from flask import abort from rethinkdb.ast import RqlMethodQuery, Table, UserError from rethinkdb.net import DefaultConnection +from werkzeug.exceptions import ServiceUnavailable from pysite.constants import DEBUG_MODE @@ -144,7 +144,7 @@ class RethinkDB: try: self.conn = self.get_connection() except rethinkdb.RqlDriverError: - abort(503, "Database connection could not be established.") + raise ServiceUnavailable("Database connection could not be established.") def teardown_request(self, _): """ -- cgit v1.2.3