hqdata 0.1.2__tar.gz

hqdata-0.1.2/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 HonestQuant
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

hqdata-0.1.2/PKG-INFO

@@ -0,0 +1,106 @@
+Metadata-Version: 2.4
+Name: hqdata
+Version: 0.1.2
+Summary: A股历史与实时行情数据统一接入、清洗与存储
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pandas>=2.3.3
+Requires-Dist: python-dotenv>=1.0.0
+Provides-Extra: tushare
+Requires-Dist: tushare>=1.4.29; extra == "tushare"
+Provides-Extra: ricequant
+Requires-Dist: rqdatac>=3.1.4; extra == "ricequant"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0; extra == "dev"
+Dynamic: license-file
+
+# hqdata - A股历史与实时行情数据统一接入、清洗与存储
+
+<p align="center">
+    <img src ="https://img.shields.io/pypi/v/hqdata.svg"/>
+    <img src ="https://img.shields.io/pypi/pyversions/hqdata.svg"/>
+</p>
+
+## 定位
+
+`hqdata` 是 HonestQuant 量化系统的**数据基础层**，职责边界清晰：
+
+- 对下：封装各数据源 SDK，屏蔽接口差异
+- 对上：提供统一的查询接口
+- 上层策略和引擎**只调用 `hqdata.api`**，不直接接触任何数据源
+
+## 支持的数据源
+
+| 数据源      | 状态   | 说明                                        |
+| ----------- | ------ | ------------------------------------------- |
+| **AKShare** | 计划中 | 免费，实时数据                              |
+| **Tushare** | 已接入 | 需满足账户积分要求，支持股票&指数的历史日线 |
+| **米筐**    | 接入中 | 需license，支持Tick                         |
+| **迅投**    | 计划中 | 需迅投终端                                  |
+| **iTick**   | 计划中 | 需注册                                      |
+
+## 安装
+
+### 方式一：通过pip安装
+
+```bash
+# 基础安装（仅包含核心功能）
+pip install hqdata
+
+# 按需安装数据源依赖
+pip install hqdata[tushare]      # Tushare 支持
+pip install hqdata[ricequant]    # 米筐支持
+pip install hqdata[tushare,ricequant]  # 同时安装两者
+
+# 复制模板并自行填入所需字段
+cp .env.example .env # 放在你运行 Python 代码的当前目录（优先）/包安装目录
+```
+
+### 方式二：本地开发
+
+```bash
+# 创建虚拟环境
+python -m venv .venv
+source .venv/bin/activate
+
+# 安装依赖 (editable 模式，改代码直接生效)
+pip install -e .
+
+# 复制模板并自行填入所需字段
+cp .env.example .env # 放在你运行 Python 代码的当前目录（优先）/包安装目录
+```
+
+## 使用
+
+以Tushare数据为例
+
+```python
+from hqdata import init_source, get_bar
+
+# 初始化 Tushare (日线数据)
+init_source("tushare")
+
+# 获取日线数据 (symbol 格式: "代码.交易所")
+df = get_bar("600000.SH", frequency="1day", start_date="20260401", end_date="20260402")
+print(df.head())
+```
+
+## 输入参数格式说明
+
+### 股票代码
+
+symbol 参数统一使用 `代码.交易所` 格式：
+
+| 交易所 | 代码 | 示例        |
+| ------ | ---- | ----------- |
+| 上交所 | SH   | `600000.SH` |
+| 深交所 | SZ   | `000001.SZ` |
+
+## 测试
+
+```bash
+pytest tests/ -v
+pytest tests/test_tushare.py::TestTushareIntegration::test_get_bar  # 运行单个测试
+```

hqdata-0.1.2/README.md

