precommit-setup

resolvers/search.py

@@ -1,4 +1,3 @@
-import json
 import time
 from typing import Any, Dict, List, Union
 
@@ -9,67 +8,68 @@ from services.rediscache import redis
 from services.schema import query
 
 
-@query.field("search_recipients")
+@query.field('search_recipients')
 @login_required
 async def search_recipients(_, info, text: str, limit: int = 50, offset: int = 0):
-    result = set([])
+    result = set()
 
     # TODO: maybe redis scan?
 
-    author_id = info.context["author_id"]
+    author_id = info.context['author_id']
 
-    existed_chats = await redis.execute("SMEMBERS", f"/chats_by_author/{author_id}")
+    existed_chats = await redis.execute('SMEMBERS', f'/chats_by_author/{author_id}')
     if isinstance(existed_chats, set):
         chats_list = list(existed_chats)
         for chat_id in chats_list[offset : (offset + limit)]:
-            members_ids = await redis.execute("SMEMBERS", f"/chats/{chat_id}/members")
+            members_ids = await redis.execute('SMEMBERS', f'/chats/{chat_id}/members')
             if isinstance(members_ids, set):
                 for member_id in members_ids:
                     author = CacheStorage.authors_by_id.get(str(member_id))
                     if author:
-                        if author["name"].startswith(text):
+                        if author['name'].startswith(text):
                             result.add(author)
 
     more_amount = limit - len(result)
     if more_amount > 0:
         result.update(CacheStorage.authors[0:more_amount])
-    return {"members": list(result), "error": None}
+    return {'members': list(result), 'error': None}
 
 
-@query.field("search_messages")
+@query.field('search_messages')
 @login_required
 async def search_messages(
     _, info, by: Dict[str, Union[str, int]], limit: int, offset: int
 ) -> Dict[str, Union[List[Dict[str, Any]], None]]:
-    messages_set = set([])
-    author_id = info.context["author_id"]
-    lookup_chats = await redis.execute("SMEMBERS", f"chats_by_author/{author_id}")
+    messages_set = set()
+    author_id = info.context['author_id']
+    lookup_chats = await redis.execute('SMEMBERS', f'chats_by_author/{author_id}')
     if isinstance(lookup_chats, set):
-
         # pre-filter lookup chats
-        by_member = by.get("author")
+        by_member = by.get('author')
         if by_member:
             lookup_chats = filter(
-                lambda ca: by_member in ca["members"],
+                lambda ca: by_member in ca['members'],
                 list(lookup_chats),
             )
 
         # load the messages from lookup chats
         for c in lookup_chats:
             chat_id = c.decode()
-            filter_method = None
+            fltr = None
+            now = int(time.time())
             if by_member:
-                filter_method = lambda mx: mx and mx["created_by"] == by_member
-            body_like = by.get("body") or ""
+                fltr = lambda mx: mx and mx['created_by'] == by_member  # noqa E731
+            body_like = by.get('body') or ''
             if isinstance(body_like, str):
-                filter_method = lambda mx: mx and body_like in mx["body"]
-            days_ago = int(by.get("days") or "0")
+                fltr = lambda mx: mx and body_like in mx['body']  # noqa E731
+            days_ago = int(by.get('days') or '0')
             if days_ago:
-                filter_method = lambda mx: mx and (int(time.time()) - mx["created_by"] < days_ago * 24 * 60 * 60)
-            if filter_method:
+                ts = days_ago * 24 * 60 * 60
+                fltr = lambda mx: mx and now - mx['created_by'] < ts  # noqa E731
+            if fltr:
                 mmm = await load_messages(chat_id, limit, offset)
                 if isinstance(mmm, list):
-                    mmm = list(filter(filter_method, mmm))
+                    mmm = list(filter(fltr, mmm))
                     messages_set |= set(mmm)
 
-    return {"messages": sorted(list(messages_set)), "error": None}
+    return {'messages': sorted(messages_set), 'error': None}