From 575ac401ab322e23ea89f336bc5fb19e25547e38 Mon Sep 17 00:00:00 2001 From: Kaxil Naik Date: Tue, 9 Feb 2021 20:53:27 +0000 Subject: [PATCH] Log migrations info in consistent way same as #13458 but for `82b7c48c147f_remove_can_read_permission_on_config_.py` migration This migration changes logging handlers so each next migration is differently formatted when doing airflow db reset. This commit fixes this behavior. --- .../82b7c48c147f_remove_can_read_permission_on_config_.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/airflow/migrations/versions/82b7c48c147f_remove_can_read_permission_on_config_.py b/airflow/migrations/versions/82b7c48c147f_remove_can_read_permission_on_config_.py index 5e85ee4a6ba82..85d0872704570 100644 --- a/airflow/migrations/versions/82b7c48c147f_remove_can_read_permission_on_config_.py +++ b/airflow/migrations/versions/82b7c48c147f_remove_can_read_permission_on_config_.py @@ -23,6 +23,7 @@ Create Date: 2021-02-04 12:45:58.138224 """ +import logging from airflow.security import permissions from airflow.www.app import create_app @@ -36,6 +37,9 @@ def upgrade(): """Remove can_read permission on config resource for User and Viewer role""" + log = logging.getLogger() + handlers = log.handlers[:] + appbuilder = create_app(config={'FAB_UPDATE_PERMS': False}).appbuilder roles_to_modify = [role for role in appbuilder.sm.get_all_roles() if role.name in ["User", "Viewer"]] can_read_on_config_perm = appbuilder.sm.find_permission_view_menu( @@ -48,6 +52,8 @@ def upgrade(): ): appbuilder.sm.del_permission_role(role, can_read_on_config_perm) + log.handlers = handlers + def downgrade(): """Add can_read permission on config resource for User and Viewer role"""