diff --git a/resolvers/draft.py b/resolvers/draft.py index 4f7e0ead..50c8b758 100644 --- a/resolvers/draft.py +++ b/resolvers/draft.py @@ -1,3 +1,4 @@ +from operator import or_ import time from sqlalchemy.sql import and_ @@ -55,7 +56,9 @@ async def load_drafts(_, info): return {"error": "User ID and author ID are required"} with local_session() as session: - drafts = session.query(Draft).filter(Draft.authors.any(Author.id == author_id)).all() + drafts = session.query(Draft).filter(or_( + Draft.authors.any(Author.id == author_id), + Draft.created_by == author_id)).all() return {"drafts": drafts}