Adding missing attributes for PinotDbApiHook and PinotAdminHook

apache · Aug 23, 2023 · 070c381 · 070c381
1 parent 9c632a0
commit 070c381
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/airflow/providers/apache/pinot/hooks/pinot.py b/airflow/providers/apache/pinot/hooks/pinot.py
@@ -56,6 +56,11 @@ class PinotAdminHook(BaseHook):
                                     "Exception" is in the output message.
     """
 
+    conn_name_attr = "conn_id"
+    default_conn_name = "pinot_admin_default"
+    conn_type = "pinot_admin"
+    hook_name = "Pinot Admin"
+
     def __init__(
         self,
         conn_id: str = "pinot_admin_default",
@@ -258,6 +263,8 @@ class PinotDbApiHook(DbApiHook):
 
     conn_name_attr = "pinot_broker_conn_id"
     default_conn_name = "pinot_broker_default"
+    conn_type = "pinot"
+    hook_name = "Pinot Broker"
     supports_autocommit = False
 
     def get_conn(self) -> Any: