diff --git a/plugins/unity_integration.py b/plugins/unity_integration.py index bc74decc5937a564b256fc82437983217e6bcdce..97d942fa0f22f7a20fc95c6b6c8b8f5b71f6fcdf 100644 --- a/plugins/unity_integration.py +++ b/plugins/unity_integration.py @@ -87,6 +87,10 @@ class UnityIntegrationView(AppBuilderBaseView): else: log.error("User creation unsuccessful.") abort(500) + else: + # set role permissions of that user, if it already exists (just take the role from SSO and assign it) + fab_user.role = sec_manager.find_role(role) + sec_manager.update_user(fab_user) # login as that user login_user(fab_user, remember=False) return redirect(url_for("Airflow.index"))