Allow setting embedded postgres db directory with PGSERVER_DATA_DIR env var (#1202)

It seems to me that it would be useful to be able to be explicit about
where the embedded database should live - as well as where it _does_
live (via the info log), when not specifying.
This commit is contained in:
Peter Gaultney
2025-06-27 17:51:23 -04:00
committed by GitHub
parent d37113850c
commit 9f3ceba541

View File

@@ -139,10 +139,10 @@ if USE_EMBEDDED_DB:
import pgserver import pgserver
# Set up data directory # Set up data directory
PGSERVER_DATA_DIR = os.path.join(BASE_DIR, "pgserver_data") PGSERVER_DATA_DIR = os.getenv("PGSERVER_DATA_DIR") or os.path.join(BASE_DIR, "pgserver_data")
os.makedirs(PGSERVER_DATA_DIR, exist_ok=True) os.makedirs(PGSERVER_DATA_DIR, exist_ok=True)
logger.debug(f"Initializing embedded Postgres DB with data directory: {PGSERVER_DATA_DIR}") logger.info(f"Initializing embedded Postgres DB with data directory: {PGSERVER_DATA_DIR}")
# Start server # Start server
PGSERVER_INSTANCE = pgserver.get_server(PGSERVER_DATA_DIR) PGSERVER_INSTANCE = pgserver.get_server(PGSERVER_DATA_DIR)