Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
34 changes: 34 additions & 0 deletions providers/trino/src/airflow/providers/trino/hooks/trino.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,40 @@ class TrinoHook(DbApiHook):
query_id = ""
_test_connection_sql = "select 1"

@classmethod
def get_ui_field_behaviour(cls) -> dict[str, Any]:
"""Return custom field behaviour."""
return {
"hidden_fields": [],
"relabeling": {},
"placeholders": {
"extra": json.dumps(
{
"auth": "authentication type",
"impersonate_as_owner": "allow impersonate as owner",
"jwt__token": "JWT token",
"jwt__file": "JWT file path",
"certs__client_cert_path": "Client certificate path",
"certs__client_key_path": "Client key path",
"kerberos__config": "Kerberos config",
"kerberos__service_name": "Kerberos service name",
"kerberos__mutual_authentication": "Kerberos mutual authentication",
"kerberos__force_preemptive": "Kerberos force preemptive",
"kerberos__hostname_override": "Kerberos hostname override",
"kerberos__sanitize_mutual_error_response": "Kerberos sanitize mutual error response",
"kerberos__principal": "Kerberos principal",
"kerberos__delegate": "Kerberos delegate",
"kerberos__ca_bundle": "Kerberos CA bundle",
"session_properties": "session properties",
"client_tags": "Trino client tags. Example ['sales','cluster1']",
"timezone": "Trino timezone",
},
indent=1,
),
"login": "Effective user for connection",
},
}

def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._placeholder: str = "?"
Expand Down