From 099182601c3ec7c6ff098445fcb43510f625cac6 Mon Sep 17 00:00:00 2001 From: Jarek Potiuk Date: Tue, 3 Jan 2023 19:55:49 +0100 Subject: [PATCH] Add correct widgets in Docker Hook The change refactored DockerHook and by mistake it missed the propoer widget specification. For some reason the tests did not catch it (to be investigated) but this should fix the failing tests. --- airflow/providers/docker/hooks/docker.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/airflow/providers/docker/hooks/docker.py b/airflow/providers/docker/hooks/docker.py index f4c3e9c44c0d1..c93804f320ef8 100644 --- a/airflow/providers/docker/hooks/docker.py +++ b/airflow/providers/docker/hooks/docker.py @@ -164,8 +164,20 @@ def __login(self, client, conn: Connection) -> None: self.log.error("Login failed") raise + @staticmethod + def get_connection_form_widgets() -> dict[str, Any]: + """Returns connection form widgets""" + from flask_appbuilder.fieldwidgets import BS3TextFieldWidget + from flask_babel import lazy_gettext + from wtforms import BooleanField, StringField + + return { + "reauth": BooleanField(lazy_gettext("Disable SSL")), + "email": StringField(lazy_gettext("Email"), widget=BS3TextFieldWidget()), + } + @classmethod - def get_connection_form_widgets(cls) -> dict[str, Any]: + def get_ui_field_behaviour(cls) -> dict[str, Any]: """Returns custom field behaviour""" return { "hidden_fields": ["schema"],