rquote 0.4.3__py3-none-any.whl → 0.4.4__py3-none-any.whl

rquote/api/stock_info.py

@@ -3,49 +3,101 @@
 股票信息相关API
 """
 import json
-from typing import List
+from typing import List, Optional, Any
 from ..utils import hget
 from ..exceptions import HTTPError
+from ..cache import Cache
+from ..cache.memory import DictCache as DictCacheAdapter
 
 
-def get_stock_concepts(i: str) -> List[str]:
+def _normalize_stock_code(code: str) -> str:
+    """标准化股票代码"""
+    return {'6': 'sh', '0': 'sz', '3': 'sz'}.get(code[0], '') + code if code[0] in ['6', '0', '3'] else code
+
+
+def _get_cache_adapter(dd: Optional[Any]) -> Optional[Cache]:
+    """获取缓存适配器"""
+    if dd is None:
+        return None
+    if isinstance(dd, dict):
+        return DictCacheAdapter(dd)
+    elif isinstance(dd, Cache):
+        return dd
+    elif hasattr(dd, 'get') and hasattr(dd, 'put'):
+        return DictCacheAdapter(dd)
+    return None
+
+
+def _fetch_stock_plate_data(normalized_code: str, data_key: str, error_msg: str) -> List[str]:
+    """获取股票板块数据的通用函数"""
+    url = f'https://proxy.finance.qq.com/ifzqgtimg/appstock/app/stockinfo/plateNew?code={normalized_code}&app=wzq&zdf=1'
+    a = hget(url)
+    if not a:
+        raise HTTPError(f'Failed to fetch {error_msg} from QQ Finance')
+    data = json.loads(a.text)
+    if data.get('code') != 0:
+        raise HTTPError('API returned error: {}'.format(data.get('msg', 'Unknown error')))
+    return data.get('data', {}).get(data_key, [])
+
+
+def get_stock_concepts(i: str, dd: Optional[Any] = None) -> List[str]:
     """
     获取指定股票所属的概念板块
     
     Args:
         i: 股票代码
+        dd: data dictionary或Cache对象，任何有get/put方法的本地缓存
     
     Returns:
         概念代码列表
     """
-    i = {'6': 'sh', '0': 'sz', '3': 'sz'}.get(i[0], '') + i if i[0] in ['6', '0', '3'] else i
-    url = f'https://proxy.finance.qq.com/ifzqgtimg/appstock/app/stockinfo/plateNew?code={i}&app=wzq&zdf=1'
-    a = hget(url)
-    if not a:
-        raise HTTPError('Failed to fetch concepts from QQ Finance')
-    data = json.loads(a.text)
-    if data.get('code') != 0:
-        raise HTTPError('API returned error: {}'.format(data.get('msg', 'Unknown error')))
-    return data.get('data', {}).get('concept', [])
+    cache = _get_cache_adapter(dd)
+    normalized_code = _normalize_stock_code(i)
+    cache_key = f'stock_concepts:{normalized_code}'
+    
+    # 尝试从缓存获取
+    if cache:
+        cached_result = cache.get(cache_key)
+        if cached_result is not None:
+            return cached_result
+    
+    # 缓存未命中，请求网络
+    result = _fetch_stock_plate_data(normalized_code, 'concept', 'concepts')
+    
+    # 存入缓存
+    if cache:
+        cache.put(cache_key, result)
+    
+    return result
 
 
-def get_stock_industry(i: str) -> List[str]:
+def get_stock_industry(i: str, dd: Optional[Any] = None) -> List[str]:
     """
     获取指定股票所属的行业板块
     
     Args:
         i: 股票代码
