From 86e76aaa5fb25b3ba2cf24de970dda95ece1b67b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= Date: Tue, 29 Aug 2017 09:53:27 +0200 Subject: [PATCH] Use UserClass instead of assigning dict to g.current_user --- pillar/cli/setup.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pillar/cli/setup.py b/pillar/cli/setup.py index f553c937..ada719ce 100644 --- a/pillar/cli/setup.py +++ b/pillar/cli/setup.py @@ -43,11 +43,10 @@ def setup_db(admin_email): # Create a default project by faking a POST request. with current_app.test_request_context(data={'project_name': 'Default Project'}): from flask import g + from pillar.auth import UserClass from pillar.api.projects import routes as proj_routes - g.current_user = {'user_id': user['_id'], - 'groups': user['groups'], - 'roles': set(user['roles'])} + g.current_user = UserClass.construct('', user) proj_routes.create_project(overrides={'url': 'default-project', 'is_private': False})