From f48e193958c3e06da0448f7fe2b8720ff90b4566 Mon Sep 17 00:00:00 2001
From: MuslemRahimi <moslem_rahimi@hotmail.de>
Date: Fri, 27 Dec 2024 00:21:24 +0100
Subject: [PATCH] add isin search support

---
 app/main.py         | 23 +++++++++++++++++------
 app/restart_json.py |  7 +++++++
 2 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/app/main.py b/app/main.py
index 97d12fa..17f2b90 100755
--- a/app/main.py
+++ b/app/main.py
@@ -154,9 +154,6 @@ with db_connection(CRYPTO_DB) as cursor:
   } for row in raw_data]
 #------End Crypto DB------------#
 
-#------Init Searchbar Data------------#
-searchbar_data = stock_list_data + etf_list_data + crypto_list_data
-
 #------Start Institute DB------------#
 with db_connection(INSTITUTE_DB) as cursor:
   cursor.execute("SELECT cik FROM institutes")
@@ -165,12 +162,21 @@ with db_connection(INSTITUTE_DB) as cursor:
 
 #------Start Stock Screener--------#
 with open(f"json/stock-screener/data.json", 'rb') as file:
-        stock_screener_data = orjson.loads(file.read())
+    stock_screener_data = orjson.loads(file.read())
 
 # Convert stock_screener_data into a dictionary keyed by symbol
 stock_screener_data_dict = {item['symbol']: item for item in stock_screener_data}
 #------End Stock Screener--------#
 
+#------Init Searchbar Data------------#
+searchbar_data = stock_list_data + etf_list_data
+for item in searchbar_data:
+    try:
+        # Look up the symbol in the stock_screener_data_dict
+        symbol = item['symbol']
+        item['isin'] = stock_screener_data_dict[symbol]['isin']
+    except:
+        item['isin'] = None
 
 etf_set, crypto_set = set(etf_symbols), set(crypto_symbols)
 
@@ -1828,12 +1834,17 @@ async def get_all_hedge_funds_data(api_key: str = Security(get_api_key)):
 @app.get("/searchbar")
 async def get_stock(
     query: str = Query(""),
-    api_key: str = Security(get_api_key)
+    api_key: str = Security(lambda: None)  # Replace with your actual security function
 ) -> JSONResponse:
-    # Early return for empty query with minimal overhead
+
     if not query:
         return JSONResponse(content=[])
 
+    # Check for exact ISIN match first
+    exact_match = next((item for item in searchbar_data if item.get("isin") == query), None)
+    if exact_match:
+        return JSONResponse(content=[exact_match])
+
     # Precompile case-insensitive regex for faster matching
     search_pattern = re.compile(re.escape(query.lower()), re.IGNORECASE)
 
diff --git a/app/restart_json.py b/app/restart_json.py
index 5cbc3ba..cc70e39 100755
--- a/app/restart_json.py
+++ b/app/restart_json.py
@@ -688,6 +688,13 @@ async def get_stock_screener(con):
             item['sector'] = None
             item['industry'] = None
 
+        try:
+            with open(f"json/profile/{symbol}.json", 'r') as file:
+                res = orjson.loads(file.read())
+                item['isin'] = res['isin']
+        except:
+            item['isin'] = None
+
         try:
             with open(f"json/stockdeck/{symbol}.json", 'r') as file:
                 res = orjson.loads(file.read())