@@ -0,0 +1,88 @@
+# hqdata - A股历史与实时行情数据统一接入、清洗与存储
+
+<p align="center">
+    <img src ="https://img.shields.io/pypi/v/hqdata.svg"/>
+    <img src ="https://img.shields.io/pypi/pyversions/hqdata.svg"/>
+</p>
+
+## 定位
+
+`hqdata` 是 HonestQuant 量化系统的**数据基础层**，职责边界清晰：
+
+- 对下：封装各数据源 SDK，屏蔽接口差异
+- 对上：提供统一的查询接口
+- 上层策略和引擎**只调用 `hqdata.api`**，不直接接触任何数据源
+
+## 支持的数据源
+
+| 数据源      | 状态   | 说明                                        |
+| ----------- | ------ | ------------------------------------------- |
+| **AKShare** | 计划中 | 免费，实时数据                              |
+| **Tushare** | 已接入 | 需满足账户积分要求，支持股票&指数的历史日线 |
+| **米筐**    | 接入中 | 需license，支持Tick                         |
+| **迅投**    | 计划中 | 需迅投终端                                  |
+| **iTick**   | 计划中 | 需注册                                      |
+
+## 安装
+
+### 方式一：通过pip安装
+
+```bash
+# 基础安装（仅包含核心功能）
+pip install hqdata
+
+# 按需安装数据源依赖
+pip install hqdata[tushare]      # Tushare 支持
+pip install hqdata[ricequant]    # 米筐支持
+pip install hqdata[tushare,ricequant]  # 同时安装两者
+
+# 复制模板并自行填入所需字段
+cp .env.example .env # 放在你运行 Python 代码的当前目录（优先）/包安装目录
+```
+
+### 方式二：本地开发
+
+```bash
+# 创建虚拟环境
+python -m venv .venv
+source .venv/bin/activate
+
+# 安装依赖 (editable 模式，改代码直接生效)
+pip install -e .
+
+# 复制模板并自行填入所需字段
+cp .env.example .env # 放在你运行 Python 代码的当前目录（优先）/包安装目录
+```
+
+## 使用
+
+以Tushare数据为例
+
+```python
+from hqdata import init_source, get_bar
+
+# 初始化 Tushare (日线数据)
+init_source("tushare")
+
+# 获取日线数据 (symbol 格式: "代码.交易所")
+df = get_bar("600000.SH", frequency="1day", start_date="20260401", end_date="20260402")
+print(df.head())
+```
+
+## 输入参数格式说明
+
+### 股票代码
+
+symbol 参数统一使用 `代码.交易所` 格式：
+
+| 交易所 | 代码 | 示例        |
+| ------ | ---- | ----------- |
+| 上交所 | SH   | `600000.SH` |
+| 深交所 | SZ   | `000001.SZ` |
+
+## 测试
+
+```bash
+pytest tests/ -v
+pytest tests/test_tushare.py::TestTushareIntegration::test_get_bar  # 运行单个测试
+```

hqdata-0.1.2/hqdata/__init__.py

@@ -0,0 +1,6 @@
+"""hqdata - A股历史与实时行情数据统一接入、清洗与存储"""
+
+from hqdata.config import *  # Load .env on import
+from hqdata.api import init_source, get_bar
+
+__all__ = ["init_source", "get_bar"]

hqdata-0.1.2/hqdata/api.py

