rquote 0.4.2__tar.gz → 0.4.4__tar.gz

{rquote-0.4.2 → rquote-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rquote
-Version: 0.4.2
+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.2**
+当前版本：**0.4.4**
 
 ## 主要特性
 
@@ -346,6 +346,50 @@ os.environ['RQUOTE_HTTP_TIMEOUT'] = '20'
 config_from_env = config.Config.from_env()
 ```
 
+### 日志配置
+
+**默认情况下，日志功能是关闭的。** 如果需要启用日志，可以通过环境变量手动开启：
+
+#### 通过环境变量开启日志
+
+```bash
+# 设置日志级别为 INFO（会同时输出到文件和控制台）
+export RQUOTE_LOG_LEVEL=INFO
+
+# 可选：自定义日志文件路径（默认为 /tmp/rquote.log）
+export RQUOTE_LOG_FILE=/path/to/your/logfile.log
+
+# 然后运行你的Python脚本
+python your_script.py
+```
+
+#### 支持的日志级别
+
+- `DEBUG`: 详细的调试信息
+- `INFO`: 一般信息（推荐）
+- `WARNING`: 警告信息
+- `ERROR`: 错误信息
+- `CRITICAL`: 严重错误
+
+#### 在Python代码中开启日志
+
+```python
+import os
+
+# 在导入 rquote 之前设置环境变量
+os.environ['RQUOTE_LOG_LEVEL'] = 'INFO'
+os.environ['RQUOTE_LOG_FILE'] = '/tmp/rquote.log'  # 可选
+
+from rquote import get_price
+
+# 现在日志已启用
+sid, name, df = get_price('sh000001')
+```
+
+#### 关闭日志
+
+如果不设置 `RQUOTE_LOG_LEVEL` 环境变量，或者设置为空值，日志功能将保持关闭状态（默认行为）。
+
 ### 使用改进的HTTP客户端
 
 ```python

{rquote-0.4.2 → rquote-0.4.4}/README.md

@@ -4,7 +4,7 @@
 
 ## 版本信息
 
-当前版本：**0.4.2**
+当前版本：**0.4.4**
 
 ## 主要特性
 
@@ -332,6 +332,50 @@ os.environ['RQUOTE_HTTP_TIMEOUT'] = '20'
 config_from_env = config.Config.from_env()
 ```
 
+### 日志配置
+
+**默认情况下，日志功能是关闭的。** 如果需要启用日志，可以通过环境变量手动开启：
+
+#### 通过环境变量开启日志
+
+```bash
+# 设置日志级别为 INFO（会同时输出到文件和控制台）
+export RQUOTE_LOG_LEVEL=INFO
+
+# 可选：自定义日志文件路径（默认为 /tmp/rquote.log）
+export RQUOTE_LOG_FILE=/path/to/your/logfile.log
+
+# 然后运行你的Python脚本
+python your_script.py
+```
+
+#### 支持的日志级别
+
+- `DEBUG`: 详细的调试信息
+- `INFO`: 一般信息（推荐）
+- `WARNING`: 警告信息
+- `ERROR`: 错误信息
+- `CRITICAL`: 严重错误
+
+#### 在Python代码中开启日志
+
+```python
+import os
+
+# 在导入 rquote 之前设置环境变量
+os.environ['RQUOTE_LOG_LEVEL'] = 'INFO'
+os.environ['RQUOTE_LOG_FILE'] = '/tmp/rquote.log'  # 可选
+
+from rquote import get_price
+
+# 现在日志已启用
+sid, name, df = get_price('sh000001')
+```
+
+#### 关闭日志
+
+如果不设置 `RQUOTE_LOG_LEVEL` 环境变量，或者设置为空值，日志功能将保持关闭状态（默认行为）。
+
 ### 使用改进的HTTP客户端
 
 ```python

{rquote-0.4.2 → rquote-0.4.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "rquote"
-version = "0.4.2"
+version = "0.4.4"
 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.4/rquote/api/stock_info.py

@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+"""
+股票信息相关API
+"""
+import json
+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 _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:
+        概念代码列表
+    """
+    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, dd: Optional[Any] = None) -> List[str]:
+    """
+    获取指定股票所属的行业板块
+    
+    Args:
+        i: 股票代码
+        dd: data dictionary或Cache对象，任何有get/put方法的本地缓存
+    
+    Returns:
+        行业代码列表
+    """
+    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.4/rquote/utils/logging.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+"""
+日志工具
+"""
+import logging
+import os
+
+
+def setup_logger():
+    """设置日志记录器"""
+    logger = logging.getLogger('rquote')
+    if not logger.handlers:
+        # 默认关闭日志，通过环境变量 RQUOTE_LOG_LEVEL 控制
+        log_level = os.getenv('RQUOTE_LOG_LEVEL', '').upper()
+        log_file = os.getenv('RQUOTE_LOG_FILE', '/tmp/rquote.log')
+        
+        # 如果设置了有效的日志级别，则启用日志
+        if log_level in ('DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'):
+            # 将字符串级别转换为logging级别
+            level_map = {
+                'DEBUG': logging.DEBUG,
+                'INFO': logging.INFO,
+                'WARNING': logging.WARNING,
+                'ERROR': logging.ERROR,
+                'CRITICAL': logging.CRITICAL,
+            }
+            logger.setLevel(level_map[log_level])
+            
+            # 添加文件handler
+            file_handler = logging.FileHandler(log_file)
+            formatter = logging.Formatter('%(asctime)-15s:%(lineno)s %(message)s')
+            file_handler.setFormatter(formatter)
+            logger.addHandler(file_handler)
+            
+            # 添加控制台handler
+            logger.addHandler(logging.StreamHandler())
+        else:
+            # 默认关闭日志：设置为CRITICAL级别，不添加handler
+            logger.setLevel(logging.CRITICAL)
+    
+    return logger
+
+
+logger = setup_logger()
+

{rquote-0.4.2 → rquote-0.4.4}/rquote.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rquote
-Version: 0.4.2
+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.2**
+当前版本：**0.4.4**
 
 ## 主要特性
 
@@ -346,6 +346,50 @@ os.environ['RQUOTE_HTTP_TIMEOUT'] = '20'
 config_from_env = config.Config.from_env()
 ```
 
+### 日志配置
+
+**默认情况下，日志功能是关闭的。** 如果需要启用日志，可以通过环境变量手动开启：
+
+#### 通过环境变量开启日志
+
+```bash
+# 设置日志级别为 INFO（会同时输出到文件和控制台）
+export RQUOTE_LOG_LEVEL=INFO
+
+# 可选：自定义日志文件路径（默认为 /tmp/rquote.log）
+export RQUOTE_LOG_FILE=/path/to/your/logfile.log
+
+# 然后运行你的Python脚本
+python your_script.py
+```
+
+#### 支持的日志级别
+
+- `DEBUG`: 详细的调试信息
+- `INFO`: 一般信息（推荐）
+- `WARNING`: 警告信息
+- `ERROR`: 错误信息
+- `CRITICAL`: 严重错误
+
+#### 在Python代码中开启日志
+
+```python
+import os
+
+# 在导入 rquote 之前设置环境变量
+os.environ['RQUOTE_LOG_LEVEL'] = 'INFO'
+os.environ['RQUOTE_LOG_FILE'] = '/tmp/rquote.log'  # 可选
+
+from rquote import get_price
+
+# 现在日志已启用
+sid, name, df = get_price('sh000001')
+```
+
+#### 关闭日志
+
+如果不设置 `RQUOTE_LOG_LEVEL` 环境变量，或者设置为空值，日志功能将保持关闭状态（默认行为）。
+
 ### 使用改进的HTTP客户端
 
 ```python

rquote-0.4.2/rquote/api/stock_info.py

@@ -1,51 +0,0 @@
-# -*- coding: utf-8 -*-
-"""
-股票信息相关API
-"""
-import json
-from typing import List
-from ..utils import hget
-from ..exceptions import HTTPError
-
-
-def get_stock_concepts(i: str) -> List[str]:
-    """
-    获取指定股票所属的概念板块
-    
-    Args:
-        i: 股票代码
-    
-    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', [])
-
-
-def get_stock_industry(i: str) -> List[str]:
-    """
-    获取指定股票所属的行业板块
-    
-    Args:
-        i: 股票代码
-    
-    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', [])
-

rquote-0.4.2/rquote/utils/logging.py

@@ -1,25 +0,0 @@
-# -*- coding: utf-8 -*-
-"""
-日志工具
-"""
-import logging
-
-
-def setup_logger():
-    """设置日志记录器"""
-    logger = logging.getLogger('rquote')
-    if not logger.handlers:
-        logger.setLevel(logging.INFO)
-        file_handler = logging.FileHandler('/tmp/rquote.log')
-        
-        formatter = logging.Formatter('%(asctime)-15s:%(lineno)s %(message)s')
-        file_handler.setFormatter(formatter)
-        
-        logger.addHandler(file_handler)
-        logger.addHandler(logging.StreamHandler())
-    
-    return logger
-
-
-logger = setup_logger()
-