From 10c35f54514b9fd5f52b6277efd87e3b16cd8ad7 Mon Sep 17 00:00:00 2001 From: Stefan Hellander Date: Wed, 5 Mar 2025 09:45:56 +0100 Subject: [PATCH] Switch statestore to Postgres --- fedn/network/storage/dbconnection.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/fedn/network/storage/dbconnection.py b/fedn/network/storage/dbconnection.py index 0e6b7af79..01e90bf07 100644 --- a/fedn/network/storage/dbconnection.py +++ b/fedn/network/storage/dbconnection.py @@ -153,12 +153,11 @@ def _setup_sql(self, statestore_config: dict) -> "DatabaseConnection": dbname = sqlite_config["dbname"] engine = create_engine(f"sqlite:///{dbname}", echo=False) elif statestore_type == "PostgreSQL": - postgres_config = statestore_config["postgres_config"] - username = os.environ.get("FEDN_STATESTORE_USERNAME", postgres_config["username"]) - password = os.environ.get("FEDN_STATESTORE_PASSWORD", postgres_config["password"]) - host = os.environ.get("FEDN_STATESTORE_HOST", postgres_config["host"]) - port = os.environ.get("FEDN_STATESTORE_PORT", postgres_config["port"]) - dbname = os.environ.get("FEDN_STATESTORE_DBNAME", postgres_config["dbname"]) + username = os.environ.get("FEDN_STATESTORE_USERNAME", statestore_config["postgres_config"]["username"]) + password = os.environ.get("FEDN_STATESTORE_PASSWORD", statestore_config["postgres_config"]["password"]) + host = os.environ.get("FEDN_STATESTORE_HOST", statestore_config["postgres_config"]["host"]) + port = os.environ.get("FEDN_STATESTORE_PORT", statestore_config["postgres_config"]["port"]) + dbname = os.environ.get("FEDN_STATESTORE_DBNAME", statestore_config["postgres_config"]["dbname"]) engine = create_engine(f"postgresql://{username}:{password}@{host}:{port}/{dbname}", echo=False) Session = sessionmaker(engine) # noqa: N806