From 1a1721634b05b82ca1db541cf9b2222859560709 Mon Sep 17 00:00:00 2001 From: Jarek Potiuk Date: Mon, 3 Jul 2023 23:30:30 +0200 Subject: [PATCH] Add proxy_user template check --- airflow/providers/apache/hive/hooks/hive.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/airflow/providers/apache/hive/hooks/hive.py b/airflow/providers/apache/hive/hooks/hive.py index 4ad3b9d12c692..89fcda466ffc9 100644 --- a/airflow/providers/apache/hive/hooks/hive.py +++ b/airflow/providers/apache/hive/hooks/hive.py @@ -152,6 +152,8 @@ def _prepare_cli_cmd(self) -> list[Any]: proxy_user = self._get_proxy_user() if ";" in template: raise RuntimeError("The principal should not contain the ';' character") + if ";" in proxy_user: + raise RuntimeError("The proxy_user should not contain the ';' character") jdbc_url += f";principal={template};{proxy_user}" elif self.auth: jdbc_url += ";auth=" + self.auth