search-result-schema-fix

services/search.py

@@ -1,21 +1,21 @@
 import asyncio
 import json
+import logging
 from typing import List
 
 import aiohttp
 
-from orm.shout import Shout
-from services.rediscache import redis
+from orm.shout import Shout  # Adjust the import as needed
+from services.rediscache import redis  # Adjust the import as needed
 
 
 class SearchService:
     lock = asyncio.Lock()
-    cache = {}
 
     @staticmethod
     async def init(session):
         async with SearchService.lock:
-            print("[services.search] did nothing")
+            logging.info("[services.search] Initializing SearchService")
             SearchService.cache = {}
 
     @staticmethod
@@ -26,12 +26,15 @@ class SearchService:
                 # Use aiohttp to send a request to ElasticSearch
                 async with aiohttp.ClientSession() as session:
                     search_url = f"https://search.discours.io/search?q={text}"
-                    async with session.get(search_url) as response:
-                        if response.status == 200:
-                            payload = await response.json()
-                            await redis.execute("SET", text, json.dumps(payload))
-                            return payload[offset : offset + limit]
-                        else:
-                            print(f"[services.search] response: {response.status} {response.text}")
+                    try:
+                        async with session.get(search_url) as response:
+                            if response.status == 200:
+                                payload = await response.json()
+                                await redis.execute("SET", text, json.dumps(payload))  # use redis as cache
+                                return payload[offset : offset + limit]
+                            else:
+                                logging.error(f"[services.search] response: {response.status}  {await response.text()}")
+                    except Exception as e:
+                        logging.error(f"[services.search] error: {e}")
         else:
             return json.loads(cached)[offset : offset + limit]