tryna get tempbo working

2025-12-17 19:31:34 +00:00 · 2023-09-14 19:14:00 -04:00
1 changed files with 9 additions and 0 deletions
--- a/plugins/postgres/dbt/adapters/postgres/connections.py
+++ b/plugins/postgres/dbt/adapters/postgres/connections.py
@@ -32,7 +32,10 @@ class PostgresCredentials(Credentials):
    sslkey: Optional[str] = None
    sslrootcert: Optional[str] = None
    application_name: Optional[str] = "dbt"
+    endpoint: Optional[str] = None
    retries: int = 1
+    options: Optional[str] = None
+    # options: Dict[str, Any] = field(default_factory=dict)

    _ALIASES = {"dbname": "database", "pass": "password"}

@@ -130,6 +133,12 @@ class PostgresConnectionManager(SQLConnectionManager):
        if credentials.application_name:
            kwargs["application_name"] = credentials.application_name

+        if credentials.options:
+            kwargs["options"] = credentials.options
+
+        if credentials.endpoint:
+            kwargs["endpoint"] = credentials.endpoint
+
        def connect():
            handle = psycopg2.connect(
                dbname=credentials.database,