From c346481ade66eb96928eed1a3c6f6b6fb528b48e Mon Sep 17 00:00:00 2001 From: Untone Date: Mon, 18 Mar 2024 15:01:10 +0300 Subject: [PATCH] update-after-debug --- resolvers/author.py | 20 +++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) diff --git a/resolvers/author.py b/resolvers/author.py index 80dde1ea..941f9f37 100644 --- a/resolvers/author.py +++ b/resolvers/author.py @@ -23,13 +23,19 @@ from services.logger import root_logger as logger @mutation.field('update_author') @login_required async def update_author(_, info, profile): - user_id = info.context['user_id'] - with local_session() as session: - author = session.query(Author).where(Author.user == user_id).first() - Author.update(author, profile) - session.add(author) - session.commit() - return {'error': None, 'author': author} + user_id = info.context.get('user_id') + if not user_id: + return {'error': 'unauthorized', 'author': None} + try: + with local_session() as session: + author = session.query(Author).where(Author.user == user_id).first() + if author: + Author.update(author, profile) + session.add(author) + session.commit() + return {'error': None, 'author': author} + except Exception as exc: + return {'error': exc, 'author': None} @query.field('get_authors_all')