akshare-one 0.3.3__tar.gz → 0.3.5__tar.gz

{akshare_one-0.3.3 → akshare_one-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-one
-Version: 0.3.3
+Version: 0.3.5
 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,7 +9,7 @@ Keywords: akshare,financial-data,stock-data,quant
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: akshare>=1.17.20
+Requires-Dist: akshare>=1.17.21
 Requires-Dist: cachetools>=5.5.2
 Provides-Extra: talib
 Requires-Dist: ta-lib>=0.6.4; extra == "talib"
@@ -40,7 +40,9 @@ Dynamic: license-file
 | Stock news | `get_news_data` |
 | Financial data | `get_balance_sheet`/`get_income_statement`/`get_cash_flow` |
 | Internal transactions | `get_inner_trade_data` |
-| Technical indicators | See [indicators.py](akshare_one/indicators.py) |
+| Basic stock info | `get_basic_info` |
+| Financial metrics | `get_financial_metrics` |
+| Technical indicators | See [indicators.py](akshare-one/indicators.py) |
 
 ## 📦 Quick Installation
 
@@ -67,4 +69,6 @@ df_sma = get_sma(df, window=20)
 
 ## 📚 Documentation
 
-Detailed API reference: [docs/api.md](docs/api.md)
+Full API documentation is now available on GitHub Pages:
+
+https://zwldarren.github.io/akshare-one/

{akshare_one-0.3.3 → akshare_one-0.3.5}/README.md

@@ -23,7 +23,9 @@
 | Stock news | `get_news_data` |
 | Financial data | `get_balance_sheet`/`get_income_statement`/`get_cash_flow` |
 | Internal transactions | `get_inner_trade_data` |
-| Technical indicators | See [indicators.py](akshare_one/indicators.py) |
+| Basic stock info | `get_basic_info` |
+| Financial metrics | `get_financial_metrics` |
+| Technical indicators | See [indicators.py](akshare-one/indicators.py) |
 
 ## 📦 Quick Installation
 
@@ -50,4 +52,6 @@ df_sma = get_sma(df, window=20)
 
 ## 📚 Documentation
 
-Detailed API reference: [docs/api.md](docs/api.md)
+Full API documentation is now available on GitHub Pages:
+
+https://zwldarren.github.io/akshare-one/

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/__init__.py

@@ -71,7 +71,7 @@ def get_hist_data(
 
     Returns:
         pd.DataFrame:
-        - timestamp: 时间戳(UTC时区)
+        - timestamp: 时间戳
         - open: 开盘价
         - high: 最高价
         - low: 最低价
@@ -92,7 +92,7 @@ def get_hist_data(
 
 def get_realtime_data(
     symbol: Optional[str] = None,
-    source: Literal["eastmoney", "eastmoney_direct", "xueqiu"] = "xueqiu",
+    source: Literal["eastmoney", "eastmoney_direct", "xueqiu"] = "eastmoney_direct",
 ) -> pd.DataFrame:
     """Get real-time market quotes
 
@@ -183,13 +183,13 @@ def get_cash_flow(symbol: str, source: Literal["sina"] = "sina") -> pd.DataFrame
 
 
 def get_financial_metrics(
-    symbol: str, source: Literal["eastmoney"] = "eastmoney"
+    symbol: str, source: Literal["eastmoney_direct"] = "eastmoney_direct"
 ) -> pd.DataFrame:
     """获取三大财务报表关键指标
 
     Args:
         symbol: 股票代码 (如 "600600")
-        source: 数据源 ('eastmoney')
+        source: 数据源 ('eastmoney_direct')
 
     Returns:
         pd.DataFrame: 财务关键指标数据

akshare_one-0.3.5/akshare_one/modules/cache.py

@@ -0,0 +1,27 @@
+from cachetools import TTLCache, cached
+import os
+
+# 缓存配置
+CACHE_CONFIG = {
+    "hist_data_cache": TTLCache(maxsize=1000, ttl=3600),  # 历史数据缓存1小时
+    "realtime_cache": TTLCache(maxsize=500, ttl=60),  # 实时数据缓存1分钟
+    "news_cache": TTLCache(maxsize=500, ttl=3600),  # 新闻数据缓存1小时
+    "financial_cache": TTLCache(maxsize=500, ttl=86400),  # 财务数据缓存24小时
+    "info_cache": TTLCache(maxsize=500, ttl=86400),  # 信息数据缓存24小时
+}
+
+
+def cache(cache_key, key=None):
+    cache_enabled = os.getenv("AKSHARE_ONE_CACHE_ENABLED", "true").lower() in (
+        "1",
+        "true",
+        "yes",
+        "on",
+    )
+
+    def decorator(func):
+        if cache_enabled:
+            return cached(CACHE_CONFIG[cache_key], key=key)(func)
+        return func
+
+    return decorator

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/financial/factory.py

@@ -1,4 +1,4 @@
-from .eastmoney import EastMoneyFinancialReport
+from .eastmoney_direct import EastMoneyDirectFinancialReport
 from .sina import SinaFinancialReport
 from .base import FinancialDataProvider
 
@@ -10,7 +10,7 @@ class FinancialDataFactory:
 
     _providers = {
         "sina": SinaFinancialReport,
-        "eastmoney": EastMoneyFinancialReport,
+        "eastmoney_direct": EastMoneyDirectFinancialReport,
     }
 
     @classmethod

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/financial/sina.py

@@ -1,8 +1,7 @@
-from cachetools import cached
 import pandas as pd
 import akshare as ak
 
-from akshare_one.modules.cache import CACHE_CONFIG
+from ..cache import cache
 from .base import FinancialDataProvider
 
 
@@ -19,9 +18,8 @@ class SinaFinancialReport(FinancialDataProvider):
             f"sh{symbol}" if not symbol.startswith(("sh", "sz", "bj")) else symbol
         )
 
-    @cached(
-        CACHE_CONFIG["financial_cache"],
-        key=lambda self, symbol=None: f"sina_balance_{self.symbol}",
+    @cache(
+        "financial_cache", key=lambda self, symbol=None: f"sina_balance_{self.symbol}"
     )
     def get_balance_sheet(self) -> pd.DataFrame:
         """获取资产负债表数据
@@ -35,9 +33,8 @@ class SinaFinancialReport(FinancialDataProvider):
         raw_df = ak.stock_financial_report_sina(stock=self.stock, symbol="资产负债表")
         return self._clean_balance_data(raw_df)
 
-    @cached(
-        CACHE_CONFIG["financial_cache"],
-        key=lambda self, symbol=None: f"sina_income_{self.symbol}",
+    @cache(
+        "financial_cache", key=lambda self, symbol=None: f"sina_income_{self.symbol}"
     )
     def get_income_statement(self) -> pd.DataFrame:
         """获取利润表数据
@@ -51,10 +48,7 @@ class SinaFinancialReport(FinancialDataProvider):
         raw_df = ak.stock_financial_report_sina(stock=self.stock, symbol="利润表")
         return self._clean_income_data(raw_df)
 
-    @cached(
-        CACHE_CONFIG["financial_cache"],
-        key=lambda self, symbol=None: f"sina_cash_{self.symbol}",
-    )
+    @cache("financial_cache", key=lambda self, symbol=None: f"sina_cash_{self.symbol}")
     def get_cash_flow(self) -> pd.DataFrame:
         """获取现金流量表数据
 

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/historical/base.py

@@ -37,7 +37,7 @@ class HistoricalDataProvider(ABC):
 
         Returns:
             pd.DataFrame:
-            - timestamp (UTC)
+            - timestamp
             - open
             - high
             - low

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/historical/eastmoney.py

@@ -1,15 +1,14 @@
-from cachetools import cached
 from .base import HistoricalDataProvider
 import akshare as ak
 import pandas as pd
-from ..cache import CACHE_CONFIG
+from ..cache import cache
 
 
 class EastMoneyHistorical(HistoricalDataProvider):
     """Adapter for EastMoney historical stock data API"""
 
-    @cached(
-        cache=CACHE_CONFIG["hist_data_cache"],
+    @cache(
+        "hist_data_cache",
         key=lambda self: f"eastmoney_hist_{self.symbol}_{self.interval}_{self.interval_multiplier}_{self.adjust}",
     )
     def get_hist_data(self) -> pd.DataFrame:
@@ -157,7 +156,7 @@ class EastMoneyHistorical(HistoricalDataProvider):
         return resampled.reset_index()
 
     def _clean_minute_data(self, raw_df: pd.DataFrame, period: str) -> pd.DataFrame:
-        """Cleans and standardizes minute/hour level data, converting timestamps to UTC"""
+        """Cleans and standardizes minute/hour level data"""
         column_map = {
             "1": {
                 "时间": "timestamp",
@@ -188,16 +187,14 @@ class EastMoneyHistorical(HistoricalDataProvider):
         df = raw_df.rename(columns=mapping)
 
         if "timestamp" in df.columns:
-            df["timestamp"] = (
-                pd.to_datetime(df["timestamp"])
-                .dt.tz_localize("Asia/Shanghai")
-                .dt.tz_convert("UTC")
+            df["timestamp"] = pd.to_datetime(df["timestamp"]).dt.tz_localize(
+                "Asia/Shanghai"
             )
 
         return self._select_standard_columns(df)
 
     def _clean_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
-        """Cleans and standardizes daily and higher-level data, converting timestamps to UTC"""
+        """Cleans and standardizes daily and higher-level data"""
         column_map = {
             "日期": "timestamp",
             "开盘": "open",
@@ -217,10 +214,8 @@ class EastMoneyHistorical(HistoricalDataProvider):
         df = raw_df.rename(columns=available_columns)
 
         if "timestamp" in df.columns:
-            df["timestamp"] = (
-                pd.to_datetime(df["timestamp"])
-                .dt.tz_localize("Asia/Shanghai")
-                .dt.tz_convert("UTC")
+            df["timestamp"] = pd.to_datetime(df["timestamp"]).dt.tz_localize(
+                "Asia/Shanghai"
             )
 
         if "volume" in df.columns:

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/historical/eastmoney_direct.py

@@ -1,9 +1,8 @@
 import pandas as pd
-from cachetools import cached
 from .base import HistoricalDataProvider
-from ..cache import CACHE_CONFIG
-from ..eastmoney.client import EastMoneyClient
-from ..eastmoney.utils import parse_kline_data, resample_historical_data
+from ..cache import cache
+from akshare_one.eastmoney.client import EastMoneyClient
+from akshare_one.eastmoney.utils import parse_kline_data, resample_historical_data
 
 
 class EastMoneyDirectHistorical(HistoricalDataProvider):
@@ -13,8 +12,8 @@ class EastMoneyDirectHistorical(HistoricalDataProvider):
         super().__init__(*args, **kwargs)
         self.client = EastMoneyClient()
 
-    @cached(
-        cache=CACHE_CONFIG["hist_data_cache"],
+    @cache(
+        "hist_data_cache",
         key=lambda self: f"eastmoney_direct_hist_{self.symbol}_{self.interval}_{self.interval_multiplier}_{self.adjust}",
     )
     def get_hist_data(self) -> pd.DataFrame:

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/historical/sina.py

@@ -195,7 +195,7 @@ class SinaHistorical(HistoricalDataProvider):
         return resampled.reset_index()
 
     def _clean_minute_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
-        """Cleans and standardizes minute/hour level data, converting timestamps to UTC"""
+        """Cleans and standardizes minute/hour level data"""
         column_map = {
             "date": "timestamp",
             "open": "open",
@@ -208,16 +208,14 @@ class SinaHistorical(HistoricalDataProvider):
         df = raw_df.rename(columns=column_map)
 
         if "timestamp" in df.columns:
-            df["timestamp"] = (
-                pd.to_datetime(df["timestamp"])
-                .dt.tz_localize("Asia/Shanghai")
-                .dt.tz_convert("UTC")
+            df["timestamp"] = pd.to_datetime(df["timestamp"]).dt.tz_localize(
+                "Asia/Shanghai"
             )
 
         return self._select_standard_columns(df)
 
     def _clean_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
-        """Cleans and standardizes daily and higher-level data, converting timestamps to UTC"""
+        """Cleans and standardizes daily and higher-level data"""
         column_map = {
             "date": "timestamp",
             "open": "open",
@@ -230,10 +228,8 @@ class SinaHistorical(HistoricalDataProvider):
         df = raw_df.rename(columns=column_map)
 
         if "timestamp" in df.columns:
-            df["timestamp"] = (
-                pd.to_datetime(df["timestamp"])
-                .dt.tz_localize("Asia/Shanghai")
-                .dt.tz_convert("UTC")
+            df["timestamp"] = pd.to_datetime(df["timestamp"]).dt.tz_localize(
+                "Asia/Shanghai"
             )
 
         if "volume" in df.columns:

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/info/eastmoney.py

@@ -1,8 +1,7 @@
-from cachetools import cached
 import pandas as pd
 import akshare as ak
 
-from ..cache import CACHE_CONFIG
+from ..cache import cache
 from .base import InfoDataProvider
 
 
@@ -19,8 +18,8 @@ class EastmoneyInfo(InfoDataProvider):
         "上市时间": "listing_date",
     }
 
-    @cached(
-        CACHE_CONFIG["info_cache"],
+    @cache(
+        "info_cache",
         key=lambda self, symbol=None: f"eastmoney_{symbol}",
     )
     def get_basic_info(self) -> pd.DataFrame:

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/insider/base.py

@@ -16,7 +16,7 @@ class InsiderDataProvider(ABC):
             - issuer: 股票名称
             - name: 变动人
             - title: 董监高职务
-            - transaction_date: 变动日期(UTC时区)
+            - transaction_date: 变动日期
             - transaction_shares: 变动股数
             - transaction_price_per_share: 成交均价
             - shares_owned_after_transaction: 变动后持股数

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/insider/xueqiu.py

@@ -1,16 +1,15 @@
-from cachetools import cached
 import pandas as pd
 import akshare as ak
 from .base import InsiderDataProvider
 from ..utils import convert_xieqiu_symbol
-from ..cache import CACHE_CONFIG
+from ..cache import cache
 
 
 class XueQiuInsider(InsiderDataProvider):
     """Provider for XueQiu insider trading data"""
 
-    @cached(
-        cache=CACHE_CONFIG["financial_cache"],
+    @cache(
+        "financial_cache",
         key=lambda self, symbol=None: f"xueqiu_insider_{symbol if symbol else 'all'}",
     )
     def get_inner_trade_data(self) -> pd.DataFrame:
@@ -87,17 +86,13 @@ class XueQiuInsider(InsiderDataProvider):
 
         # Convert date format
         if "transaction_date" in df.columns:
-            df["transaction_date"] = (
-                pd.to_datetime(df["transaction_date"])
-                .dt.tz_localize("Asia/Shanghai")
-                .dt.tz_convert("UTC")
-            )
+            df["transaction_date"] = pd.to_datetime(
+                df["transaction_date"]
+            ).dt.tz_localize("Asia/Shanghai")
 
         if "filing_date" in df.columns:
-            df["filing_date"] = (
-                pd.to_datetime(df["filing_date"])
-                .dt.tz_localize("Asia/Shanghai")
-                .dt.tz_convert("UTC")
+            df["filing_date"] = pd.to_datetime(df["filing_date"]).dt.tz_localize(
+                "Asia/Shanghai"
             )
 
         # Convert numeric columns

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/news/base.py

@@ -15,7 +15,7 @@ class NewsDataProvider(ABC):
             - keyword: 关键词
             - title: 新闻标题
             - content: 新闻内容
-            - publish_time: 发布时间 (UTC)
+            - publish_time: 发布时间
             - source: 文章来源
             - url: 新闻链接
         """

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/news/eastmoney.py

@@ -1,14 +1,13 @@
-from cachetools import cached
 import pandas as pd
 import akshare as ak
 
-from ..cache import CACHE_CONFIG
+from ..cache import cache
 from .base import NewsDataProvider
 
 
 class EastMoneyNews(NewsDataProvider):
-    @cached(
-        CACHE_CONFIG["news_cache"],
+    @cache(
+        "news_cache",
         key=lambda self: f"eastmoney_news_{self.symbol}",
     )
     def get_news_data(self) -> pd.DataFrame:
@@ -29,11 +28,8 @@ class EastMoneyNews(NewsDataProvider):
 
         df = raw_df.rename(columns=column_mapping)
 
-        # Convert time to UTC
-        df["publish_time"] = (
-            pd.to_datetime(df["publish_time"])
-            .dt.tz_localize("Asia/Shanghai")
-            .dt.tz_convert("UTC")
+        df["publish_time"] = pd.to_datetime(df["publish_time"]).dt.tz_localize(
+            "Asia/Shanghai"
         )
 
         required_columns = [

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/realtime/eastmoney.py

@@ -1,14 +1,13 @@
-from cachetools import cached
 import pandas as pd
 import akshare as ak
 
-from ..cache import CACHE_CONFIG
+from ..cache import cache
 from .base import RealtimeDataProvider
 
 
 class EastmoneyRealtime(RealtimeDataProvider):
-    @cached(
-        CACHE_CONFIG["realtime_cache"],
+    @cache(
+        "realtime_cache",
         key=lambda self, symbol=None: f"eastmoney_{symbol if symbol else 'all'}",
     )
     def get_current_data(self) -> pd.DataFrame:
@@ -36,10 +35,7 @@ class EastmoneyRealtime(RealtimeDataProvider):
 
         df = raw_df.rename(columns=column_mapping)
 
-        # Change time to UTC
-        df = df.assign(
-            timestamp=lambda x: pd.Timestamp.now(tz="Asia/Shanghai").tz_convert("UTC")
-        )
+        df = df.assign(timestamp=lambda x: pd.Timestamp.now(tz="Asia/Shanghai"))
 
         required_columns = [
             "symbol",

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/realtime/eastmoney_direct.py

@@ -1,9 +1,8 @@
 import pandas as pd
-from cachetools import cached
 from .base import RealtimeDataProvider
-from ..cache import CACHE_CONFIG
-from ..eastmoney.client import EastMoneyClient
-from ..eastmoney.utils import parse_realtime_data
+from ..cache import cache
+from akshare_one.eastmoney.client import EastMoneyClient
+from akshare_one.eastmoney.utils import parse_realtime_data
 
 
 class EastMoneyDirectRealtime(RealtimeDataProvider):
@@ -13,8 +12,8 @@ class EastMoneyDirectRealtime(RealtimeDataProvider):
         super().__init__(symbol)
         self.client = EastMoneyClient()
 
-    @cached(
-        cache=CACHE_CONFIG["realtime_cache"],
+    @cache(
+        "realtime_cache",
         key=lambda self: f"eastmoney_direct_realtime_{self.symbol}",
     )
     def get_current_data(self) -> pd.DataFrame:

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one/modules/realtime/xueqiu.py

@@ -1,14 +1,13 @@
-from cachetools import cached
 import pandas as pd
 import akshare as ak
 from ..utils import convert_xieqiu_symbol
-from ..cache import CACHE_CONFIG
+from ..cache import cache
 from .base import RealtimeDataProvider
 
 
 class XueQiuRealtime(RealtimeDataProvider):
-    @cached(
-        cache=CACHE_CONFIG["realtime_cache"],
+    @cache(
+        "realtime_cache",
         key=lambda self, symbol=None: f"xueqiu_{symbol}",
     )
     def get_current_data(self) -> pd.DataFrame:
@@ -43,9 +42,7 @@ class XueQiuRealtime(RealtimeDataProvider):
             ),
             "timestamp": pd.to_datetime(
                 raw_df.loc[raw_df["item"] == "时间", "value"].values[0]
-            )
-            .tz_localize("Asia/Shanghai")
-            .tz_convert("UTC"),
+            ).tz_localize("Asia/Shanghai"),
             "volume": int(raw_df.loc[raw_df["item"] == "成交量", "value"].values[0])
             / 100,
             "amount": float(raw_df.loc[raw_df["item"] == "成交额", "value"].values[0]),

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-one
-Version: 0.3.3
+Version: 0.3.5
 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,7 +9,7 @@ Keywords: akshare,financial-data,stock-data,quant
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: akshare>=1.17.20
+Requires-Dist: akshare>=1.17.21
 Requires-Dist: cachetools>=5.5.2
 Provides-Extra: talib
 Requires-Dist: ta-lib>=0.6.4; extra == "talib"
@@ -40,7 +40,9 @@ Dynamic: license-file
 | Stock news | `get_news_data` |
 | Financial data | `get_balance_sheet`/`get_income_statement`/`get_cash_flow` |
 | Internal transactions | `get_inner_trade_data` |
-| Technical indicators | See [indicators.py](akshare_one/indicators.py) |
+| Basic stock info | `get_basic_info` |
+| Financial metrics | `get_financial_metrics` |
+| Technical indicators | See [indicators.py](akshare-one/indicators.py) |
 
 ## 📦 Quick Installation
 
@@ -67,4 +69,6 @@ df_sma = get_sma(df, window=20)
 
 ## 📚 Documentation
 
-Detailed API reference: [docs/api.md](docs/api.md)
+Full API documentation is now available on GitHub Pages:
+
+https://zwldarren.github.io/akshare-one/

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one.egg-info/SOURCES.txt

@@ -10,10 +10,7 @@ akshare_one.egg-info/requires.txt
 akshare_one.egg-info/top_level.txt
 akshare_one/modules/cache.py
 akshare_one/modules/utils.py
-akshare_one/modules/eastmoney/client.py
-akshare_one/modules/eastmoney/utils.py
 akshare_one/modules/financial/base.py
-akshare_one/modules/financial/eastmoney.py
 akshare_one/modules/financial/factory.py
 akshare_one/modules/financial/sina.py
 akshare_one/modules/historical/base.py

{akshare_one-0.3.3 → akshare_one-0.3.5}/akshare_one.egg-info/requires.txt

@@ -1,4 +1,4 @@
-akshare>=1.17.20
+akshare>=1.17.21
 cachetools>=5.5.2
 
 [talib]

{akshare_one-0.3.3 → akshare_one-0.3.5}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "akshare-one"
-version = "0.3.3"
+version = "0.3.5"
 description = "Standardized interface for Chinese financial market data, built on AKShare with unified data formats and simplified APIs"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "akshare>=1.17.20",
+    "akshare>=1.17.21",
     "cachetools>=5.5.2",
 ]
 license = "MIT"
@@ -22,12 +22,16 @@ talib = [
 
 [dependency-groups]
 dev = [
+    "mkdocs-material>=9.6.15",
     "pre-commit>=4.2.0",
     "pytest>=8.4.1",
     "pytest-cov>=6.2.1",
     "ruff>=0.12.1",
 ]
 
+[tool.setuptools]
+packages = ["akshare_one"]
+
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 python_files = "test_*.py"

akshare_one-0.3.5/tests/test_info.py

@@ -0,0 +1,82 @@
+import unittest
+from cachetools import TTLCache
+import pandas as pd
+import os
+import time
+from akshare_one import get_basic_info
+from akshare_one.modules.cache import CACHE_CONFIG
+
+
+class TestInfo(unittest.TestCase):
+    def test_get_info(self):
+        """测试获取股票基本信息"""
+        df = get_basic_info("600405")
+        self.assertIsInstance(df, pd.DataFrame)
+        self.assertFalse(df.empty)
+        self.assertEqual(df.shape[0], 1)
+
+        expected_columns = [
+            "price",
+            "symbol",
+            "name",
+            "total_shares",
+            "float_shares",
+            "total_market_cap",
+            "float_market_cap",
+            "industry",
+            "listing_date",
+        ]
+        for col in expected_columns:
+            self.assertIn(col, df.columns)
+
+        self.assertEqual(df["symbol"].iloc[0], "600405")
+        self.assertIsInstance(df["listing_date"].iloc[0], pd.Timestamp)
+
+    def test_cache(self):
+        """测试缓存功能是否生效"""
+        cache = CACHE_CONFIG["info_cache"]
+        cache.clear()
+
+        # 测试缓存命中
+        os.environ["AKSHARE_ONE_CACHE_ENABLED"] = "true"
+
+        # 第一次调用 - 应该缓存未命中
+        initial_size = cache.currsize
+        df1 = get_basic_info("600405")
+        self.assertEqual(cache.currsize, initial_size + 1)  # 缓存应增加
+
+        # 第二次调用 - 应该缓存命中
+        df2 = get_basic_info("600405")
+        self.assertEqual(cache.currsize, initial_size + 1)  # 缓存大小不变
+        pd.testing.assert_frame_equal(df1, df2)
+
+        # 测试缓存禁用
+        os.environ["AKSHARE_ONE_CACHE_ENABLED"] = "false"
+        disabled_size = cache.currsize
+        get_basic_info("600405")
+        self.assertEqual(cache.currsize, disabled_size)
+
+        # 测试缓存过期
+        os.environ["AKSHARE_ONE_CACHE_ENABLED"] = "true"
+
+        # 创建临时缓存并替换原缓存
+        original_cache = cache
+        temp_cache = TTLCache(maxsize=1000, ttl=1)
+        CACHE_CONFIG["info_cache"] = temp_cache
+
+        # 填充缓存
+        get_basic_info("600405")
+        expired_size = temp_cache.currsize
+
+        # 等待缓存过期
+        time.sleep(1.1)
+
+        # 过期后调用 - 应该缓存未命中
+        get_basic_info("600405")
+        self.assertEqual(temp_cache.currsize, expired_size)  # 缓存被替换，大小不变
+
+        CACHE_CONFIG["info_cache"] = original_cache
+
+
+if __name__ == "__main__":
+    unittest.main()

akshare_one-0.3.3/akshare_one/modules/cache.py

@@ -1,10 +0,0 @@
-from cachetools import TTLCache
-
-# 缓存配置
-CACHE_CONFIG = {
-    "hist_data_cache": TTLCache(maxsize=1000, ttl=3600),  # 历史数据缓存1小时
-    "realtime_cache": TTLCache(maxsize=500, ttl=60),  # 实时数据缓存1分钟
-    "news_cache": TTLCache(maxsize=500, ttl=3600),
-    "financial_cache": TTLCache(maxsize=500, ttl=86400),  # 财务数据缓存24小时
-    "info_cache": TTLCache(maxsize=500, ttl=86400),  # 信息数据缓存24小时
-}

akshare_one-0.3.3/akshare_one/modules/eastmoney/client.py

@@ -1,88 +0,0 @@
-import requests
-from typing import Dict, Any
-
-
-class EastMoneyClient:
-    """
-    A client for interacting directly with EastMoney's data APIs.
-    This class handles session management, request signing, and API calls.
-    """
-
-    def __init__(self):
-        self.session = requests.Session()
-        self.session.headers.update(
-            {
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/132.0.0.0 Safari/537.36",
-                "Referer": "https://quote.eastmoney.com/",
-                "Accept": "application/json, text/plain, */*",
-                "Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8",
-            }
-        )
-
-    def _get_security_id(self, symbol: str) -> str:
-        """
-        Converts a stock symbol to EastMoney's internal secid format.
-        e.g., '600519' -> '1.600519', '000001' -> '0.000001'
-        """
-        symbol = symbol.upper()
-        if symbol.startswith("SZ"):
-            market = "0"
-            code = symbol[2:]
-        elif symbol.startswith("SH"):
-            market = "1"
-            code = symbol[2:]
-        elif symbol.startswith("HK"):
-            market = "116"
-            code = symbol[2:]
-        elif len(symbol) == 6:
-            if symbol.startswith(("000", "001", "002", "003", "300", "200")):
-                market = "0"
-            elif symbol.startswith(("600", "601", "603", "605", "688", "900")):
-                market = "1"
-            else:
-                market = "0"  # Default to SZ for ambiguity
-            code = symbol
-        elif len(symbol) == 5:  # HK Market
-            market = "116"
-            code = symbol
-        else:
-            market = "0"
-            code = symbol
-        return f"{market}.{code}"
-
-    def fetch_historical_klines(
-        self, symbol: str, klt: str, fqt: str, start_date: str, end_date: str
-    ) -> Dict[str, Any]:
-        """
-        Fetches historical K-line (candlestick) data.
-        """
-        url = "https://push2his.eastmoney.com/api/qt/stock/kline/get"
-        secid = self._get_security_id(symbol)
-        params = {
-            "fields1": "f1,f2,f3,f4,f5,f6",
-            "fields2": "f51,f52,f53,f54,f55,f56,f57,f58,f59,f60,f61",
-            "klt": klt,
-            "fqt": fqt,
-            "secid": secid,
-            "beg": start_date,
-            "end": end_date,
-        }
-        response = self.session.get(url, params=params)
-        response.raise_for_status()
-        return response.json()
-
-    def fetch_realtime_quote(self, symbol: str) -> Dict[str, Any]:
-        """
-        Fetches real-time quote data for a single stock.
-        """
-        url = "https://push2.eastmoney.com/api/qt/stock/get"
-        secid = self._get_security_id(symbol)
-        params = {
-            "invt": "2",
-            "fltt": "2",
-            "fields": "f43,f57,f58,f169,f170,f46,f60,f44,f51,f168,f47,f164,f163,f116,f60,f45,f52,f50,f48,f167,f117,f71,f161,f49,f530",
-            "secid": secid,
-        }
-        response = self.session.get(url, params=params)
-        response.raise_for_status()
-        return response.json()

akshare_one-0.3.3/akshare_one/modules/eastmoney/utils.py

@@ -1,104 +0,0 @@
-import pandas as pd
-from typing import Dict, Any
-
-
-def parse_kline_data(data: Dict[str, Any]) -> pd.DataFrame:
-    """
-    Parses K-line data from the API response into a pandas DataFrame.
-    """
-    klines = data.get("data", {}).get("klines", [])
-    if not klines:
-        return pd.DataFrame(
-            columns=["timestamp", "open", "high", "low", "close", "volume"]
-        )
-
-    records = []
-    for kline in klines:
-        parts = kline.split(",")
-        if len(parts) >= 6:
-            records.append(
-                {
-                    "timestamp": parts[0],
-                    "open": float(parts[1]),
-                    "close": float(parts[2]),
-                    "high": float(parts[3]),
-                    "low": float(parts[4]),
-                    "volume": int(parts[5]),
-                }
-            )
-
-    df = pd.DataFrame(records)
-    if not df.empty:
-        df["timestamp"] = pd.to_datetime(df["timestamp"])
-        df["timestamp"] = (
-            df["timestamp"].dt.tz_localize("Asia/Shanghai").dt.tz_convert("UTC")
-        )
-        df = df[["timestamp", "open", "high", "low", "close", "volume"]]
-    return df
-
-
-def parse_realtime_data(data: Dict[str, Any]) -> pd.DataFrame:
-    """
-    Parses real-time quote data from the API response into a pandas DataFrame.
-    """
-    stock_data = data.get("data")
-    if not stock_data:
-        return pd.DataFrame()
-
-    df = pd.DataFrame(
-        [
-            {
-                "symbol": stock_data.get("f57"),
-                "price": stock_data.get("f43"),
-                "change": stock_data.get("f169"),
-                "pct_change": stock_data.get("f170"),
-                "volume": stock_data.get("f47"),
-                "amount": stock_data.get("f48"),
-                "open": stock_data.get("f46"),
-                "high": stock_data.get("f44"),
-                "low": stock_data.get("f45"),
-                "prev_close": stock_data.get("f60"),
-            }
-        ]
-    )
-    df["timestamp"] = pd.Timestamp.now(tz="Asia/Shanghai").tz_convert("UTC")
-    return df
-
-
-def resample_historical_data(
-    df: pd.DataFrame, interval: str, multiplier: int
-) -> pd.DataFrame:
-    """
-    Resamples historical data to a specified frequency.
-    """
-    if df.empty or multiplier <= 1:
-        return df
-
-    df = df.set_index("timestamp")
-
-    freq_map = {
-        "day": f"{multiplier}D",
-        "week": f"{multiplier}W-MON",
-        "month": f"{multiplier}MS",
-        "year": f"{multiplier * 12}MS",
-    }
-    freq = freq_map.get(interval)
-
-    if not freq:
-        return df.reset_index()
-
-    resampled = (
-        df.resample(freq)
-        .agg(
-            {
-                "open": "first",
-                "high": "max",
-                "low": "min",
-                "close": "last",
-                "volume": "sum",
-            }
-        )
-        .dropna()
-    )
-
-    return resampled.reset_index()

akshare_one-0.3.3/akshare_one/modules/financial/eastmoney.py

@@ -1,184 +0,0 @@
-from cachetools import cached
-import pandas as pd
-import requests
-
-from akshare_one.modules.cache import CACHE_CONFIG
-from .base import FinancialDataProvider
-
-
-class EastMoneyFinancialReport(FinancialDataProvider):
-    _balance_sheet_rename_map = {
-        "REPORT_DATE": "report_date",
-        "TOTAL_ASSETS": "total_assets",
-        "FIXED_ASSET": "fixed_assets_net",
-        "MONETARYFUNDS": "cash_and_equivalents",
-        "ACCOUNTS_RECE": "accounts_receivable",
-        "INVENTORY": "inventory",
-        "TOTAL_LIABILITIES": "total_liabilities",
-        "ACCOUNTS_PAYABLE": "trade_and_non_trade_payables",
-        "ADVANCE_RECEIVABLES": "deferred_revenue",
-        "TOTAL_EQUITY": "shareholders_equity",
-    }
-
-    _income_statement_rename_map = {
-        "REPORT_DATE": "report_date",
-        "TOTAL_OPERATE_INCOME": "revenue",
-        "TOTAL_OPERATE_COST": "total_operating_costs",
-        "OPERATE_PROFIT": "operating_profit",
-        "PARENT_NETPROFIT": "net_income_common_stock",
-    }
-
-    _cash_flow_rename_map = {
-        "REPORT_DATE": "report_date",
-        "NETCASH_OPERATE": "net_cash_flow_from_operations",
-        "NETCASH_INVEST": "net_cash_flow_from_investing",
-        "NETCASH_FINANCE": "net_cash_flow_from_financing",
-        "CCE_ADD": "change_in_cash_and_equivalents",
-    }
-
-    def __init__(self, symbol):
-        super().__init__(symbol)
-
-    def get_income_statement(self):
-        pass
-
-    def get_balance_sheet(self):
-        pass
-
-    def get_cash_flow(self):
-        pass
-
-    @cached(
-        CACHE_CONFIG["financial_cache"],
-        key=lambda self, symbol=None: f"financial_metrics_{self.symbol}",
-    )
-    def get_financial_metrics(self) -> pd.DataFrame:
-        """获取三大财务报表关键指标"""
-        balance_sheet = self._fetch_balance_sheet()
-        income_statement = self._fetch_income_statement()
-        cash_flow = self._fetch_cash_flow()
-
-        if balance_sheet.empty and income_statement.empty and cash_flow.empty:
-            return pd.DataFrame()
-
-        merged = pd.merge(
-            balance_sheet, income_statement, on="report_date", how="outer"
-        )
-        merged = pd.merge(merged, cash_flow, on="report_date", how="outer")
-
-        # Convert report_date to datetime and format as YYYY-MM-DD
-        merged["report_date"] = pd.to_datetime(merged["report_date"]).dt.strftime(
-            "%Y-%m-%d"
-        )
-
-        # Sort by report_date in descending order (most recent first)
-        merged = merged.sort_values("report_date", ascending=False).reset_index(
-            drop=True
-        )
-
-        return merged
-
-    def _fetch_balance_sheet(self) -> pd.DataFrame:
-        """
-        Get stock balance sheet data from East Money API
-        """
-        try:
-            # API endpoint and parameters
-            api_url = "https://datacenter-web.eastmoney.com/api/data/v1/get"
-            params = {
-                "reportName": "RPT_DMSK_FN_BALANCE",
-                "filter": f'(SECURITY_CODE="{self.symbol}")',
-                "pageNumber": "1",
-                "pageSize": "1000",
-                "sortColumns": "REPORT_DATE",
-                "sortTypes": "-1",
-                "columns": ",".join(self._balance_sheet_rename_map.keys()),
-            }
-
-            # Fetch data from API
-            response = requests.get(api_url, params=params)
-            response.raise_for_status()
-            data = response.json()
-
-            # Extract the actual data
-            if data.get("result") and data["result"].get("data"):
-                df = pd.DataFrame(data["result"]["data"])
-                df.rename(columns=self._balance_sheet_rename_map, inplace=True)
-                return df
-            else:
-                print("No balance sheet data found in API response")
-                return pd.DataFrame()
-
-        except Exception as e:
-            print(f"Error occurred: {str(e)}")
-            return pd.DataFrame()
-
-    def _fetch_income_statement(self) -> pd.DataFrame:
-        """
-        Get stock income statement data from East Money API
-        """
-        try:
-            # API endpoint and parameters
-            api_url = "https://datacenter-web.eastmoney.com/api/data/v1/get"
-            params = {
-                "reportName": "RPT_DMSK_FN_INCOME",
-                "filter": f'(SECURITY_CODE="{self.symbol}")',
-                "pageNumber": "1",
-                "pageSize": "1000",
-                "sortColumns": "REPORT_DATE",
-                "sortTypes": "-1",
-                "columns": ",".join(self._income_statement_rename_map.keys()),
-            }
-
-            # Fetch data from API
-            response = requests.get(api_url, params=params)
-            response.raise_for_status()
-            data = response.json()
-
-            # Extract the actual data
-            if data.get("result") and data["result"].get("data"):
-                df = pd.DataFrame(data["result"]["data"])
-                df.rename(columns=self._income_statement_rename_map, inplace=True)
-                return df
-            else:
-                print("No income statement data found in API response")
-                return pd.DataFrame()
-
-        except Exception as e:
-            print(f"Error occurred: {str(e)}")
-            return pd.DataFrame()
-
-    def _fetch_cash_flow(self) -> pd.DataFrame:
-        """
-        Get stock cash flow statement data from East Money API
-        """
-        try:
-            # API endpoint and parameters
-            api_url = "https://datacenter-web.eastmoney.com/api/data/v1/get"
-            params = {
-                "reportName": "RPT_DMSK_FN_CASHFLOW",
-                "filter": f'(SECURITY_CODE="{self.symbol}")',
-                "pageNumber": "1",
-                "pageSize": "1000",
-                "sortColumns": "REPORT_DATE",
-                "sortTypes": "-1",
-                "columns": ",".join(self._cash_flow_rename_map.keys()),
-            }
-
-            # Fetch data from API
-            response = requests.get(api_url, params=params)
-            response.raise_for_status()
-            data = response.json()
-
-            # Extract the actual data
-            if data.get("result") and data["result"].get("data"):
-                df = pd.DataFrame(data["result"]["data"])
-                df.rename(columns=self._cash_flow_rename_map, inplace=True)
-                return df
-            else:
-                print("No cash flow statement data found in API response")
-                return pd.DataFrame()
-
-        except Exception as e:
-            print(f"Error occurred: {str(e)}")
-            return pd.DataFrame()

akshare_one-0.3.3/tests/test_info.py

@@ -1,33 +0,0 @@
-import unittest
-import pandas as pd
-from akshare_one import get_basic_info
-
-
-class TestInfo(unittest.TestCase):
-    def test_get_info(self):
-        """测试获取股票基本信息"""
-        df = get_basic_info("600405")
-        self.assertIsInstance(df, pd.DataFrame)
-        self.assertFalse(df.empty)
-        self.assertEqual(df.shape[0], 1)
-
-        expected_columns = [
-            "price",
-            "symbol",
-            "name",
-            "total_shares",
-            "float_shares",
-            "total_market_cap",
-            "float_market_cap",
-            "industry",
-            "listing_date",
-        ]
-        for col in expected_columns:
-            self.assertIn(col, df.columns)
-
-        self.assertEqual(df["symbol"].iloc[0], "600405")
-        self.assertIsInstance(df["listing_date"].iloc[0], pd.Timestamp)
-
-
-if __name__ == "__main__":
-    unittest.main()