allow user to set debug level via env var (#650)

* allow user to set debug level via env var * add changie
dbt-labs · Oct 30, 2023 · 969eaa6 · 969eaa6
1 parent d5d2d74
commit 969eaa6
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/.changes/unreleased/Features-20231030-101055.yaml b/.changes/unreleased/Features-20231030-101055.yaml
@@ -0,0 +1,6 @@
+kind: Features
+body: allow user to set debug level for redshift-connector via env var
+time: 2023-10-30T10:10:55.976191-07:00
+custom:
+  Author: colin-rogers-dbt
+  Issue: "650"
diff --git a/dbt/adapters/redshift/impl.py b/dbt/adapters/redshift/impl.py
@@ -1,3 +1,4 @@
+import os
 from dataclasses import dataclass
 from typing import Optional, Set, Any, Dict, Type
 from collections import namedtuple
@@ -14,9 +15,15 @@
 
 from dbt.adapters.redshift import RedshiftConnectionManager, RedshiftRelation
 
-
 logger = AdapterLogger("Redshift")
-
+packages = ["redshift_connector", "redshift_connector.core"]
+if os.getenv("DBT_REDSHIFT_CONNECTOR_DEBUG_LOGGING"):
+    level = "DEBUG"
+else:
+    level = "ERROR"
+for package in packages:
+    logger.debug(f"Setting {package} to {level}")
+    logger.set_adapter_dependency_log_level(package, level)
 
 GET_RELATIONS_MACRO_NAME = "redshift__get_relations"