@@ -0,0 +1,91 @@
+"""hqdata public API - only entry point for upper layers"""
+
+from datetime import date
+from typing import Literal, Optional
+import pandas as pd
+
+# Singleton source instance
+_source: Optional["BaseSource"] = None
+
+
+def init_source(source_type: Literal["ricequant", "tushare"], **kwargs) -> None:
+    """Initialize data source.
+
+    Args:
+        source_type: "ricequant" or "tushare"
+        **kwargs: Source-specific credentials
+    """
+    global _source
+    if source_type == "ricequant":
+        from hqdata.sources.ricequant import RicequantSource
+
+        _source = RicequantSource(**kwargs)
+    elif source_type == "tushare":
+        from hqdata.sources.tushare import TushareSource
+
+        _source = TushareSource(**kwargs)
+    else:
+        raise ValueError(f"Unknown source type: {source_type}")
+
+
+def get_stock_list(list_status: str = "L") -> pd.DataFrame:
+    """Get basic info for stocks.
+
+    Args:
+        list_status: Listing status — "L" (listed), "D" (delisted), "P" (suspended)
+
+    Returns:
+        DataFrame with columns: symbol, name, industry, market, exchange,
+        curr_type, list_status, list_date, delist_date, is_hs
+    """
+    if _source is None:
+        raise RuntimeError("Data source not initialized. Call init_source() first.")
+    return _source.get_stock_list(list_status)
+
+
+def get_bar(
+    symbol: str,
+    frequency: str = "1day",
+    start_date: Optional[str] = None,
+    end_date: Optional[str] = None,
+) -> pd.DataFrame:
+    """Get K-line/bar data for a stock.
+
+    Args:
+        symbol: Stock symbol with exchange (e.g., "600000.SH" or "000001.SZ")
+        frequency: Bar frequency ("1day" | "1min" | "5min" | "15min" | "30min" | "60min" | "1week" | "1month")
+        start_date: Start date in YYYYMMDD format (defaults to today)
+        end_date: End date in YYYYMMDD format (defaults to today)
+
+    Returns:
+        DataFrame with columns: symbol, date, open, high, low, close, pre_close, change, pct_change, volume, amount
+    """
+    if _source is None:
+        raise RuntimeError("Data source not initialized. Call init_source() first.")
+    today = date.today().strftime("%Y%m%d")
+    start_date = start_date or today
+    end_date = end_date or today
+    return _source.get_bar(symbol, frequency, start_date, end_date)
+
+
+def get_index_bar(
+    symbol: str,
+    start_date: Optional[str] = None,
+    end_date: Optional[str] = None,
+) -> pd.DataFrame:
+    """Get daily bar data for an index.
+
+    Args:
+        symbol: Index code with exchange (e.g., "000001.SH", "399300.SZ")
+        start_date: Start date in YYYYMMDD format (defaults to today)
+        end_date: End date in YYYYMMDD format (defaults to today)
+
+    Returns:
+        DataFrame with columns: symbol, date, open, high, low, close, pre_close, change, pct_change, volume, amount
+    """
+    if _source is None:
+        raise RuntimeError("Data source not initialized. Call init_source() first.")
+    today = date.today().strftime("%Y%m%d")
+    start_date = start_date or today
+    end_date = end_date or today
+    return _source.get_index_bar(symbol, start_date, end_date)

hqdata-0.1.2/hqdata/config.py

@@ -0,0 +1,14 @@
+"""Configuration loading from .env file"""
+
+from pathlib import Path
+from dotenv import load_dotenv
+
+# Load .env from current working directory first (user's project root)
+# Then fall back to package directory (for development)
+_cwd_env = Path.cwd() / ".env"
+_pkg_env = Path(__file__).parent.parent / ".env"
+
+if _cwd_env.exists():
+    load_dotenv(_cwd_env)
+elif _pkg_env.exists():
+    load_dotenv(_pkg_env)

hqdata-0.1.2/hqdata/sources/__init__.py

@@ -0,0 +1 @@
+"""hqdata data sources"""

hqdata-0.1.2/hqdata/sources/base.py

@@ -0,0 +1,70 @@
+"""Base class for data sources"""
+
+import os
+from abc import ABC, abstractmethod
+from typing import Optional
+import pandas as pd
+
+
+class BaseSource(ABC):
+    """Abstract base class for data source adapters."""
+
+    @staticmethod
+    def _get_env(param: Optional[str], env_var: str, error_msg: str) -> str:
+        value = param or os.getenv(env_var)
+        if not value:
+            raise ValueError(error_msg)
+        return value
+
+    @abstractmethod
+    def get_bar(
+        self,
+        symbol: str,
+        frequency: str = "1day",
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+    ) -> pd.DataFrame:
+        """Get K-line/bar data for a stock.
+
+        Args:
+            symbol: Stock symbol with exchange (e.g., "600000.SH" or "000001.SZ")
+            frequency: Bar frequency ("1min", "5min", "15min", "30min", "60min", "1day", "1week", "1month")
+            start_date: Start date in YYYYMMDD format
+            end_date: End date in YYYYMMDD format
+
+        Returns:
+            DataFrame with columns: symbol, date, open, high, low, close, pre_close, change, pct_change, volume, amount
+        """
+        pass
+
+    @abstractmethod
+    def get_stock_list(self, list_status: str = "L") -> pd.DataFrame:
+        """Get basic info for stocks.
+
+        Args:
+            list_status: Listing status — "L" (listed), "D" (delisted), "P" (suspended)
+
+        Returns:
+            DataFrame with columns: symbol, name, industry, market, exchange,
+            curr_type, list_status, list_date, delist_date, is_hs
+        """
+        pass
+
+    @abstractmethod
+    def get_index_bar(
+        self,
+        symbol: str,
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+    ) -> pd.DataFrame:
+        """Get daily bar data for an index.
+
+        Args:
+            symbol: Index code with exchange (e.g., "000300.SH", "000905.SH")
+            start_date: Start date in YYYYMMDD format
+            end_date: End date in YYYYMMDD format
+
+        Returns:
+            DataFrame with columns: symbol, date, open, high, low, close, pre_close, change, pct_change, volume, amount
+        """
+        pass

