add support for dbt-athena adapter (OpenLineage#3110)

* add support for dbt-athena adapter Signed-off-by: Moritz <[email protected]> * fix athena namespace format for dbt integration Signed-off-by: Moritz <[email protected]> --------- Signed-off-by: Moritz <[email protected]>
nubank · Oct 4, 2024 · 78b788e · 78b788e
1 parent f34e8ab
commit 78b788e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/integration/common/openlineage/common/provider/dbt/processor.py b/integration/common/openlineage/common/provider/dbt/processor.py
@@ -40,6 +40,7 @@ class Adapter(Enum):
     DATABRICKS = "databricks"
     SQLSERVER = "sqlserver"
     DREMIO = "dremio"
+    ATHENA = "athena"
 
     @staticmethod
     def adapters() -> str:
@@ -583,6 +584,8 @@ def extract_namespace(self, profile: Dict) -> str:
             return f"mssql://{profile['server']}:{profile['port']}"
         elif self.adapter_type == Adapter.DREMIO:
             return f"dremio://{profile['software_host']}:{profile['port']}"
+        elif self.adapter_type == Adapter.ATHENA:
+            return f"awsathena://athena.{profile['region_name']}.amazonaws.com"
         elif self.adapter_type == Adapter.SPARK:
             port = ""