rquote 0.4.8__py3-none-any.whl → 0.4.9__py3-none-any.whl

rquote/api/lists.py

@@ -191,8 +191,11 @@ def get_cnindex_stocks(index_type='hs300'):
     # 指数类型到 TYPE 值的映射
     index_type_map = {
         'hs300': '1',
+        'sz50': '2',
         'zz500': '3',
-        'zz1000': '7'
+        'kc500': '4',
+        'zz1000': '7',
+        'zz2000': '13'
     }
     
     if index_type not in index_type_map:
@@ -202,6 +205,10 @@ def get_cnindex_stocks(index_type='hs300'):
     
     # 构建 URL
     base_url = 'https://datacenter-web.eastmoney.com/api/data/v1/get'
+    page_size = 500
+    all_data = []
+    
+    # 先获取第一页，确定总数
     params = {
         'callback': 'jQuery112308471143523381743_1763517709888',
         'sortColumns': 'SECURITY_CODE',
@@ -232,7 +239,34 @@ def get_cnindex_stocks(index_type='hs300'):
     
     # 返回 result.data 中的数据列表
     if data.get('result') and data['result'].get('data'):
-        return data['result']['data']
+        all_data.extend(data['result']['data'])
+        total = int(index_type[2:])
+        
+        # 如果总数超过 page_size，需要分页获取
+        if total > page_size:
+            total_pages = (total + page_size - 1) // page_size  # 向上取整
+            
+            # 从第二页开始获取
+            for page_num in range(2, total_pages + 1):
+                params['pageNumber'] = str(page_num)
+                url = f'{base_url}?{urlencode(params)}'
+                
+                a = hget(url)
+                if not a:
+                    logger.warning(f'Failed to fetch page {page_num} for {index_type}')
+                    continue
+                
+                json_str = a.text.split('(', 1)[1].rstrip(');')
+                page_data = json.loads(json_str)
+                
+                if page_data.get('result') and page_data['result'].get('data'):
+                    all_data.extend(page_data['result']['data'])
+                
+                # 避免请求过快
+                time.sleep(0.1)
+        
+        logger.debug(f'Fetched {len(all_data)} stocks for {index_type} (total: {total})')
+        return all_data
     else:
         logger.warning(f'No data found in response for {index_type}')
         return []

{rquote-0.4.8.dist-info → rquote-0.4.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rquote
-Version: 0.4.8
+Version: 0.4.9
 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.8**
+当前版本：**0.4.9**
 
 ## 主要特性
 

{rquote-0.4.8.dist-info → rquote-0.4.9.dist-info}/RECORD

@@ -4,7 +4,7 @@ rquote/exceptions.py,sha256=lJH2GC5dDhMoW_OtlBc03wlUn684-7jNPyF1NjmfVIE,569
 rquote/plots.py,sha256=UQn4sjhIzVwagfhUDM738b2HHjKo4tRdU2UCs_1-FbY,2341
 rquote/utils.py,sha256=bH0ZFIo-ZelNztzPS6BXFShXE3yGA9USI_P9INN0Y-s,310
 rquote/api/__init__.py,sha256=17VgPOKsbQX8kKcuU8fKqn4qOHxMamV9GBTXTCcbdvw,706
-rquote/api/lists.py,sha256=WGyliA9pkpPrN019xkhKL-5nzrmTGrsb8bnG8Zz4cRs,7395
+rquote/api/lists.py,sha256=uCOESePsrWGfY8gZE70aqk8xJS8nMqOK1dD8zTPrpi4,8676
 rquote/api/price.py,sha256=I5lZl6cUQRlE4AtzNbR-uGZt1ho9vgP1cgNFDjaigMA,3575
 rquote/api/stock_info.py,sha256=h_AbgsR8CLWz5zA2PtGsS3ROQ3qcw_hnRAtG3USeMos,2988
 rquote/api/tick.py,sha256=nEcjuAjtBHUaD8KPRLg643piVa21PhKDQvkVWNwvvME,1431
@@ -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.8.dist-info/METADATA,sha256=qZGWiccqGzatqJTn5pAgAvzk3OHIj2QqCJAziMuxFQU,14478
-rquote-0.4.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rquote-0.4.8.dist-info/top_level.txt,sha256=CehAiaZx7Fo8HGoV2zd5GhILUW1jQEN8YS-cWMlrK9Y,7
-rquote-0.4.8.dist-info/RECORD,,
+rquote-0.4.9.dist-info/METADATA,sha256=QjWXGAaetxCapZOpA6GZHR-NykNLKSnZRJ6hStkVNwc,14478
+rquote-0.4.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rquote-0.4.9.dist-info/top_level.txt,sha256=CehAiaZx7Fo8HGoV2zd5GhILUW1jQEN8YS-cWMlrK9Y,7
+rquote-0.4.9.dist-info/RECORD,,