hqdata-0.1.2/hqdata/sources/ricequant.py

@@ -0,0 +1,74 @@
+"""Ricequant (米筐) data source adapter"""
+
+from typing import Optional
+import pandas as pd
+
+from hqdata.sources.base import BaseSource
+
+
+def _get_rqdatac():
+    """Lazy import rqdatac to support optional installation."""
+    from rqdatac import init, get_price
+    from rqdatac.share.errors import RQDataError
+    return init, get_price, RQDataError
+
+
+class RicequantSource(BaseSource):
+    """Ricequant data source adapter.
+
+    Requires rqdatac >= 3.1.4 and valid RQData credentials.
+    Credentials can be provided via arguments or environment variables
+    (RQDATA_USERNAME, RQDATA_PASSWORD).
+    """
+
+    def __init__(
+        self,
+        username: Optional[str] = None,
+        password: Optional[str] = None,
+    ):
+        """Initialize ricequant connection.
+
+        Args:
+            username: RQData username (email), defaults to RQDATA_USERNAME env var
+            password: RQData password/token, defaults to RQDATA_PASSWORD env var
+        """
+        username = BaseSource._get_env(
+            username, "RQDATA_USERNAME",
+            "RQData credentials not provided. Set username/password in init_source() "
+            "or set RQDATA_USERNAME and RQDATA_PASSWORD environment variables."
+        )
+        password = BaseSource._get_env(
+            password, "RQDATA_PASSWORD",
+            "RQData credentials not provided. Set username/password in init_source() "
+            "or set RQDATA_USERNAME and RQDATA_PASSWORD environment variables."
+        )
+
+        init, _, _ = _get_rqdatac()
+        init(
+            username,
+            password,
+            address=("rqdatad-pro.ricequant.com", 16011),
+            use_pool=True,
+            max_pool_size=1,
+            auto_load_plugins=False,
+        )
+
+    def get_stock_list(self, list_status: str = "L") -> pd.DataFrame:
+        raise NotImplementedError("Ricequant get_stock_list not implemented yet.")
+
+    def get_bar(
+        self,
+        symbol: str,
+        frequency: str = "1day",
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+    ) -> pd.DataFrame:
+        raise NotImplementedError("Ricequant get_bar not implemented yet.")
+
+    def get_index_bar(
+        self,
+        symbol: str,
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+    ) -> pd.DataFrame:
+        raise NotImplementedError("Ricequant get_index_bar not implemented yet.")

hqdata-0.1.2/hqdata/sources/tushare.py

