fix bug

microsoft · May 17, 2024 · 4c24ec2 · 4c24ec2
1 parent 6f765db
commit 4c24ec2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/data_collector/utils.py b/scripts/data_collector/utils.py
@@ -204,6 +204,9 @@ def _get_symbol():
         if len(_symbols) < 3900:
             raise ValueError("request error")
 
+        _symbols = [_symbol + '.ss' if _symbol.startswith('6') else _symbol + '.sz' if _symbol.startswith(('0', '3')) else None for _symbol in _symbols]
+        _symbols = [_symbol for _symbol in _symbols if _symbol is not None]
+
         return set(_symbols)
 
     if _HS_SYMBOLS is None: