From 695734abe67f56f34a5e99aa98cd558e2b39545b Mon Sep 17 00:00:00 2001 From: Devin Buhl Date: Fri, 14 Apr 2023 09:58:02 -0400 Subject: [PATCH] Allow defining postgres connection settings via env vars --- bazarr/app/database.py | 20 +++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) diff --git a/bazarr/app/database.py b/bazarr/app/database.py index 24fed4d2a..2b14a5c32 100644 --- a/bazarr/app/database.py +++ b/bazarr/app/database.py @@ -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,