rquote 0.3.6__py3-none-any.whl → 0.3.9__py3-none-any.whl

rquote/__init__.py

@@ -7,6 +7,9 @@ Copyright (c) 2021 Roi ZHAO
 
 '''
 
+import re
+from pathlib import Path
+
 # API函数
 from .api import (
     get_price,
@@ -34,7 +37,38 @@ from . import exceptions
 from .cache import MemoryCache, Cache
 from .utils.http import HTTPClient
 
-__version__ = '0.3.6'
+
+def _get_version():
+    """从 pyproject.toml 读取版本号"""
+    # 优先尝试从已安装的包中读取版本
+    try:
+        from importlib import metadata
+        return metadata.version("rquote")
+    except Exception:
+        pass
+    
+    # 如果包未安装，从 pyproject.toml 读取
+    try:
+        # 获取项目根目录（__init__.py 的父目录的父目录）
+        current_file = Path(__file__).resolve()
+        project_root = current_file.parent.parent
+        pyproject_path = project_root / "pyproject.toml"
+        
+        if pyproject_path.exists():
+            with open(pyproject_path, 'r', encoding='utf-8') as f:
+                content = f.read()
+                # 使用正则表达式匹配 version = "x.x.x"
+                match = re.search(r'version\s*=\s*["\']([^"\']+)["\']', content)
+                if match:
+                    return match.group(1)
+    except Exception:
+        pass
+    
+    # 如果都失败了，返回默认版本
+    return "0.0.0"
+
+
+__version__ = _get_version()
 
 __all__ = [
     # API函数

rquote/api/lists.py

@@ -160,7 +160,7 @@ def get_industry_stocks(node):
     获取指定行业板块的股票列表
     
     Args:
-        node: 行业节点代码
+        node: 行业节点名称
     
     Returns:
         股票列表

rquote/api/price.py

@@ -11,6 +11,33 @@ from ..utils.date import check_date_format
 from ..exceptions import SymbolError
 
 
+def _normalize_dataframe_index(df: pd.DataFrame) -> pd.DataFrame:
+    """
+    统一处理 DataFrame 索引，转换为 DatetimeIndex
+    
+    Args:
+        df: 输入的 DataFrame
+    
+    Returns:
+        处理后的 DataFrame，索引为 DatetimeIndex
+    """
+    if df.empty:
+        return df
+    
+    # 如果已经是 DatetimeIndex，直接返回
+    if isinstance(df.index, pd.DatetimeIndex):
+        return df
+    
+    # 尝试转换为 DatetimeIndex
+    try:
+        df.index = pd.to_datetime(df.index)
+    except (ValueError, TypeError) as e:
+        # 如果转换失败，保持原样（可能是其他类型的索引）
+        pass
+    
+    return df
+
+
 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]:
     '''
