diff options
| -rw-r--r-- | pysite/database.py | 4 |
1 files changed, 2 insertions, 2 deletions
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, _): """ |