From d9783d8c9a38fa3370b2e2eb84e8f858e335a6bd Mon Sep 17 00:00:00 2001 From: Jonathan Zernik Date: Fri, 31 Jul 2020 20:21:29 -0700 Subject: [PATCH] Fix search path for schema in db (#188) --- squeakserver/server/postgres_db.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/squeakserver/server/postgres_db.py b/squeakserver/server/postgres_db.py index 65c7b71b..a9f95b30 100644 --- a/squeakserver/server/postgres_db.py +++ b/squeakserver/server/postgres_db.py @@ -51,7 +51,10 @@ class PostgresDb: create_schema_sql = sql.SQL("CREATE SCHEMA IF NOT EXISTS {};").format( sql.Identifier(schema_name) ) - use_schema_sql = sql.SQL("ALTER ROLE {} SET search_path TO {};").format( + use_schema_sql = sql.SQL("SET search_path TO {};").format( + sql.Identifier(schema_name), + ) + use_schema_for_user_sql = sql.SQL("ALTER ROLE {} SET search_path TO {};").format( sql.Identifier(self.user), sql.Identifier(schema_name), ) @@ -60,6 +63,7 @@ class PostgresDb: logger.info("Creating schema: {}".format(schema_name)) curs.execute(create_schema_sql) curs.execute(use_schema_sql) + curs.execute(use_schema_for_user_sql) def init(self): """ Create the tables and indices in the database. """