From 6243c27390cacfefe586a26e0143930d0f555b27 Mon Sep 17 00:00:00 2001
From: Untone <anton.rewin@gmail.com>
Date: Mon, 11 Mar 2024 11:56:14 +0300
Subject: [PATCH] handle-no-userid

---
 resolvers/follower.py | 56 ++++++++++++++++++++++++++-----------------
 1 file changed, 34 insertions(+), 22 deletions(-)

diff --git a/resolvers/follower.py b/resolvers/follower.py
index 1fdea8cf..639346a0 100644
--- a/resolvers/follower.py
+++ b/resolvers/follower.py
@@ -77,6 +77,8 @@ async def unfollow(_, info, what, slug):
     follows = None
     try:
         user_id = info.context.get('user_id')
+        if not user_id:
+            return {"error": "unauthorized"}
         follower_query = select(Author).filter(Author.user == user_id)
         [follower] = get_with_stat(follower_query)
         if follower:
@@ -111,28 +113,38 @@ async def unfollow(_, info, what, slug):
 
 
 async def get_follows_by_user_id(user_id: str):
-    if user_id:
-        author = await redis.execute('GET', f'user:{user_id}:author')
-        follows = DEFAULT_FOLLOWS
-        day_old = int(time.time()) - author.get('last_seen', 0) > 24 * 60 * 60
-        if day_old:
-            author_id = json.loads(str(author)).get('id')
-            if author_id:
-                topics = author_follows_topics(author_id)
-                authors = author_follows_authors(author_id)
-                follows = {
-                    'topics': topics,
-                    'authors': authors,
-                    'communities': [
-                        {'id': 1, 'name': 'Дискурс', 'slug': 'discours', 'pic': ''}
-                    ],
-                }
-        else:
-            logger.debug(f'getting follows for {user_id} from redis')
-            res = await redis.execute('GET', f'user:{user_id}:follows')
-            if isinstance(res, str):
-                follows = json.loads(res)
-        return follows
+    if not user_id:
+        return {"error": "unauthorized"}
+    author = await redis.execute('GET', f'user:{user_id}:author')
+    if isinstance(author, str):
+        author = json.loads(author)
+    if not author:
+        with local_session() as session:
+            author = session.query(Author).filter(Author.user == user_id).first()
+            if not author:
+                return {"error": "cant find author"}
+            author = author.dict()
+    last_seen = author.get('last_seen', 0) if isinstance(author, dict) else 0
+    follows = DEFAULT_FOLLOWS
+    day_old = int(time.time()) - last_seen > 24 * 60 * 60
+    if day_old:
+        author_id = json.loads(str(author)).get('id')
+        if author_id:
+            topics = author_follows_topics(author_id)
+            authors = author_follows_authors(author_id)
+            follows = {
+                'topics': topics,
+                'authors': authors,
+                'communities': [
+                    {'id': 1, 'name': 'Дискурс', 'slug': 'discours', 'pic': ''}
+                ],
+            }
+    else:
+        logger.debug(f'getting follows for {user_id} from redis')
+        res = await redis.execute('GET', f'user:{user_id}:follows')
+        if isinstance(res, str):
+            follows = json.loads(res)
+    return follows
 
 
 def reactions_follow(author_id, shout_id, auto=False):