Connect to all schemas in the database? #123

Activity

1. vincentsarago commented on Sep 21, 2023

   

   vincentsarago

   on Sep 21, 2023

   Member

   More actions

   @swdmike I don't think it's currently possible but you could customize this

   tipg/tipg/database.py

   Lines 23 to 67 in fdb6ec6

   	class connection_factory:
   	"""Connection creation."""
   	schemas: List[str]
   	user_sql_files: List[pathlib.Path]
   	def __init__(
   	self,
   	schemas: Optional[List[str]] = None,
   	user_sql_files: Optional[List[pathlib.Path]] = None,
   	) -> None:
   	"""Init."""
   	self.schemas = schemas or []
   	self.user_sql_files = user_sql_files or []
   	async def __call__(self, conn: asyncpg.Connection):
   	"""Create connection."""
   	await conn.set_type_codec(
   	"json", encoder=orjson.dumps, decoder=orjson.loads, schema="pg_catalog"
   	)
   	await conn.set_type_codec(
   	"jsonb", encoder=orjson.dumps, decoder=orjson.loads, schema="pg_catalog"
   	)
   	# Note: we add `pg_temp as the first element of the schemas list to make sure
   	# we register the custom functions and `dbcatalog` in it.
   	schemas = ",".join(["pg_temp", *self.schemas])
   	logger.debug(f"Looking for Tables and Functions in {schemas} schemas")
   	await conn.execute(
   	f"""
   	SELECT set_config(
   	'search_path',
   	'{schemas},' || current_setting('search_path', false),
   	false
   	);
   	"""
   	)
   	# Register custom SQL functions/table/views in pg_temp
   	for sqlfile in self.user_sql_files:
   	await conn.execute(sqlfile.read_text())
   	# Register TiPG functions in `pg_temp`
   	await conn.execute(DB_CATALOG_FILE.read_text())

   to first list all the schemas and then construct the search path

   tipg/tipg/database.py

   Line 49 in fdb6ec6

   schemas = ",".join(["pg_temp", *self.schemas])
2. developmentseed

   locked and limited conversation to collaborators on Sep 21, 2023
3. vincentsarago

   converted this issue into a discussion #124 on Sep 21, 2023