@@ -20,6 +47,7 @@ def get_price(i: str, sdate: str = '', edate: str = '', freq: str = 'day',
         i: 股票代码
         sdate: 开始日期
         edate: 结束日期
+        freq: 频率，默认'day' (日线)，可选：'week', 'month', 'min'
         dd: data dictionary或Cache对象，任何有get/put方法的本地缓存
         days: 获取天数，覆盖sdate
         fq: 复权方式，qfq为前复权
@@ -48,7 +76,12 @@ def get_price(i: str, sdate: str = '', edate: str = '', freq: str = 'day',
     market = MarketFactory.create_from_symbol(i, cache=cache)
     
     # 调用市场的get_price方法
-    return market.get_price(i, sdate=sdate, edate=edate, freq=freq, days=days, fq=fq)
+    symbol, name, df = market.get_price(i, sdate=sdate, edate=edate, freq=freq, days=days, fq=fq)
+    
+    # 统一后处理：转换索引为 DatetimeIndex
+    df = _normalize_dataframe_index(df)
+    
+    return symbol, name, df
 
 
 def get_price_longer(i: str, l: int = 2, dd=None) -> Tuple[str, str, pd.DataFrame]:
@@ -64,10 +97,24 @@ def get_price_longer(i: str, l: int = 2, dd=None) -> Tuple[str, str, pd.DataFram
         (symbol, name, DataFrame)
     """
     _, name, a = get_price(i, dd=dd)
-    d1 = a.index.format()[0]
+    # 使用 DatetimeIndex 的格式化方法（get_price 已统一转换为 DatetimeIndex）
+    if isinstance(a.index, pd.DatetimeIndex) and len(a.index) > 0:
+        d1 = a.index[0].strftime('%Y%m%d')
+    else:
+        # 降级处理：如果索引不是 DatetimeIndex（理论上不应该发生），尝试格式化
+        try:
+            d1 = str(a.index[0])[:8] if len(str(a.index[0])) >= 8 else str(a.index[0])
+        except:
+            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()
         d1 = d0
     return i, name, a
 
+
+if __name__ == '__main__':
+    i = 'usTSLA.N'
+    a = get_price(i, freq='min')
+    print(a)

rquote/api/stock_info.py

@@ -18,6 +18,7 @@ def get_stock_concepts(i: str) -> List[str]:
     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:
@@ -38,6 +39,7 @@ def get_stock_industry(i: str) -> List[str]:
     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:

rquote/markets/cn_stock.py

@@ -82,7 +82,10 @@ class CNStockMarket(Market):
             name = data['data']['name']
             df = pd.DataFrame([i.split(',') for i in data['data']['klines']], 
                              columns=['date', 'open', 'close', 'high', 'low', 'vol', 'money', 'p'])
-            df = df.set_index(['date']).astype(float)
+            df = df.set_index(['date'])
+            # 转换数值列
+            for col in ['open', 'close', 'high', 'low', 'vol', 'money', 'p']:
+                df[col] = pd.to_numeric(df[col], errors='coerce')
             
             result = (symbol, name, df)
             self._put_cache(symbol, result)

rquote/markets/future.py

@@ -27,7 +27,10 @@ class FutureMarket(Market):
         
         # 特殊处理BTC
         if symbol[2:5].lower() == 'btc':
-            return self._get_btc_price(symbol)
+            if freq in ('min', '1min', 'minute'):
+                return self._get_btc_minute_price(symbol)
+            else:
+                return self._get_btc_price(symbol)
         
         cache_key = f"{symbol}:{sdate}:{edate}:{freq}:{days}"
         cached = self._get_cached(cache_key)
@@ -49,7 +52,7 @@ class FutureMarket(Market):
             return self._get_price_fallback(symbol, future_code, freq)
     
     def _get_btc_price(self, symbol: str) -> Tuple[str, str, pd.DataFrame]:
-        """获取比特币价格"""
+        """获取比特币日线价格"""
         url = 'https://quotes.sina.cn/fx/api/openapi.php/BtcService.getDayKLine?symbol=btcbtcusd'
         response = hget(url)
         if not response:
@@ -60,12 +63,95 @@ class FutureMarket(Market):
                           columns=['date', 'open', 'high', 'low', 'close', 'vol', 'amount'])
         for col in ['open', 'high', 'low', 'close', 'vol', 'amount']:
             df[col] = pd.to_numeric(df[col], errors='coerce')
-        df = df.set_index('date').astype(float)
+        df = df.set_index('date')
         
         result = (symbol, 'BTC', df)
         self._put_cache(symbol, result)
         return result
     
+    def _get_btc_minute_price(self, symbol: str, datalen: int = 1440) -> Tuple[str, str, pd.DataFrame]:
+        """
+        获取比特币分钟级价格
+        
+        Args:
+            symbol: 股票代码（如 'fuBTC'）
+            datalen: 数据长度，默认1440（24小时，每分钟1条）
+        
+        Returns:
+            (symbol, name, DataFrame)
+        """
+        cache_key = f"{symbol}:min:{datalen}"
+        cached = self._get_cached(cache_key)
+        if cached:
+            return cached
+        
+        url = f'https://quotes.sina.cn/fx/api/openapi.php/BtcService.getMinKline?symbol=btcbtcusd&scale=1&datalen={datalen}&callback=var%20_btcbtcusd'
+        response = hget(url)
+        if not response:
+            raise DataSourceError("Failed to fetch BTC minute data")
+        
+        # 解析 JavaScript callback 格式: var _btcbtcusd({...})
+        text = response.text
+        
+        # 移除开头的注释和脚本标签
+        if '*/' in text:
+            text = text.split('*/', 1)[1]
+        text = text.strip()
+        
+        # 查找 JSON 部分（从第一个 { 开始）
+        json_start = text.find('{')
+        if json_start == -1:
+            raise DataSourceError("Invalid BTC minute data format: no JSON found")
+        
+        # 提取 JSON 部分，需要找到匹配的最后一个 }
+        # 格式: var _btcbtcusd({...}) 或 var _btcbtcusd({...});
+        json_str = text[json_start:]
+        # 移除末尾可能的 ); 或 )
+        json_str = json_str.rstrip(');').rstrip(')')
+        
+        try:
+            data = json.loads(json_str)
+        except json.JSONDecodeError as e:
+            raise DataSourceError(f"Failed to parse BTC minute data JSON: {e}")
+        
+        # 检查返回状态
+        if data.get('result', {}).get('status', {}).get('code') != 0:
+            raise DataSourceError(f"BTC API error: {data.get('result', {}).get('status', {}).get('msg', 'Unknown error')}")
+        
+        # 提取数据
+        kline_data = data.get('result', {}).get('data', [])
+        if not kline_data:
+            raise DataSourceError("No BTC minute data returned")
+        
+        # 转换为 DataFrame
+        # 数据格式: {"d":"2025-11-16 15:35:00","o":"95835.37","h":"95919.90","l":"95835.37","c":"95919.89","v":"6","a":"551441.4297"}
+        records = []
+        for item in kline_data:
+            records.append({
+                'date': item.get('d', ''),
+                'open': item.get('o', '0'),
+                'high': item.get('h', '0'),
+                'low': item.get('l', '0'),
+                'close': item.get('c', '0'),
+                'vol': item.get('v', '0'),
+                'amount': item.get('a', '0')
+            })
+        
+        df = pd.DataFrame(records)
+        if df.empty:
+            raise DataSourceError("Empty BTC minute data")
+        
+        # 转换数据类型
+        for col in ['open', 'high', 'low', 'close', 'vol', 'amount']:
+            df[col] = pd.to_numeric(df[col], errors='coerce')
+        
+        # 设置索引
+        df = df.set_index('date')
+        
+        result = (symbol, 'BTC', df)
+        self._put_cache(cache_key, result)
+        return result
+    
     def _get_price_fallback(self, symbol: str, future_code: str, freq: str) -> Tuple[str, str, pd.DataFrame]:
         """降级方法"""
         from ..utils.helpers import load_js_var_json
@@ -84,7 +170,7 @@ class FutureMarket(Market):
             df.columns = ['date', 'open', 'high', 'low', 'close', 'vol', 'p', 's']
             for col in ['open', 'high', 'low', 'close', 'vol', 'p', 's']:
                 df[col] = pd.to_numeric(df[col], errors='coerce')
-            df = df.set_index('date').astype(float)
+            df = df.set_index('date')
             result = (symbol, future_code, df)
         
         self._put_cache(f"{symbol}:{freq}", result)

rquote/markets/hk_stock.py

@@ -35,7 +35,7 @@ class HKStockMarket(Market):
             )
             
             parser = KlineParser()
-            name, df = parser.parse_tencent_kline(raw_data, symbol)
+            name, df = parser.parse_tencent_kline(raw_data, symbol, fq=fq)
             
             result = (symbol, name, df)
             self._put_cache(cache_key, result)

rquote/parsers/kline.py

@@ -11,7 +11,7 @@ class KlineParser:
     """K线数据解析器"""
     
     @staticmethod
-    def parse_tencent_kline(data: Dict[str, Any], symbol: str) -> Tuple[str, pd.DataFrame]:
+    def parse_tencent_kline(data: Dict[str, Any], symbol: str, fq: str = 'qfq') -> Tuple[str, pd.DataFrame]:
         """
         解析腾讯K线数据
         
@@ -27,9 +27,17 @@ class KlineParser:
             if not symbol_data:
                 raise ParseError(f"No data for symbol {symbol}")
             
-            # 查找时间键
-            time_keys = ['day', 'qfqday', 'hfqday', 'week', 'qfqweek', 'hfqweek',
-                        'month', 'qfqmonth', 'hfqmonth']
+            # 查找时间键，优先使用与fq参数匹配的键
+            # 根据fq参数确定优先级：qfq -> qfqday优先，hfq -> hfqday优先，否则day优先
+            if fq == 'qfq':
+                time_keys = ['qfqday', 'day', 'hfqday', 'qfqweek', 'week', 'hfqweek',
+                            'qfqmonth', 'month', 'hfqmonth']
+            elif fq == 'hfq':
+                time_keys = ['hfqday', 'day', 'qfqday', 'hfqweek', 'week', 'qfqweek',
+                            'hfqmonth', 'month', 'qfqmonth']
+            else:
+                time_keys = ['day', 'qfqday', 'hfqday', 'week', 'qfqweek', 'hfqweek',
+                            'month', 'qfqmonth', 'hfqmonth']
             tk = None
             for tkt in time_keys:
                 if tkt in symbol_data:

{rquote-0.3.6.dist-info → rquote-0.3.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rquote
-Version: 0.3.6
+Version: 0.3.9
 Summary: Mostly day quotes of cn/hk/us/fund/future markets, side with quote list fetch
 Requires-Python: >=3.6.1
 Description-Content-Type: text/markdown
@@ -83,7 +83,7 @@ sid, name, df = get_price('sh000001', dd=cache_dict)
 - `i`: 股票代码，使用新浪/腾讯的id形式
 - `sdate`: 开始日期 (可选，格式：YYYY-MM-DD)
 - `edate`: 结束日期 (可选，格式：YYYY-MM-DD)
-- `freq`: 频率，默认'day' (日线)，可选：'week', 'month', 'min'
+- `freq`: 频率，默认'day' (日线)，可选：(港A)'week', 'month', (美股)'min'
 - `days`: 获取天数，默认320天
 - `fq`: 复权方式，默认'qfq' (前复权)，可选：'hfq' (后复权)
 - `dd`: 本地缓存字典 (可选，已废弃，建议使用MemoryCache)

{rquote-0.3.6.dist-info → rquote-0.3.9.dist-info}/RECORD

@@ -1,12 +1,12 @@
-rquote/__init__.py,sha256=l0zLfBtpfQB401d5EE4GLja3EAWu03LwWXopqFtwMmY,1159
+rquote/__init__.py,sha256=-U5Uq4eT3Hhl2EkVmBgr5TAfU-ZfFzpOaGeJafnhyos,2208
 rquote/config.py,sha256=noep_VzY_nJehnkPQb4mkwzpeYLwkU1riqofQJ6Hhw0,1108
 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=ptizO--im80HaxlzxkJo9BKdJPEnbu00R9UDgcoA0mU,656
-rquote/api/lists.py,sha256=jSvxtlBsGh67ruAq0_ASCqsEXMh6xC9lZEKQWS97UCY,5331
-rquote/api/price.py,sha256=a6ARAX66iPy79Bhxjj8Sn-Gk3p84Ukw0UDywViWTTzA,2092
-rquote/api/stock_info.py,sha256=aQwAhMQi8EymV5uVwOdkOLKYYnKfMRRMm6SqvHVYl-w,1349
+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/tick.py,sha256=nEcjuAjtBHUaD8KPRLg643piVa21PhKDQvkVWNwvvME,1431
 rquote/cache/__init__.py,sha256=IXGSRpvSgBlcM6twLuJEOEockbb09_VqORXdQpfwpCA,138
 rquote/cache/base.py,sha256=orzG4Yo-6gzVG027j1-LTZPT718JohnCdLDnOLoLUQ4,515
@@ -19,20 +19,20 @@ rquote/factors/__init__.py,sha256=_ZbH2XxYtXwCJpvRVdNvGncoPSpMqrtlYmf1_fMGIjM,11
 rquote/factors/technical.py,sha256=dPDs3pDEDRV9iQJBrSoKpGFLQMjOqyoBdN2rUntpOUU,4235
 rquote/markets/__init__.py,sha256=k4F8cZgb-phqemMqhZXFPdOKsR4P--DD3d5i21vKhbg,365
 rquote/markets/base.py,sha256=DjvxRcJqwUsBTxnsE28Gd-zJLFsCGwdQpezLRAZ_9sQ,1347
-rquote/markets/cn_stock.py,sha256=xf-uXDvkdPBPuc-4-VhEsX0OrxSsaMVsCl1wmSqnVl0,8064
+rquote/markets/cn_stock.py,sha256=fyF7jJHFUrI5jwuqBKHXpsIE51H4kbyc3q-uuviPLGk,8224
 rquote/markets/factory.py,sha256=4Txpuok0LBOLT_vAiIU-NslwVnYF7sKHCdlacAboxpo,2875
-rquote/markets/future.py,sha256=AbA-gpmBhbyrRJn8Vu-ouQLCd-TlacXSVZQ2VHYn2pg,3907
-rquote/markets/hk_stock.py,sha256=iz0BriBDNGLkUCPExcCbe2jmuoqJinxx_cbDtKblABc,1445
+rquote/markets/future.py,sha256=7AqViPp0S9OQZsaU2hkJzh4My6gYFqLo1OUW2mVMSDo,7215
+rquote/markets/hk_stock.py,sha256=NlWaXQgXttpcQVFZjflcEkMTmXMxeP2C6Y7OGG50u7E,1452
 rquote/markets/us_stock.py,sha256=17mTg50g3ImOnGM4Re1MRSyvbD2mgFW6wjtMh86IEXA,2465
 rquote/parsers/__init__.py,sha256=q4g-FgpzxKBPfhJiQH3B5MEeZWUIXlyre-vAnOnfYmA,110
-rquote/parsers/kline.py,sha256=O1hieGTebSppcXUl3vS1CNzYftk6f2ElS0a9Tx8mqo4,3481
+rquote/parsers/kline.py,sha256=g6k8W76-4hpYsuBgvwmb5G6ZkzHOJDX-JrVVXYksw4c,4020
 rquote/utils/__init__.py,sha256=-ZHABqFHQeJrCCsgnqEYWR57jl7GduCKn2V3hpFi-pE,348
 rquote/utils/date.py,sha256=nhK3xQ2kFvKhdkPw-2HR2V0PSzBfXmX8L7laG7VmG2E,913
 rquote/utils/helpers.py,sha256=V07n9BtRS8bEJH023Kca78-unk7iD3B9hn2UjELetYs,354
 rquote/utils/http.py,sha256=X0Alhnu0CNqyQeOt6ivUWmh2XwrWxXd2lSpQOKDdnzw,3249
 rquote/utils/logging.py,sha256=cbeRH4ODazn7iyQmGoEBT2lH5LX4Ca3zDfs_20J1T28,566
 rquote/utils/web.py,sha256=I8_pcThW6VUvahuRHdtp32iZwr85hEt1hB6TgznMy_U,3854
-rquote-0.3.6.dist-info/METADATA,sha256=IJ2xLFbl5KeQ_LktJ58cXTvmUC48ua4xzO2q6obGTN4,11248
-rquote-0.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rquote-0.3.6.dist-info/top_level.txt,sha256=CehAiaZx7Fo8HGoV2zd5GhILUW1jQEN8YS-cWMlrK9Y,7
-rquote-0.3.6.dist-info/RECORD,,
+rquote-0.3.9.dist-info/METADATA,sha256=gFYIx3AMMzbJgDx9BlXn-Xn3T9PgkTuYkQTXthyjeIA,11262
+rquote-0.3.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rquote-0.3.9.dist-info/top_level.txt,sha256=CehAiaZx7Fo8HGoV2zd5GhILUW1jQEN8YS-cWMlrK9Y,7
+rquote-0.3.9.dist-info/RECORD,,