rquote 0.4.8__tar.gz → 0.5.0__tar.gz

{rquote-0.4.8 → rquote-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rquote
-Version: 0.4.8
+Version: 0.5.0
 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.5.0**
 
 ## 主要特性
 
@@ -172,15 +172,18 @@ sid, nm, df = get_price('fuM2601', freq='min')
 | 2024-02-06 | 2680.48 | 2789.49 | 2802.93 | 2669.67 | 502849313  |
 | 2024-02-07 | 2791.51 | 2829.70 | 2829.70 | 2770.53 | 547117439  |
 
-#### `get_price_longer(i, l=2, dd={})`
+#### `get_price_longer(i, l=2, edate='', freq='day', fq='qfq', dd=None)`
 
-获取更长时间的历史数据，默认获取2年数据
+获取更长时间的历史数据，默认获取2年数据，并且参数与 `get_price` 保持一致
 
 ```python
 from rquote import get_price_longer
 
-# 获取3年的历史数据
+# 获取3年的历史数据（默认日线、前复权，以最新交易日为结束）
 sid, nm, df = get_price_longer('sh000001', l=3)
+
+# 指定结束日期与频率（例如获取到 2024-02-01 的周线数据）
+sid, nm, df = get_price_longer('sh000001', l=3, edate='2024-02-01', freq='week', fq='qfq')
 ```
 
 ### 股票列表获取
@@ -238,8 +241,11 @@ zz1000_stocks = get_cnindex_stocks('zz1000')
 
 支持的指数类型：
 - `'hs300'`: 沪深300
+- `'sz50'`: 上证50
 - `'zz500'`: 中证500
+- `'kc500'`: 科创500
 - `'zz1000'`: 中证1000
+- `'zz2000'`: 中证2000
 
 ### 基金和期货
 

{rquote-0.4.8 → rquote-0.5.0}/README.md

@@ -4,7 +4,7 @@
 
 ## 版本信息
 
-当前版本：**0.4.8**
+当前版本：**0.5.0**
 
 ## 主要特性
 
@@ -158,15 +158,18 @@ sid, nm, df = get_price('fuM2601', freq='min')
 | 2024-02-06 | 2680.48 | 2789.49 | 2802.93 | 2669.67 | 502849313  |
 | 2024-02-07 | 2791.51 | 2829.70 | 2829.70 | 2770.53 | 547117439  |
 
-#### `get_price_longer(i, l=2, dd={})`
+#### `get_price_longer(i, l=2, edate='', freq='day', fq='qfq', dd=None)`
 
-获取更长时间的历史数据，默认获取2年数据
+获取更长时间的历史数据，默认获取2年数据，并且参数与 `get_price` 保持一致
 
 ```python
 from rquote import get_price_longer
 
-# 获取3年的历史数据
+# 获取3年的历史数据（默认日线、前复权，以最新交易日为结束）
 sid, nm, df = get_price_longer('sh000001', l=3)
+
+# 指定结束日期与频率（例如获取到 2024-02-01 的周线数据）
+sid, nm, df = get_price_longer('sh000001', l=3, edate='2024-02-01', freq='week', fq='qfq')
 ```
 
 ### 股票列表获取
@@ -224,8 +227,11 @@ zz1000_stocks = get_cnindex_stocks('zz1000')
 
 支持的指数类型：
 - `'hs300'`: 沪深300
+- `'sz50'`: 上证50
 - `'zz500'`: 中证500
+- `'kc500'`: 科创500
 - `'zz1000'`: 中证1000
+- `'zz2000'`: 中证2000
 
 ### 基金和期货
 

{rquote-0.4.8 → rquote-0.5.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "rquote"
-version = "0.4.8"
+version = "0.5.0"
 description = "Mostly day quotes of cn/hk/us/fund/future markets, side with quote list fetch"
 readme = "README.md"
 # requires-python = ">=3.6.1" # duckdb requires higher python version

{rquote-0.4.8 → rquote-0.5.0}/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 → rquote-0.5.0}/rquote/api/price.py

@@ -38,7 +38,7 @@ def _normalize_dataframe_index(df: pd.DataFrame) -> pd.DataFrame:
     return df
 
 
-def get_price(i: str, sdate: str = '', edate: str = '', freq: str = 'day', 
+def get_price(i: str, sdate: str = '', edate: str = '', freq: str = 'day',
               days: int = 320, fq: str = 'qfq', dd=None) -> Tuple[str, str, pd.DataFrame]:
     '''
     获取价格数据
@@ -84,19 +84,24 @@ def get_price(i: str, sdate: str = '', edate: str = '', freq: str = 'day',
     return symbol, name, df
 
 
-def get_price_longer(i: str, l: int = 2, dd=None) -> Tuple[str, str, pd.DataFrame]:
+def get_price_longer(i: str, l: int = 2, edate: str = '', freq: str = 'day',
+                     fq: str = 'qfq', dd=None) -> Tuple[str, str, pd.DataFrame]:
     """
     获取更长时间的历史数据
     
     Args:
         i: 股票代码
         l: 年数
+        edate: 结束日期，同 ``get_price``
+        freq: 频率，同 ``get_price``
+        fq: 复权方式，同 ``get_price``
         dd: 缓存对象
     
     Returns:
         (symbol, name, DataFrame)
     """
-    _, name, a = get_price(i, dd=dd)
+    # 首段数据直接复用 get_price，透传 edate/freq/fq 参数
+    _, name, a = get_price(i, edate=edate, freq=freq, fq=fq, dd=dd)
     # 使用 DatetimeIndex 的格式化方法（get_price 已统一转换为 DatetimeIndex）
     if isinstance(a.index, pd.DatetimeIndex) and len(a.index) > 0:
         d1 = a.index[0].strftime('%Y%m%d')
@@ -104,12 +109,16 @@ def get_price_longer(i: str, l: int = 2, dd=None) -> Tuple[str, str, pd.DataFram
         # 降级处理：如果索引不是 DatetimeIndex（理论上不应该发生），尝试格式化
         try:
             d1 = str(a.index[0])[:8] if len(str(a.index[0])) >= 8 else str(a.index[0])
-        except:
+        except Exception:
             d1 = a.index.format()[0] if hasattr(a.index, 'format') else str(a.index[0])
     
     for y in range(1, l):
         d0 = str(int(d1[:4]) - 1) + d1[4:]
-        a = pd.concat((get_price(i, d0, d1, dd=dd)[2], a), 0).drop_duplicates()
+        # 逐年向前补数据，保持与 get_price 相同的 freq/fq 配置
+        a = pd.concat(
+            (get_price(i, d0, d1, freq=freq, fq=fq, dd=dd)[2], a),
+            0
+        ).drop_duplicates()
         d1 = d0
     return i, name, a
 

{rquote-0.4.8 → rquote-0.5.0}/rquote.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rquote
-Version: 0.4.8
+Version: 0.5.0
 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.5.0**
 
 ## 主要特性
 
@@ -172,15 +172,18 @@ sid, nm, df = get_price('fuM2601', freq='min')
 | 2024-02-06 | 2680.48 | 2789.49 | 2802.93 | 2669.67 | 502849313  |
 | 2024-02-07 | 2791.51 | 2829.70 | 2829.70 | 2770.53 | 547117439  |
 
-#### `get_price_longer(i, l=2, dd={})`
+#### `get_price_longer(i, l=2, edate='', freq='day', fq='qfq', dd=None)`
 
-获取更长时间的历史数据，默认获取2年数据
+获取更长时间的历史数据，默认获取2年数据，并且参数与 `get_price` 保持一致
 
 ```python
 from rquote import get_price_longer
 
-# 获取3年的历史数据
+# 获取3年的历史数据（默认日线、前复权，以最新交易日为结束）
 sid, nm, df = get_price_longer('sh000001', l=3)
+
+# 指定结束日期与频率（例如获取到 2024-02-01 的周线数据）
+sid, nm, df = get_price_longer('sh000001', l=3, edate='2024-02-01', freq='week', fq='qfq')
 ```
 
 ### 股票列表获取
@@ -238,8 +241,11 @@ zz1000_stocks = get_cnindex_stocks('zz1000')
 
 支持的指数类型：
 - `'hs300'`: 沪深300
+- `'sz50'`: 上证50
 - `'zz500'`: 中证500
+- `'kc500'`: 科创500
 - `'zz1000'`: 中证1000
+- `'zz2000'`: 中证2000
 
 ### 基金和期货