akshare-one 0.2.0__tar.gz → 0.2.1__tar.gz

{akshare_one-0.2.0 → akshare_one-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-one
-Version: 0.2.0
+Version: 0.2.1
 Summary: Standardized interface for Chinese financial market data, built on AKShare with unified data formats and simplified APIs
 License-Expression: MIT
 Project-URL: Homepage, https://github.com/zwldarren/akshare-one
@@ -9,35 +9,35 @@ Keywords: akshare,financial-data,stock-data,quant
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: akshare>=1.16.94
-Requires-Dist: cachetools>=5.5.2
+Requires-Dist: akshare>=1.16.98
+Requires-Dist: cachetools>=6.0.0
 Dynamic: license-file
 
 <div align="center">
   <h1>AKShare One</h1>
   <div>
-    <strong>English</strong> | <a href="README_zh.md">中文</a>
+    <a href="README_zh.md">中文</a> | <strong>English</strong>
   </div>
 </div>
 
-**AKShare One** is a standardized interface for Chinese financial market data, built as a wrapper around [AKShare](https://github.com/akfamily/akshare) to solve inconsistencies in input/output formats across different data sources.
+**AKShare One** is a data interface for obtaining Chinese A-shares, based on [AKShare](https://github.com/akfamily/akshare). It aims to simplify AKShare's usage and unify input/output formats from different data sources, making it easier to pass data to LLM.
 
 ## ✨ Features
 
 - 📊 Unified stock code formats across data sources
 - 🏗️ Standardized return data structures
-- 🛠️ Simplified API parameters
+- 🛠️ Simplified API parameter design
 - ⏱️ Automatic timestamp and adjustment handling
 
 ## 🚀 Core Features
 
-| Feature | Interface |
-|---------|-----------|
+| Function | Interface |
+|------|------|
 | Historical data | `get_hist_data` |
 | Real-time quotes | `get_realtime_data` |
 | Stock news | `get_news_data` |
 | Financial data | `get_balance_sheet`/`get_income_statement`/`get_cash_flow` |
-| Insider trading | `get_inner_trade_data` |
+| Internal transactions | `get_inner_trade_data` |
 
 ## 📦 Quick Installation
 

{akshare_one-0.2.0 → akshare_one-0.2.1}/README.md

@@ -1,28 +1,28 @@
 <div align="center">
   <h1>AKShare One</h1>
   <div>
-    <strong>English</strong> | <a href="README_zh.md">中文</a>
+    <a href="README_zh.md">中文</a> | <strong>English</strong>
   </div>
 </div>
 
-**AKShare One** is a standardized interface for Chinese financial market data, built as a wrapper around [AKShare](https://github.com/akfamily/akshare) to solve inconsistencies in input/output formats across different data sources.
+**AKShare One** is a data interface for obtaining Chinese A-shares, based on [AKShare](https://github.com/akfamily/akshare). It aims to simplify AKShare's usage and unify input/output formats from different data sources, making it easier to pass data to LLM.
 
 ## ✨ Features
 
 - 📊 Unified stock code formats across data sources
 - 🏗️ Standardized return data structures
-- 🛠️ Simplified API parameters
+- 🛠️ Simplified API parameter design
 - ⏱️ Automatic timestamp and adjustment handling
 
 ## 🚀 Core Features
 
-| Feature | Interface |
-|---------|-----------|
+| Function | Interface |
+|------|------|
 | Historical data | `get_hist_data` |
 | Real-time quotes | `get_realtime_data` |
 | Stock news | `get_news_data` |
 | Financial data | `get_balance_sheet`/`get_income_statement`/`get_cash_flow` |
-| Insider trading | `get_inner_trade_data` |
+| Internal transactions | `get_inner_trade_data` |
 
 ## 📦 Quick Installation
 

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/insider.py

@@ -3,19 +3,16 @@
 包含上市公司内部交易相关功能
 """
 
-from typing import Optional
 import pandas as pd
 from .modules.insider.factory import InsiderDataFactory
 
 
-def get_inner_trade_data(
-    symbol: Optional[str] = None, source: str = "xueqiu"
-) -> "pd.DataFrame":
+def get_inner_trade_data(symbol: str, source: str = "xueqiu") -> "pd.DataFrame":
     """获取雪球内部交易数据
 
     Args:
         source: 数据源 (目前支持 "xueqiu")
-        symbol: 可选股票代码，如"600000"，不传则返回所有数据
+        symbol: 股票代码，如"600000"
 
     Returns:
         pd.DataFrame:

akshare_one-0.2.1/akshare_one/modules/financial/base.py

@@ -0,0 +1,22 @@
+from abc import ABC, abstractmethod
+import pandas as pd
+
+
+class FinancialDataProvider(ABC):
+    def __init__(self, symbol: str) -> None:
+        self.symbol = symbol
+
+    @abstractmethod
+    def get_balance_sheet(self) -> pd.DataFrame:
+        """Fetches balance sheet data"""
+        pass
+
+    @abstractmethod
+    def get_income_statement(self) -> pd.DataFrame:
+        """Fetches income statement data"""
+        pass
+
+    @abstractmethod
+    def get_cash_flow(self) -> pd.DataFrame:
+        """Fetches cash flow data"""
+        pass

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/modules/financial/sina.py

@@ -3,7 +3,7 @@ import pandas as pd
 import akshare as ak
 
 from akshare_one.modules.cache import CACHE_CONFIG
-from .base import FinancialDataProvider, validate_financial_data
+from .base import FinancialDataProvider
 
 
 class SinaFinancialReport(FinancialDataProvider):
@@ -13,7 +13,6 @@ class SinaFinancialReport(FinancialDataProvider):
             f"sh{symbol}" if not symbol.startswith(("sh", "sz", "bj")) else symbol
         )
 
-    @validate_financial_data
     @cached(
         CACHE_CONFIG["financial_cache"],
         key=lambda self, symbol=None: f"sina_balance_{self.symbol}",
@@ -30,7 +29,6 @@ class SinaFinancialReport(FinancialDataProvider):
         raw_df = ak.stock_financial_report_sina(stock=self.stock, symbol="资产负债表")
         return self._clean_balance_data(raw_df)
 
-    @validate_financial_data
     @cached(
         CACHE_CONFIG["financial_cache"],
         key=lambda self, symbol=None: f"sina_income_{self.symbol}",
@@ -47,7 +45,6 @@ class SinaFinancialReport(FinancialDataProvider):
         raw_df = ak.stock_financial_report_sina(stock=self.stock, symbol="利润表")
         return self._clean_income_data(raw_df)
 
-    @validate_financial_data
     @cached(
         CACHE_CONFIG["financial_cache"],
         key=lambda self, symbol=None: f"sina_cash_{self.symbol}",

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/modules/historical/base.py

@@ -2,31 +2,6 @@ from abc import ABC, abstractmethod
 import pandas as pd
 
 
-def validate_hist_data(func):
-    """Decorator to validate historical data returned by data providers"""
-
-    def wrapper(*args, **kwargs):
-        df = func(*args, **kwargs)
-
-        if not isinstance(df, pd.DataFrame):
-            raise ValueError("Returned data must be a pandas DataFrame")
-
-        required_columns = {"timestamp", "open", "high", "low", "close", "volume"}
-        missing_cols = required_columns - set(df.columns)
-        if missing_cols:
-            raise ValueError(f"Missing required columns: {missing_cols}")
-
-        if "timestamp" in df.columns:
-            if not pd.api.types.is_datetime64_any_dtype(df["timestamp"]):
-                raise ValueError("timestamp must be datetime64 dtype")
-            if df["timestamp"].dt.tz is None or str(df["timestamp"].dt.tz) != "UTC":
-                raise ValueError("timestamp must be in UTC timezone")
-
-        return df
-
-    return wrapper
-
-
 class HistoricalDataProvider(ABC):
     def __init__(
         self,

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/modules/historical/eastmoney.py

@@ -1,5 +1,5 @@
 from cachetools import cached
-from .base import HistoricalDataProvider, validate_hist_data
+from .base import HistoricalDataProvider
 import akshare as ak
 import pandas as pd
 from ..cache import CACHE_CONFIG
@@ -8,7 +8,6 @@ from ..cache import CACHE_CONFIG
 class EastMoneyHistorical(HistoricalDataProvider):
     """Adapter for EastMoney historical stock data API"""
 
-    @validate_hist_data
     @cached(
         cache=CACHE_CONFIG["hist_data_cache"],
         key=lambda self: f"eastmoney_hist_{self.symbol}_{self.interval}_{self.interval_multiplier}_{self.adjust}",

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/modules/historical/sina.py

@@ -1,5 +1,5 @@
 from cachetools import cached
-from .base import HistoricalDataProvider, validate_hist_data
+from .base import HistoricalDataProvider
 import akshare as ak
 import pandas as pd
 from ..cache import CACHE_CONFIG
@@ -8,7 +8,6 @@ from ..cache import CACHE_CONFIG
 class SinaHistorical(HistoricalDataProvider):
     """Adapter for Sina historical stock data API"""
 
-    @validate_hist_data
     @cached(
         cache=CACHE_CONFIG["hist_data_cache"],
         key=lambda self: f"sina_hist_{self.symbol}_{self.interval}_{self.interval_multiplier}_{self.adjust}",

akshare_one-0.2.1/akshare_one/modules/insider/base.py

@@ -0,0 +1,28 @@
+from abc import ABC, abstractmethod
+import pandas as pd
+
+
+class InsiderDataProvider(ABC):
+    def __init__(self, symbol: str) -> None:
+        self.symbol = symbol
+
+    @abstractmethod
+    def get_inner_trade_data(self) -> pd.DataFrame:
+        """Fetches insider trade data
+
+        Returns:
+            pd.DataFrame:
+            - symbol: 股票代码
+            - issuer: 股票名称
+            - name: 变动人
+            - title: 董监高职务
+            - transaction_date: 变动日期(UTC时区)
+            - transaction_shares: 变动股数
+            - transaction_price_per_share: 成交均价
+            - shares_owned_after_transaction: 变动后持股数
+            - relationship: 与董监高关系
+            - is_board_director: 是否为董事会成员
+            - transaction_value: 交易金额(变动股数*成交均价)
+            - shares_owned_before_transaction: 变动前持股数
+        """
+        pass

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/modules/insider/xueqiu.py

@@ -1,7 +1,7 @@
 from cachetools import cached
 import pandas as pd
 import akshare as ak
-from .base import InsiderDataProvider, validate_insider_data
+from .base import InsiderDataProvider
 from ..utils import convert_xieqiu_symbol
 from ..cache import CACHE_CONFIG
 
@@ -9,7 +9,6 @@ from ..cache import CACHE_CONFIG
 class XueQiuInsider(InsiderDataProvider):
     """Provider for XueQiu insider trading data"""
 
-    @validate_insider_data
     @cached(
         cache=CACHE_CONFIG["financial_cache"],
         key=lambda self, symbol=None: f"xueqiu_insider_{symbol if symbol else 'all'}",

akshare_one-0.2.1/akshare_one/modules/news/base.py

@@ -0,0 +1,22 @@
+from abc import ABC, abstractmethod
+import pandas as pd
+
+
+class NewsDataProvider(ABC):
+    def __init__(self, symbol: str) -> None:
+        self.symbol = symbol
+
+    @abstractmethod
+    def get_news_data(self) -> pd.DataFrame:
+        """Fetches news data for given symbol
+
+        Returns:
+            pd.DataFrame:
+            - keyword: 关键词
+            - title: 新闻标题
+            - content: 新闻内容
+            - publish_time: 发布时间 (UTC)
+            - source: 文章来源
+            - url: 新闻链接
+        """
+        pass

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/modules/news/eastmoney.py

@@ -3,11 +3,10 @@ import pandas as pd
 import akshare as ak
 
 from ..cache import CACHE_CONFIG
-from .base import NewsDataProvider, validate_news_data
+from .base import NewsDataProvider
 
 
 class EastMoneyNews(NewsDataProvider):
-    @validate_news_data
     @cached(
         CACHE_CONFIG["news_cache"],
         key=lambda self: f"eastmoney_news_{self.symbol}",

akshare_one-0.2.1/akshare_one/modules/realtime/base.py

@@ -0,0 +1,27 @@
+from abc import ABC, abstractmethod
+import pandas as pd
+
+
+class RealtimeDataProvider(ABC):
+    def __init__(self, symbol: str) -> None:
+        self.symbol = symbol
+
+    @abstractmethod
+    def get_current_data(self) -> pd.DataFrame:
+        """Fetches realtime market data
+
+        Returns:
+            pd.DataFrame:
+            - symbol: 股票代码
+            - price: 最新价
+            - change: 涨跌额
+            - pct_change: 涨跌幅(%)
+            - timestamp: 时间戳
+            - volume: 成交量(手)
+            - amount: 成交额(元)
+            - open: 今开
+            - high: 最高
+            - low: 最低
+            - prev_close: 昨收
+        """
+        pass

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/modules/realtime/eastmoney.py

@@ -3,11 +3,10 @@ import pandas as pd
 import akshare as ak
 
 from ..cache import CACHE_CONFIG
-from .base import RealtimeDataProvider, validate_realtime_data
+from .base import RealtimeDataProvider
 
 
 class EastmoneyRealtime(RealtimeDataProvider):
-    @validate_realtime_data
     @cached(
         CACHE_CONFIG["realtime_cache"],
         key=lambda self, symbol=None: f"eastmoney_{symbol if symbol else 'all'}",

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/modules/realtime/xueqiu.py

@@ -3,11 +3,10 @@ import pandas as pd
 import akshare as ak
 from ..utils import convert_xieqiu_symbol
 from ..cache import CACHE_CONFIG
-from .base import RealtimeDataProvider, validate_realtime_data
+from .base import RealtimeDataProvider
 
 
 class XueQiuRealtime(RealtimeDataProvider):
-    @validate_realtime_data
     @cached(
         cache=CACHE_CONFIG["realtime_cache"],
         key=lambda self, symbol=None: f"xueqiu_{symbol}",

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one/stock.py

@@ -28,7 +28,7 @@ def get_hist_data(
         start_date: 开始日期 (YYYY-MM-DD)
         end_date: 结束日期 (YYYY-MM-DD)
         adjust: 复权类型 ('none','qfq','hfq')
-        source: 数据源 ('eastmoney', 'sina')
+        source: 数据源 ('eastmoney', 'sina') (default: 'eastmoney')
 
     Returns:
         pd.DataFrame:

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-one
-Version: 0.2.0
+Version: 0.2.1
 Summary: Standardized interface for Chinese financial market data, built on AKShare with unified data formats and simplified APIs
 License-Expression: MIT
 Project-URL: Homepage, https://github.com/zwldarren/akshare-one
@@ -9,35 +9,35 @@ Keywords: akshare,financial-data,stock-data,quant
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: akshare>=1.16.94
-Requires-Dist: cachetools>=5.5.2
+Requires-Dist: akshare>=1.16.98
+Requires-Dist: cachetools>=6.0.0
 Dynamic: license-file
 
 <div align="center">
   <h1>AKShare One</h1>
   <div>
-    <strong>English</strong> | <a href="README_zh.md">中文</a>
+    <a href="README_zh.md">中文</a> | <strong>English</strong>
   </div>
 </div>
 
-**AKShare One** is a standardized interface for Chinese financial market data, built as a wrapper around [AKShare](https://github.com/akfamily/akshare) to solve inconsistencies in input/output formats across different data sources.
+**AKShare One** is a data interface for obtaining Chinese A-shares, based on [AKShare](https://github.com/akfamily/akshare). It aims to simplify AKShare's usage and unify input/output formats from different data sources, making it easier to pass data to LLM.
 
 ## ✨ Features
 
 - 📊 Unified stock code formats across data sources
 - 🏗️ Standardized return data structures
-- 🛠️ Simplified API parameters
+- 🛠️ Simplified API parameter design
 - ⏱️ Automatic timestamp and adjustment handling
 
 ## 🚀 Core Features
 
-| Feature | Interface |
-|---------|-----------|
+| Function | Interface |
+|------|------|
 | Historical data | `get_hist_data` |
 | Real-time quotes | `get_realtime_data` |
 | Stock news | `get_news_data` |
 | Financial data | `get_balance_sheet`/`get_income_statement`/`get_cash_flow` |
-| Insider trading | `get_inner_trade_data` |
+| Internal transactions | `get_inner_trade_data` |
 
 ## 📦 Quick Installation
 

{akshare_one-0.2.0 → akshare_one-0.2.1}/akshare_one.egg-info/SOURCES.txt

@@ -29,4 +29,8 @@ akshare_one/modules/news/factory.py
 akshare_one/modules/realtime/base.py
 akshare_one/modules/realtime/eastmoney.py
 akshare_one/modules/realtime/factory.py
-akshare_one/modules/realtime/xueqiu.py
+akshare_one/modules/realtime/xueqiu.py
+tests/test_financial.py
+tests/test_insider.py
+tests/test_news.py
+tests/test_stock.py

akshare_one-0.2.1/akshare_one.egg-info/requires.txt

@@ -0,0 +1,2 @@
+akshare>=1.16.98
+cachetools>=6.0.0

{akshare_one-0.2.0 → akshare_one-0.2.1}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "akshare-one"
-version = "0.2.0"
+version = "0.2.1"
 description = "Standardized interface for Chinese financial market data, built on AKShare with unified data formats and simplified APIs"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
-    "akshare>=1.16.94",
-    "cachetools>=5.5.2",
+    "akshare>=1.16.98",
+    "cachetools>=6.0.0",
 ]
 license = "MIT"
 keywords = ["akshare", "financial-data", "stock-data", "quant"]
@@ -18,6 +18,13 @@ Repository = "https://github.com/zwldarren/akshare-one.git"
 [dependency-groups]
 dev = [
     "pre-commit>=4.2.0",
-    "pytest>=8.3.5",
-    "ruff>=0.11.6",
+    "pytest>=8.4.0",
+    "pytest-cov>=6.1.1",
+    "ruff>=0.11.12",
 ]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = "test_*.py"
+python_functions = "test_*"
+addopts = "-v --cov=akshare_one --cov-report=term-missing"

akshare_one-0.2.1/tests/test_financial.py

@@ -0,0 +1,64 @@
+import pytest
+from akshare_one import get_balance_sheet, get_income_statement, get_cash_flow
+
+
+class TestBalanceSheet:
+    def test_basic_balance_sheet(self):
+        """测试基本资产负债表获取功能"""
+        df = get_balance_sheet(symbol="600600")
+        assert not df.empty
+        required_columns = {
+            "report_date",
+            "currency",
+            "total_assets",
+            "current_assets",
+            "cash_and_equivalents",
+            "total_liabilities",
+            "current_liabilities",
+            "shareholders_equity",
+        }
+        assert required_columns.issubset(df.columns)
+
+    def test_invalid_symbol(self):
+        """测试无效股票代码"""
+        with pytest.raises(Exception):
+            get_balance_sheet(symbol="INVALID")
+
+
+class TestIncomeStatement:
+    def test_basic_income_statement(self):
+        """测试基本利润表获取功能"""
+        df = get_income_statement(symbol="600600")
+        assert not df.empty
+        required_columns = {
+            "report_date",
+            "currency",
+            "revenue",
+            "cost_of_revenue",
+            "operating_profit",
+            "net_income",
+            "earnings_per_share",
+        }
+        assert required_columns.issubset(df.columns)
+
+    def test_multiple_periods(self):
+        """测试多期数据获取"""
+        df = get_income_statement(symbol="600600")
+        assert len(df) >= 4  # 至少包含4个季度数据
+
+
+class TestCashFlow:
+    def test_basic_cash_flow(self):
+        """测试基本现金流量表获取功能"""
+        df = get_cash_flow(symbol="600600")
+        assert not df.empty
+        required_columns = {
+            "report_date",
+            "report_type",
+            "currency",
+            "net_cash_flow_from_operations",
+            "net_cash_flow_from_investing",
+            "net_cash_flow_from_financing",
+            "ending_cash_balance",
+        }
+        assert required_columns.issubset(df.columns)

akshare_one-0.2.1/tests/test_insider.py

@@ -0,0 +1,47 @@
+import pytest
+from akshare_one import get_inner_trade_data
+from datetime import datetime, timedelta, timezone
+
+
+class TestInnerTradeData:
+    def test_basic_inner_trade(self):
+        """测试基本内部交易数据获取功能"""
+        df = get_inner_trade_data(symbol="600405")
+        assert not df.empty
+        required_columns = {
+            "symbol",
+            "issuer",
+            "name",
+            "title",
+            "transaction_date",
+            "transaction_shares",
+            "transaction_price_per_share",
+            "shares_owned_after_transaction",
+            "relationship",
+            "is_board_director",
+            "transaction_value",
+        }
+        assert required_columns.issubset(df.columns)
+
+    def test_transaction_date_range(self):
+        """测试交易日期范围"""
+        df = get_inner_trade_data(symbol="600405")
+        now = datetime.now(timezone.utc)
+        one_year_ago = now - timedelta(days=365)
+
+        # 验证交易日期在合理范围内
+        assert all(one_year_ago <= ts <= now for ts in df["transaction_date"])
+
+    def test_transaction_value_calculation(self):
+        """测试交易金额计算正确性"""
+        df = get_inner_trade_data(symbol="600405")
+        sample = df.iloc[0]
+        calculated_value = (
+            sample["transaction_shares"] * sample["transaction_price_per_share"]
+        )
+        assert abs(sample["transaction_value"] - calculated_value) < 0.01
+
+    def test_invalid_source(self):
+        """测试无效数据源"""
+        with pytest.raises(Exception):
+            get_inner_trade_data("600405", source="invalid")

akshare_one-0.2.1/tests/test_news.py

@@ -0,0 +1,50 @@
+import pytest
+from akshare_one import get_news_data
+from datetime import datetime, timedelta, timezone
+
+
+class TestNewsData:
+    def test_basic_news_data(self):
+        """测试基本新闻数据获取功能"""
+        df = get_news_data(symbol="300059")
+        assert not df.empty
+        required_columns = {
+            "keyword",
+            "title",
+            "content",
+            "publish_time",
+            "source",
+            "url",
+        }
+        assert required_columns.issubset(df.columns)
+
+        # 验证发布时间格式
+        assert isinstance(df.iloc[0]["publish_time"], datetime)
+
+    def test_news_data_time_range(self):
+        """测试新闻数据时间范围"""
+        df = get_news_data(symbol="300059")
+        now = datetime.now(timezone.utc)
+        one_year_ago = now - timedelta(days=365)
+
+        # 验证新闻发布时间在合理范围内
+        assert all(one_year_ago <= ts <= now for ts in df["publish_time"])
+
+    def test_news_content_quality(self):
+        """测试新闻内容质量"""
+        df = get_news_data(symbol="300059")
+        sample_news = df.iloc[0]
+
+        assert len(sample_news["title"]) > 0
+        assert len(sample_news["content"]) > 0
+        assert sample_news["url"].startswith("http")
+
+    def test_invalid_symbol(self):
+        """测试无效股票代码"""
+        with pytest.raises(Exception):
+            get_news_data(symbol="INVALID")
+
+    def test_multiple_pages(self):
+        """测试多页新闻数据"""
+        df = get_news_data(symbol="300059")
+        assert len(df) >= 10  # 至少获取10条新闻

akshare_one-0.2.1/tests/test_stock.py

@@ -0,0 +1,101 @@
+import pytest
+from akshare_one import get_hist_data, get_realtime_data
+
+
+class TestHistData:
+    def test_basic_hist_data(self):
+        """测试基本历史数据获取功能"""
+        df = get_hist_data(
+            symbol="600000",
+            interval="day",
+            start_date="2024-01-01",
+            end_date="2024-01-31",
+        )
+        assert not df.empty
+        assert set(df.columns) == {
+            "timestamp",
+            "open",
+            "high",
+            "low",
+            "close",
+            "volume",
+        }
+        assert len(df) > 0
+
+    def test_hist_data_with_adjust(self):
+        """测试复权历史数据"""
+        df_qfq = get_hist_data(
+            symbol="600000",
+            interval="day",
+            adjust="qfq",
+            start_date="2024-01-01",
+            end_date="2024-01-31",
+        )
+        df_hfq = get_hist_data(
+            symbol="600000",
+            interval="day",
+            adjust="hfq",
+            start_date="2024-01-01",
+            end_date="2024-01-31",
+        )
+        assert not df_qfq.equals(df_hfq)
+
+    def test_minute_hist_data(self):
+        """测试新浪数据源的分钟级历史数据"""
+        df = get_hist_data(
+            symbol="600900",
+            interval="minute",
+            interval_multiplier=5,
+            start_date="2025-05-01",
+            end_date="2024-05-30",
+            source="sina",
+        )
+        assert not df.empty
+        assert len(df) > 0
+
+    def test_invalid_symbol(self):
+        """测试无效股票代码"""
+        with pytest.raises(Exception):
+            get_hist_data(
+                symbol="INVALID",
+                interval="day",
+                start_date="2024-01-01",
+                end_date="2024-01-31",
+            )
+
+
+class TestRealtimeData:
+    def test_basic_realtime_data(self):
+        """测试基本实时数据获取"""
+        df = get_realtime_data(symbol="600000")
+        assert not df.empty
+        assert set(df.columns) == {
+            "symbol",
+            "price",
+            "change",
+            "pct_change",
+            "timestamp",
+            "volume",
+            "amount",
+            "open",
+            "high",
+            "low",
+            "prev_close",
+        }
+
+    def test_all_realtime_data(self):
+        """测试获取所有股票实时数据"""
+        df = get_realtime_data()
+        assert not df.empty
+        assert "600000" in df["symbol"].values
+
+    def test_xueqiu_source(self):
+        """测试雪球数据源"""
+        df = get_realtime_data(symbol="600000", source="xueqiu")
+        assert not df.empty
+        assert df.iloc[0]["symbol"] == "600000"
+
+    def test_invalid_source(self):
+        """测试无效数据源"""
+        with pytest.raises(Exception):
+            get_realtime_data(symbol="600000", source="invalid")

akshare_one-0.2.0/akshare_one/modules/financial/base.py

@@ -1,41 +0,0 @@
-from abc import ABC, abstractmethod
-import pandas as pd
-
-
-def validate_financial_data(func):
-    """Decorator to validate financial data returned by data providers"""
-
-    def wrapper(*args, **kwargs):
-        df = func(*args, **kwargs)
-
-        if not isinstance(df, pd.DataFrame):
-            raise ValueError("Returned data must be a pandas DataFrame")
-
-        # Validate report_date if present
-        if "report_date" in df.columns:
-            if not pd.api.types.is_datetime64_any_dtype(df["report_date"]):
-                raise ValueError("report_date must be datetime64 dtype")
-
-        return df
-
-    return wrapper
-
-
-class FinancialDataProvider(ABC):
-    def __init__(self, symbol: str) -> None:
-        self.symbol = symbol
-
-    @abstractmethod
-    def get_balance_sheet(self) -> pd.DataFrame:
-        """Fetches balance sheet data"""
-        pass
-
-    @abstractmethod
-    def get_income_statement(self) -> pd.DataFrame:
-        """Fetches income statement data"""
-        pass
-
-    @abstractmethod
-    def get_cash_flow(self) -> pd.DataFrame:
-        """Fetches cash flow data"""
-        pass

akshare_one-0.2.0/akshare_one/modules/insider/base.py

@@ -1,78 +0,0 @@
-from abc import ABC, abstractmethod
-import pandas as pd
-from typing import Optional
-
-
-def validate_insider_data(func):
-    """Decorator to validate insider trading data returned by data providers"""
-
-    def wrapper(*args, **kwargs):
-        df = func(*args, **kwargs)
-
-        if not isinstance(df, pd.DataFrame):
-            raise ValueError("Returned data must be a pandas DataFrame")
-
-        # Required fields for insider trading data
-        required_fields = {
-            "symbol",
-            "issuer",
-            "name",
-            "transaction_date",
-            "transaction_shares",
-            "transaction_price_per_share",
-        }
-        if not required_fields.issubset(df.columns):
-            missing = required_fields - set(df.columns)
-            raise ValueError(f"Missing required fields: {missing}")
-
-        # Validate timestamp if present
-        if "transaction_date" in df.columns:
-            if not pd.api.types.is_datetime64_any_dtype(df["transaction_date"]):
-                raise ValueError("transaction_date must be datetime64 dtype")
-            if (
-                df["transaction_date"].dt.tz is None
-                or str(df["transaction_date"].dt.tz) != "UTC"
-            ):
-                raise ValueError("transaction_date must be in UTC timezone")
-
-        # Validate numeric fields
-        numeric_fields = {
-            "transaction_shares",
-            "transaction_price_per_share",
-            "transaction_value",
-            "shares_owned_before_transaction",
-            "shares_owned_after_transaction",
-        }
-        for field in numeric_fields & set(df.columns):
-            if not pd.api.types.is_numeric_dtype(df[field]):
-                raise ValueError(f"{field} must be numeric")
-
-        return df
-
-    return wrapper
-
-
-class InsiderDataProvider(ABC):
-    def __init__(self, symbol: Optional[str] = None) -> None:
-        self.symbol = symbol
-
-    @abstractmethod
-    def get_inner_trade_data(self, symbol: Optional[str] = None) -> pd.DataFrame:
-        """Fetches insider trade data
-
-        Returns:
-            pd.DataFrame:
-            - symbol: 股票代码
-            - issuer: 股票名称
-            - name: 变动人
-            - title: 董监高职务
-            - transaction_date: 变动日期(UTC时区)
-            - transaction_shares: 变动股数
-            - transaction_price_per_share: 成交均价
-            - shares_owned_after_transaction: 变动后持股数
-            - relationship: 与董监高关系
-            - is_board_director: 是否为董事会成员
-            - transaction_value: 交易金额(变动股数*成交均价)
-            - shares_owned_before_transaction: 变动前持股数
-        """
-        pass

akshare_one-0.2.0/akshare_one/modules/news/base.py

@@ -1,51 +0,0 @@
-from abc import ABC, abstractmethod
-import pandas as pd
-
-
-def validate_news_data(func):
-    """Decorator to validate news data returned by data providers"""
-
-    def wrapper(*args, **kwargs):
-        df = func(*args, **kwargs)
-
-        if not isinstance(df, pd.DataFrame):
-            raise ValueError("Returned data must be a pandas DataFrame")
-
-        # Required fields
-        required_fields = {"title", "publish_time"}
-        if not required_fields & set(df.columns):
-            raise ValueError(f"Must contain all required fields: {required_fields}")
-
-        # Validate publish_time if present
-        if "publish_time" in df.columns:
-            if not pd.api.types.is_datetime64_any_dtype(df["publish_time"]):
-                raise ValueError("publish_time must be datetime64 dtype")
-            if (
-                df["publish_time"].dt.tz is None
-                or str(df["publish_time"].dt.tz) != "UTC"
-            ):
-                raise ValueError("publish_time must be in UTC timezone")
-
-        return df
-
-    return wrapper
-
-
-class NewsDataProvider(ABC):
-    def __init__(self, symbol: str) -> None:
-        self.symbol = symbol
-
-    @abstractmethod
-    def get_news_data(self) -> pd.DataFrame:
-        """Fetches news data for given symbol
-
-        Returns:
-            pd.DataFrame:
-            - keyword: 关键词
-            - title: 新闻标题
-            - content: 新闻内容
-            - publish_time: 发布时间 (UTC)
-            - source: 文章来源
-            - url: 新闻链接
-        """
-        pass

akshare_one-0.2.0/akshare_one/modules/realtime/base.py

@@ -1,68 +0,0 @@
-from abc import ABC, abstractmethod
-import pandas as pd
-
-
-def validate_realtime_data(func):
-    """Decorator to validate realtime data returned by data providers"""
-
-    def wrapper(*args, **kwargs):
-        df = func(*args, **kwargs)
-
-        if not isinstance(df, pd.DataFrame):
-            raise ValueError("Returned data must be a pandas DataFrame")
-
-        # At least one of these core fields must be present
-        core_fields = {"timestamp", "price", "volume"}
-        if not core_fields & set(df.columns):
-            raise ValueError(f"Must contain at least one of: {core_fields}")
-
-        # Validate timestamp if present
-        if "timestamp" in df.columns:
-            if not pd.api.types.is_datetime64_any_dtype(df["timestamp"]):
-                raise ValueError("timestamp must be datetime64 dtype")
-            if df["timestamp"].dt.tz is None or str(df["timestamp"].dt.tz) != "UTC":
-                raise ValueError("timestamp must be in UTC timezone")
-
-        # Validate numeric fields if present
-        numeric_fields = {
-            "price",
-            "change",
-            "pct_change",
-            "open",
-            "high",
-            "low",
-            "prev_close",
-            "amount",
-        }
-        for field in numeric_fields & set(df.columns):
-            if not pd.api.types.is_numeric_dtype(df[field]):
-                raise ValueError(f"{field} must be numeric")
-
-        return df
-
-    return wrapper
-
-
-class RealtimeDataProvider(ABC):
-    def __init__(self, symbol: str) -> None:
-        self.symbol = symbol
-
-    @abstractmethod
-    def get_current_data(self) -> pd.DataFrame:
-        """Fetches realtime market data
-
-        Returns:
-            pd.DataFrame:
-            - symbol: 股票代码
-            - price: 最新价
-            - change: 涨跌额
-            - pct_change: 涨跌幅(%)
-            - timestamp: 时间戳
-            - volume: 成交量(手)
-            - amount: 成交额(元)
-            - open: 今开
-            - high: 最高
-            - low: 最低
-            - prev_close: 昨收
-        """
-        pass

akshare_one-0.2.0/akshare_one.egg-info/requires.txt

@@ -1,2 +0,0 @@
-akshare>=1.16.94
-cachetools>=5.5.2