@@ -0,0 +1,113 @@
+"""Tushare data source adapter"""
+
+from typing import Optional
+import pandas as pd
+
+from hqdata.sources.base import BaseSource
+
+
+def _get_tushare():
+    """Lazy import tushare to support optional installation."""
+    import tushare as ts
+    return ts
+
+
+class TushareSource(BaseSource):
+    """Tushare data source adapter.
+
+    Requires tushare >= 1.4.0 and valid TUSHARE_TOKEN.
+    Token can be set via environment variable TUSHARE_TOKEN.
+    """
+
+    def __init__(self, token: Optional[str] = None):
+        """Initialize tushare connection.
+
+        Args:
+            token: Tushare token, defaults to TUSHARE_TOKEN env var
+        """
+        token = BaseSource._get_env(
+            token, "TUSHARE_TOKEN",
+            "Tushare token not provided. Set token in init_source() "
+            "or set TUSHARE_TOKEN environment variable."
+        )
+        ts = _get_tushare()
+        ts.set_token(token)
+        self.pro = ts.pro_api()
+
+    @staticmethod
+    def _rename_daily_columns(df: pd.DataFrame) -> pd.DataFrame:
+        return df.rename(columns={
+            "ts_code": "symbol",
+            "trade_date": "date",
+            "vol": "volume",
+            "pct_chg": "pct_change",
+        })
+
+    _STOCK_LIST_FIELDS = (
+        "symbol,name,industry,market,exchange,curr_type,list_status,list_date,delist_date,is_hs"
+    )
+
+    def get_stock_list(self, list_status: str = "L") -> pd.DataFrame:
+        """Get basic info for stocks.
+
+        Args:
+            list_status: Listing status — "L" (listed), "D" (delisted), "P" (suspended)
+
+        Returns:
+            DataFrame with columns: symbol, name, industry, market, exchange,
+            curr_type, list_status, list_date, delist_date, is_hs
+        """
+        return self.pro.stock_basic(list_status=list_status, fields=self._STOCK_LIST_FIELDS)
+
+    def get_bar(
+        self,
+        symbol: str,
+        frequency: str = "1day",
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+    ) -> pd.DataFrame:
+        """Get daily bar data for a stock.
+
+        Args:
+            symbol: Stock symbol with exchange (e.g., "600000.SH" or "000001.SZ")
+            frequency: Bar frequency ("1day" only, other frequencies not supported)
+            start_date: Start date in YYYYMMDD format
+            end_date: End date in YYYYMMDD format
+
+        Returns:
+            DataFrame with columns: symbol, date, open, high, low, close, pre_close, change, pct_change, volume, amount
+        """
+        if frequency != "1day":
+            raise NotImplementedError(
+                f"Tushare only supports '1day' frequency, got '{frequency}'"
+            )
+
+        df = self.pro.daily(ts_code=symbol, start_date=start_date, end_date=end_date)
+
+        if df is not None and not df.empty:
+            df = self._rename_daily_columns(df).sort_values("date")
+
+        return df
+
+    def get_index_bar(
+        self,
+        symbol: str,
+        start_date: Optional[str] = None,
+        end_date: Optional[str] = None,
+    ) -> pd.DataFrame:
+        """Get daily bar data for an index.
+
+        Args:
+            symbol: Index code with exchange (e.g., "000300.SH", "000905.SH")
+            start_date: Start date in YYYYMMDD format
+            end_date: End date in YYYYMMDD format
+
+        Returns:
+            DataFrame with columns: symbol, date, open, high, low, close, pre_close, change, pct_change, volume, amount
+        """
+        df = self.pro.index_daily(ts_code=symbol, start_date=start_date, end_date=end_date)
+
+        if df is not None and not df.empty:
+            df = self._rename_daily_columns(df).sort_values("date")
+
+        return df

hqdata-0.1.2/hqdata.egg-info/PKG-INFO

@@ -0,0 +1,106 @@
+Metadata-Version: 2.4
+Name: hqdata
+Version: 0.1.2
+Summary: A股历史与实时行情数据统一接入、清洗与存储
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pandas>=2.3.3
+Requires-Dist: python-dotenv>=1.0.0
+Provides-Extra: tushare
+Requires-Dist: tushare>=1.4.29; extra == "tushare"
+Provides-Extra: ricequant
+Requires-Dist: rqdatac>=3.1.4; extra == "ricequant"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0; extra == "dev"
+Dynamic: license-file
+
+# hqdata - A股历史与实时行情数据统一接入、清洗与存储
+
+<p align="center">
+    <img src ="https://img.shields.io/pypi/v/hqdata.svg"/>
+    <img src ="https://img.shields.io/pypi/pyversions/hqdata.svg"/>
+</p>
+
+## 定位
+
+`hqdata` 是 HonestQuant 量化系统的**数据基础层**，职责边界清晰：
+
+- 对下：封装各数据源 SDK，屏蔽接口差异
+- 对上：提供统一的查询接口
+- 上层策略和引擎**只调用 `hqdata.api`**，不直接接触任何数据源
+
+## 支持的数据源
+
+| 数据源      | 状态   | 说明                                        |
+| ----------- | ------ | ------------------------------------------- |
+| **AKShare** | 计划中 | 免费，实时数据                              |
+| **Tushare** | 已接入 | 需满足账户积分要求，支持股票&指数的历史日线 |
+| **米筐**    | 接入中 | 需license，支持Tick                         |
+| **迅投**    | 计划中 | 需迅投终端                                  |
+| **iTick**   | 计划中 | 需注册                                      |
+
+## 安装
+
+### 方式一：通过pip安装
+
+```bash
+# 基础安装（仅包含核心功能）
+pip install hqdata
+
+# 按需安装数据源依赖
+pip install hqdata[tushare]      # Tushare 支持
+pip install hqdata[ricequant]    # 米筐支持
+pip install hqdata[tushare,ricequant]  # 同时安装两者
+
+# 复制模板并自行填入所需字段
+cp .env.example .env # 放在你运行 Python 代码的当前目录（优先）/包安装目录
+```
+
+### 方式二：本地开发
+
+```bash
+# 创建虚拟环境
+python -m venv .venv
+source .venv/bin/activate
+
+# 安装依赖 (editable 模式，改代码直接生效)
+pip install -e .
+
+# 复制模板并自行填入所需字段
+cp .env.example .env # 放在你运行 Python 代码的当前目录（优先）/包安装目录
+```
+
+## 使用
+
+以Tushare数据为例
+
+```python
+from hqdata import init_source, get_bar
+
+# 初始化 Tushare (日线数据)
+init_source("tushare")
+
+# 获取日线数据 (symbol 格式: "代码.交易所")
+df = get_bar("600000.SH", frequency="1day", start_date="20260401", end_date="20260402")
+print(df.head())
+```
+
+## 输入参数格式说明
+
+### 股票代码
+
+symbol 参数统一使用 `代码.交易所` 格式：
+
+| 交易所 | 代码 | 示例        |
+| ------ | ---- | ----------- |
+| 上交所 | SH   | `600000.SH` |
+| 深交所 | SZ   | `000001.SZ` |
+
+## 测试
+
+```bash
+pytest tests/ -v
+pytest tests/test_tushare.py::TestTushareIntegration::test_get_bar  # 运行单个测试
+```