+        dd: data dictionary或Cache对象，任何有get/put方法的本地缓存
     
     Returns:
         行业代码列表
     """
-    i = {'6': 'sh', '0': 'sz', '3': 'sz'}.get(i[0], '') + i if i[0] in ['6', '0', '3'] else i
-    url = f'https://proxy.finance.qq.com/ifzqgtimg/appstock/app/stockinfo/plateNew?code={i}&app=wzq&zdf=1'
-    a = hget(url)
-    if not a:
-        raise HTTPError('Failed to fetch industry from QQ Finance')
-    data = json.loads(a.text)
-    if data.get('code') != 0:
-        raise HTTPError('API returned error: {}'.format(data.get('msg', 'Unknown error')))
-    return data.get('data', {}).get('plate', [])
+    cache = _get_cache_adapter(dd)
+    normalized_code = _normalize_stock_code(i)
+    cache_key = f'stock_industry:{normalized_code}'
+    
+    # 尝试从缓存获取
+    if cache:
+        cached_result = cache.get(cache_key)
+        if cached_result is not None:
+            return cached_result
+    
+    # 缓存未命中，请求网络
+    result = _fetch_stock_plate_data(normalized_code, 'plate', 'industry')
+    
+    # 存入缓存
+    if cache:
+        cache.put(cache_key, result)
+    
+    return result
 

{rquote-0.4.3.dist-info → rquote-0.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rquote
-Version: 0.4.3
+Version: 0.4.4
 Summary: Mostly day quotes of cn/hk/us/fund/future markets, side with quote list fetch
 Requires-Python: >=3.9.0
 Description-Content-Type: text/markdown
@@ -18,7 +18,7 @@ Requires-Dist: duckdb>=0.9.0; extra == "persistent"
 
 ## 版本信息
 
-当前版本：**0.4.3**
+当前版本：**0.4.4**
 
 ## 主要特性
 

{rquote-0.4.3.dist-info → rquote-0.4.4.dist-info}/RECORD

@@ -6,7 +6,7 @@ rquote/utils.py,sha256=bH0ZFIo-ZelNztzPS6BXFShXE3yGA9USI_P9INN0Y-s,310
 rquote/api/__init__.py,sha256=ptizO--im80HaxlzxkJo9BKdJPEnbu00R9UDgcoA0mU,656
 rquote/api/lists.py,sha256=fRebS02Fi0qe6KpWBA-9W1UG0It6__DmRlNimtMa7L8,5331
 rquote/api/price.py,sha256=I5lZl6cUQRlE4AtzNbR-uGZt1ho9vgP1cgNFDjaigMA,3575
-rquote/api/stock_info.py,sha256=912ICdIBr8z2lKWDbq3gG0E94czTPvbx9aXsKUi-QkE,1537
+rquote/api/stock_info.py,sha256=h_AbgsR8CLWz5zA2PtGsS3ROQ3qcw_hnRAtG3USeMos,2988
 rquote/api/tick.py,sha256=nEcjuAjtBHUaD8KPRLg643piVa21PhKDQvkVWNwvvME,1431
 rquote/cache/__init__.py,sha256=S393I5Wmp0QooaRka9n7bvDUdEbg3jUhm6u815T86rM,317
 rquote/cache/base.py,sha256=orzG4Yo-6gzVG027j1-LTZPT718JohnCdLDnOLoLUQ4,515
@@ -33,7 +33,7 @@ rquote/utils/helpers.py,sha256=V07n9BtRS8bEJH023Kca78-unk7iD3B9hn2UjELetYs,354
 rquote/utils/http.py,sha256=X0Alhnu0CNqyQeOt6ivUWmh2XwrWxXd2lSpQOKDdnzw,3249
 rquote/utils/logging.py,sha256=fs2YF1Srux4LLTdk_Grjm5g1f4mzewI38VVSAI82goA,1471
 rquote/utils/web.py,sha256=I8_pcThW6VUvahuRHdtp32iZwr85hEt1hB6TgznMy_U,3854
-rquote-0.4.3.dist-info/METADATA,sha256=R9zS71G3lxp8C43qrY67DBn2Y2t_1P7rTtCztfj7Kq8,14344
-rquote-0.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rquote-0.4.3.dist-info/top_level.txt,sha256=CehAiaZx7Fo8HGoV2zd5GhILUW1jQEN8YS-cWMlrK9Y,7
-rquote-0.4.3.dist-info/RECORD,,
+rquote-0.4.4.dist-info/METADATA,sha256=19kwtEiSyrMll1Ju8Ei1nR4CHCohDhht02hLXsHL47k,14344
+rquote-0.4.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rquote-0.4.4.dist-info/top_level.txt,sha256=CehAiaZx7Fo8HGoV2zd5GhILUW1jQEN8YS-cWMlrK9Y,7
+rquote-0.4.4.dist-info/RECORD,,