diff --git a/resolvers/reader.py b/resolvers/reader.py index 2aec7723..be839349 100644 --- a/resolvers/reader.py +++ b/resolvers/reader.py @@ -230,8 +230,8 @@ async def load_shouts_by(_, _info, options): return shouts -@query.field('load_shouts_drafts') @login_required +@query.field('load_shouts_drafts') async def load_shouts_drafts(_, info): user_id = info.context.get('user_id') shouts = [] diff --git a/services/auth.py b/services/auth.py index cc7686b0..e01ccccb 100644 --- a/services/auth.py +++ b/services/auth.py @@ -83,15 +83,14 @@ def login_required(f): @wraps(f) async def decorated_function(*args, **kwargs): info = args[1] - context = info.context - req = context.get('request') + req = info.context.get('request') authorized = await check_auth(req) if authorized: user_id, user_roles = authorized if user_id and user_roles: logger.info(f' got {user_id} roles: {user_roles}') - context['user_id'] = user_id.strip() - context['roles'] = user_roles + info.context['user_id'] = user_id.strip() + info.context['roles'] = user_roles return await f(*args, **kwargs) return decorated_function