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 5e85ee4a6ba82f..85d0872704570c 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"""