@@ -27,7 +27,9 @@ class ParadeDbExtension(DatabaseDockerContainerExtension):
2727 def __init__ (self ):
2828 # Get configuration from environment variables
2929 postgres_user = os .environ .get (ENV_POSTGRES_USER , DEFAULT_POSTGRES_USER )
30- postgres_password = os .environ .get (ENV_POSTGRES_PASSWORD , DEFAULT_POSTGRES_PASSWORD )
30+ postgres_password = os .environ .get (
31+ ENV_POSTGRES_PASSWORD , DEFAULT_POSTGRES_PASSWORD
32+ )
3133 postgres_db = os .environ .get (ENV_POSTGRES_DB , DEFAULT_POSTGRES_DB )
3234 postgres_port = int (os .environ .get (ENV_POSTGRES_PORT , DEFAULT_POSTGRES_PORT ))
3335
@@ -53,14 +55,16 @@ def __init__(self):
5355 def get_connection_info (self ) -> dict :
5456 """Return connection information for ParadeDB."""
5557 info = super ().get_connection_info ()
56- info .update ({
57- "database" : self .postgres_db ,
58- "user" : self .postgres_user ,
59- "password" : self .postgres_password ,
60- "port" : self .postgres_port ,
61- "connection_string" : (
62- f"postgresql://{ self .postgres_user } :{ self .postgres_password } "
63- f"@{ self .container_host } :{ self .postgres_port } /{ self .postgres_db } "
64- ),
65- })
58+ info .update (
59+ {
60+ "database" : self .postgres_db ,
61+ "user" : self .postgres_user ,
62+ "password" : self .postgres_password ,
63+ "port" : self .postgres_port ,
64+ "connection_string" : (
65+ f"postgresql://{ self .postgres_user } :{ self .postgres_password } "
66+ f"@{ self .container_host } :{ self .postgres_port } /{ self .postgres_db } "
67+ ),
68+ }
69+ )
6670 return info
0 commit comments