Allow defining postgres connection settings via env vars

This commit is contained in:
Devin Buhl 2023-04-14 09:58:02 -04:00 committed by GitHub
parent fab9f2ea85
commit 695734abe6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 7 deletions

View File

@ -22,7 +22,7 @@ from .get_args import args
logger = logging.getLogger(__name__)
postgresql = settings.postgresql.getboolean('enabled')
postgresql = (os.getenv("POSTGRES_ENABLED", settings.postgresql.enabled).lower() == 'true')
region = make_region().configure('dogpile.cache.memory')
@ -32,13 +32,19 @@ if postgresql:
(OperationalError, 'server closed the connection unexpectedly'),
)
postgres_database = os.getenv("POSTGRES_DATABASE", settings.postgresql.database)
postgres_username = os.getenv("POSTGRES_USERNAME", settings.postgresql.username)
postgres_password = os.getenv("POSTGRES_PASSWORD", settings.postgresql.password)
postgres_host = os.getenv("POSTGRES_HOST", settings.postgresql.host)
postgres_port = os.getenv("POSTGRES_PORT", settings.postgresql.port)
logger.debug(
f"Connecting to PostgreSQL database: {settings.postgresql.host}:{settings.postgresql.port}/{settings.postgresql.database}")
database = ReconnectPostgresqlDatabase(settings.postgresql.database,
user=settings.postgresql.username,
password=settings.postgresql.password,
host=settings.postgresql.host,
port=settings.postgresql.port,
f"Connecting to PostgreSQL database: {postgres_host}:{postgres_port}/{postgres_database}")
database = ReconnectPostgresqlDatabase(postgres_database,
user=postgres_username,
password=postgres_password,
host=postgres_host,
port=postgres_port,
autocommit=True,
autorollback=True,
autoconnect=True,