1
0
mirror of https://github.com/quay/quay.git synced 2026-01-27 18:42:52 +03:00
Files
quay/util/config/validators/validate_database.py
2020-02-05 19:55:07 -08:00

24 lines
706 B
Python

from peewee import OperationalError
from data.database import validate_database_precondition
from util.config.validators import BaseValidator, ConfigValidationException
class DatabaseValidator(BaseValidator):
name = "database"
@classmethod
def validate(cls, validator_context):
"""
Validates connecting to the database.
"""
config = validator_context.config
try:
validate_database_precondition(config["DB_URI"], config.get("DB_CONNECTION_ARGS", {}))
except OperationalError as ex:
if ex.args and len(ex.args) > 1:
raise ConfigValidationException(ex.args[1])
else:
raise ex