Merge pull request #30 from rickyking/fix/handle-security-lookup-errors-v2

fix: handle GraphQL errors in security_id_to_symbol gracefully

Author: gboudreau

ws_api/wealthsimple_api.py

@@ -713,10 +713,14 @@ def _activity_add_description(self, act):
     def security_id_to_symbol(self, security_id: str) -> str:
         security_symbol = f"[{security_id}]"
         if self.security_market_data_cache_getter:
-            market_data = self.get_security_market_data(security_id)
-            if isinstance(market_data, dict) and market_data['stock']:
-                stock = market_data['stock']
-                security_symbol = f"{stock['primaryExchange']}:{stock['symbol']}"
+            try:
+                market_data = self.get_security_market_data(security_id)
+                if isinstance(market_data, dict) and market_data.get('stock'):
+                    stock = market_data['stock']
+                    security_symbol = f"{stock['primaryExchange']}:{stock['symbol']}"
+            except WSApiException:
+                # Some securities cannot be looked up (e.g., delisted or special securities)
+                pass
         return security_symbol
 
     def get_etf_details(self, funding_id):