hqdata-0.1.2/hqdata.egg-info/SOURCES.txt

@@ -0,0 +1,17 @@
+LICENSE
+README.md
+pyproject.toml
+hqdata/__init__.py
+hqdata/api.py
+hqdata/config.py
+hqdata.egg-info/PKG-INFO
+hqdata.egg-info/SOURCES.txt
+hqdata.egg-info/dependency_links.txt
+hqdata.egg-info/requires.txt
+hqdata.egg-info/top_level.txt
+hqdata/sources/__init__.py
+hqdata/sources/base.py
+hqdata/sources/ricequant.py
+hqdata/sources/tushare.py
+tests/test_ricequant.py
+tests/test_tushare.py

hqdata-0.1.2/hqdata.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

hqdata-0.1.2/hqdata.egg-info/requires.txt

@@ -0,0 +1,12 @@
+pandas>=2.3.3
+python-dotenv>=1.0.0
+
+[dev]
+pytest>=7.0
+pytest-cov>=4.0
+
+[ricequant]
+rqdatac>=3.1.4
+
+[tushare]
+tushare>=1.4.29

hqdata-0.1.2/hqdata.egg-info/top_level.txt

@@ -0,0 +1 @@
+hqdata

hqdata-0.1.2/pyproject.toml

@@ -0,0 +1,33 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "hqdata"
+version = "0.1.2"
+description = "A股历史与实时行情数据统一接入、清洗与存储"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "pandas>=2.3.3",
+    "python-dotenv>=1.0.0",
+]
+
+[project.optional-dependencies]
+tushare = [
+    "tushare>=1.4.29",
+]
+ricequant = [
+    "rqdatac>=3.1.4",
+]
+dev = [
+    "pytest>=7.0",
+    "pytest-cov>=4.0",
+]
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["hqdata*"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]

hqdata-0.1.2/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

hqdata-0.1.2/tests/test_ricequant.py

@@ -0,0 +1,34 @@
+"""Tests for ricequant source"""
+
+import os
+import pytest
+from unittest.mock import patch
+
+import hqdata.config  # 加载 .env
+from hqdata.sources.ricequant import RicequantSource
+
+
+class TestRicequantSource:
+    """Unit tests for RicequantSource."""
+
+    @patch.dict(os.environ, {}, clear=True)
+    def test_init_missing_credentials_raises(self):
+        with pytest.raises(ValueError, match="RQDATA"):
+            RicequantSource(username=None, password=None)
+
+
+class TestRicequantIntegration:
+    """Integration tests using real Ricequant API data."""
+
+    @pytest.fixture(autouse=True)
+    def setup(self):
+        token = os.getenv("RQDATA_USERNAME")
+        password = os.getenv("RQDATA_PASSWORD")
+        if not token or not password:
+            pytest.skip("RQDATA_USERNAME or RQDATA_PASSWORD not set")
+        self.source = RicequantSource(username=token, password=password)
+
+    def test_get_bar_not_implemented(self):
+        """get_bar raises NotImplementedError until Ricequant adapter is implemented."""
+        with pytest.raises(NotImplementedError):
+            self.source.get_bar("600000.SH", "1day", "20260101", "20260101")

