From 354bda0efa0d6af3ca6f74ccbfadc198145ff6c0 Mon Sep 17 00:00:00 2001
From: Untone <anton.rewin@gmail.com>
Date: Thu, 13 Mar 2025 22:21:43 +0300
Subject: [PATCH] drafts-fix

---
 resolvers/draft.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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}