hqdata-0.1.2/tests/test_tushare.py

@@ -0,0 +1,78 @@
+"""Tests for tushare source"""
+
+import os
+import pytest
+from unittest.mock import patch
+import pandas as pd
+
+import hqdata.config  # 加载 .env
+from hqdata.sources.tushare import TushareSource
+
+
+class TestTushareSource:
+    """Unit tests for TushareSource."""
+
+    @patch.dict(os.environ, {}, clear=True)
+    def test_init_missing_token_raises(self):
+        with pytest.raises(ValueError, match="TUSHARE_TOKEN"):
+            TushareSource(token=None)
+
+
+class TestTushareIntegration:
+    """Integration tests using real Tushare API data."""
+
+    @pytest.fixture(autouse=True)
+    def setup(self):
+        token = os.getenv("TUSHARE_TOKEN")
+        if not token:
+            pytest.skip("TUSHARE_TOKEN not set")
+        self.source = TushareSource(token=token)
+
+    def test_get_stock_list(self):
+        """Test get_stock_list returns well-formed data for listed stocks."""
+        df = self.source.get_stock_list(list_status="L")
+        expected_columns = {"symbol", "name", "industry", "market", "exchange",
+                            "curr_type", "list_status", "list_date", "delist_date", "is_hs"}
+        excluded_columns = {"ts_code", "area", "fullname", "enname", "cnspell", "act_name", "act_ent_type"}
+
+        assert not df.empty
+        assert expected_columns.issubset(df.columns), f"Missing columns: {expected_columns - set(df.columns)}"
+        assert not excluded_columns.intersection(df.columns), f"Unexpected columns present: {excluded_columns & set(df.columns)}"
+        assert (df["list_status"] == "L").all()
+        assert df["symbol"].is_unique
+
+    def test_get_bar(self):
+        """Test get_bar returns well-formed data for both markets."""
+        expected_columns = {"symbol", "date", "open", "high", "low", "close",
+                            "pre_close", "change", "pct_change", "volume", "amount"}
+
+        for symbol in ("000001.SZ", "600000.SH"):
+            df = self.source.get_bar(symbol, "1day", "20260101", "20260401")
+            assert not df.empty, f"{symbol} returned empty DataFrame"
+            assert expected_columns.issubset(df.columns), f"Missing columns: {expected_columns - set(df.columns)}"
+            assert (df["high"] >= df["low"]).all(), "high < low found"
+            assert (df["high"] >= df["close"]).all(), "high < close found"
+            assert (df["low"] <= df["close"]).all(), "low > close found"
+            assert (df["volume"] > 0).all(), "non-positive volume found"
+            assert (df["amount"] > 0).all(), "non-positive amount found"
+
+    def test_get_bar_unsupported_frequency(self):
+        """Test that unsupported frequency raises NotImplementedError."""
+        with pytest.raises(NotImplementedError):
+            self.source.get_bar("000001.SZ", "1min", "20260101", "20260101")
+
+    def test_get_index_bar(self):
+        """Test get_index_bar returns well-formed data for major indexes."""
+        expected_columns = {"symbol", "date", "open", "high", "low", "close",
+                            "pre_close", "change", "pct_change", "volume", "amount"}
+
+        for symbol in ("000300.SH", "000905.SH", "000852.SH", "932000.CSI"):
+            df = self.source.get_index_bar(symbol, "20260101", "20260401")
+            print(df)
+            assert not df.empty, f"{symbol} returned empty DataFrame"
+            assert expected_columns.issubset(df.columns), f"Missing columns: {expected_columns - set(df.columns)}"
+            assert (df["high"] >= df["low"]).all(), "high < low found"
+            assert (df["high"] >= df["close"]).all(), "high < close found"
+            assert (df["low"] <= df["close"]).all(), "low > close found"
+            assert (df["volume"] > 0).all(), "non-positive volume found"
+            assert (df["amount"] > 0).all(), "non-positive amount found"