akshare-one 0.2.2__py3-none-any.whl → 0.3.0__py3-none-any.whl

akshare_one/modules/realtime/eastmoney.py

@@ -1,57 +1,57 @@
-from cachetools import cached
-import pandas as pd
-import akshare as ak
-
-from ..cache import CACHE_CONFIG
-from .base import RealtimeDataProvider
-
-
-class EastmoneyRealtime(RealtimeDataProvider):
-    @cached(
-        CACHE_CONFIG["realtime_cache"],
-        key=lambda self, symbol=None: f"eastmoney_{symbol if symbol else 'all'}",
-    )
-    def get_current_data(self) -> pd.DataFrame:
-        """获取沪深京A股实时行情数据"""
-        raw_df = ak.stock_zh_a_spot_em()
-        df = self._clean_spot_data(raw_df)
-        if self.symbol:
-            df = df[df["symbol"] == self.symbol].reset_index(drop=True)
-        return df
-
-    def _clean_spot_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
-        """清理和标准化实时行情数据"""
-        column_mapping = {
-            "代码": "symbol",
-            "最新价": "price",
-            "涨跌额": "change",
-            "涨跌幅": "pct_change",
-            "成交量": "volume",
-            "成交额": "amount",
-            "今开": "open",
-            "最高": "high",
-            "最低": "low",
-            "昨收": "prev_close",
-        }
-
-        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")
-        )
-
-        required_columns = [
-            "symbol",
-            "price",
-            "change",
-            "pct_change",
-            "timestamp",
-            "volume",
-            "amount",
-            "open",
-            "high",
-            "low",
-            "prev_close",
-        ]
-        return df[required_columns]
+from cachetools import cached
+import pandas as pd
+import akshare as ak
+
+from ..cache import CACHE_CONFIG
+from .base import RealtimeDataProvider
+
+
+class EastmoneyRealtime(RealtimeDataProvider):
+    @cached(
+        CACHE_CONFIG["realtime_cache"],
+        key=lambda self, symbol=None: f"eastmoney_{symbol if symbol else 'all'}",
+    )
+    def get_current_data(self) -> pd.DataFrame:
+        """获取沪深京A股实时行情数据"""
+        raw_df = ak.stock_zh_a_spot_em()
+        df = self._clean_spot_data(raw_df)
+        if self.symbol:
+            df = df[df["symbol"] == self.symbol].reset_index(drop=True)
+        return df
+
+    def _clean_spot_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
+        """清理和标准化实时行情数据"""
+        column_mapping = {
+            "代码": "symbol",
+            "最新价": "price",
+            "涨跌额": "change",
+            "涨跌幅": "pct_change",
+            "成交量": "volume",
+            "成交额": "amount",
+            "今开": "open",
+            "最高": "high",
+            "最低": "low",
+            "昨收": "prev_close",
+        }
+
+        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")
+        )
+
+        required_columns = [
+            "symbol",
+            "price",
+            "change",
+            "pct_change",
+            "timestamp",
+            "volume",
+            "amount",
+            "open",
+            "high",
+            "low",
+            "prev_close",
+        ]
+        return df[required_columns]

akshare_one/modules/realtime/eastmoney_direct.py

@@ -1,37 +1,37 @@
-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
-
-
-class EastMoneyDirectRealtime(RealtimeDataProvider):
-    """Direct implementation for EastMoney realtime stock data API"""
-
-    def __init__(self, symbol: str):
-        super().__init__(symbol)
-        self.client = EastMoneyClient()
-
-    @cached(
-        cache=CACHE_CONFIG["realtime_cache"],
-        key=lambda self: f"eastmoney_direct_realtime_{self.symbol}",
-    )
-    def get_current_data(self) -> pd.DataFrame:
-        """Get real-time stock data"""
-        try:
-            raw_data = self.client.fetch_realtime_quote(self.symbol)
-
-            if raw_data.get("rc") != 0:
-                raise ValueError(f"API returned error: {raw_data.get('msg')}")
-
-            df = parse_realtime_data(raw_data)
-
-            # Ensure the output matches the base class definition
-            if self.symbol:
-                df = df[df["symbol"] == self.symbol].reset_index(drop=True)
-
-            return df
-
-        except Exception as e:
-            raise ValueError(f"Failed to get real-time data for {self.symbol}: {e}")
+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
+
+
+class EastMoneyDirectRealtime(RealtimeDataProvider):
+    """Direct implementation for EastMoney realtime stock data API"""
+
+    def __init__(self, symbol: str):
+        super().__init__(symbol)
+        self.client = EastMoneyClient()
+
+    @cached(
+        cache=CACHE_CONFIG["realtime_cache"],
+        key=lambda self: f"eastmoney_direct_realtime_{self.symbol}",
+    )
+    def get_current_data(self) -> pd.DataFrame:
+        """Get real-time stock data"""
+        try:
+            raw_data = self.client.fetch_realtime_quote(self.symbol)
+
+            if raw_data.get("rc") != 0:
+                raise ValueError(f"API returned error: {raw_data.get('msg')}")
+
+            df = parse_realtime_data(raw_data)
+
+            # Ensure the output matches the base class definition
+            if self.symbol:
+                df = df[df["symbol"] == self.symbol].reset_index(drop=True)
+
+            return df
+
+        except Exception as e:
+            raise ValueError(f"Failed to get real-time data for {self.symbol}: {e}")

akshare_one/modules/realtime/factory.py

@@ -1,48 +1,48 @@
-from .eastmoney import EastmoneyRealtime
-from .xueqiu import XueQiuRealtime
-from .base import RealtimeDataProvider
-from .eastmoney_direct import EastMoneyDirectRealtime
-
-
-class RealtimeDataFactory:
-    """
-    Factory class for creating realtime data providers
-    """
-
-    _providers = {
-        "eastmoney": EastmoneyRealtime,
-        "xueqiu": XueQiuRealtime,
-        "eastmoney_direct": EastMoneyDirectRealtime,
-    }
-
-    @classmethod
-    def get_provider(cls, provider_name: str, **kwargs) -> RealtimeDataProvider:
-        """
-        Get a realtime data provider by name
-
-        Args:
-            provider_name: Name of the provider (e.g., 'eastmoney')
-            **kwargs: Additional arguments to pass to the provider's constructor
-
-        Returns:
-            RealtimeDataProvider: An instance of the requested provider
-
-        Raises:
-            ValueError: If the requested provider is not found
-        """
-        provider_class = cls._providers.get(provider_name.lower())
-        if not provider_class:
-            raise ValueError(f"Unknown realtime data provider: {provider_name}")
-
-        return provider_class(**kwargs)
-
-    @classmethod
-    def register_provider(cls, name: str, provider_class: type):
-        """
-        Register a new realtime data provider
-
-        Args:
-            name: Name to associate with this provider
-            provider_class: The provider class to register
-        """
-        cls._providers[name.lower()] = provider_class
+from .eastmoney import EastmoneyRealtime
+from .xueqiu import XueQiuRealtime
+from .base import RealtimeDataProvider
+from .eastmoney_direct import EastMoneyDirectRealtime
+
+
+class RealtimeDataFactory:
+    """
+    Factory class for creating realtime data providers
+    """
+
+    _providers = {
+        "eastmoney": EastmoneyRealtime,
+        "xueqiu": XueQiuRealtime,
+        "eastmoney_direct": EastMoneyDirectRealtime,
+    }
+
+    @classmethod
+    def get_provider(cls, provider_name: str, **kwargs) -> RealtimeDataProvider:
+        """
+        Get a realtime data provider by name
+
+        Args:
+            provider_name: Name of the provider (e.g., 'eastmoney')
+            **kwargs: Additional arguments to pass to the provider's constructor
+
+        Returns:
+            RealtimeDataProvider: An instance of the requested provider
+
+        Raises:
+            ValueError: If the requested provider is not found
+        """
+        provider_class = cls._providers.get(provider_name.lower())
+        if not provider_class:
+            raise ValueError(f"Unknown realtime data provider: {provider_name}")
+
+        return provider_class(**kwargs)
+
+    @classmethod
+    def register_provider(cls, name: str, provider_class: type):
+        """
+        Register a new realtime data provider
+
+        Args:
+            name: Name to associate with this provider
+            provider_class: The provider class to register
+        """
+        cls._providers[name.lower()] = provider_class

akshare_one/modules/realtime/xueqiu.py

@@ -1,60 +1,60 @@
-from cachetools import cached
-import pandas as pd
-import akshare as ak
-from ..utils import convert_xieqiu_symbol
-from ..cache import CACHE_CONFIG
-from .base import RealtimeDataProvider
-
-
-class XueQiuRealtime(RealtimeDataProvider):
-    @cached(
-        cache=CACHE_CONFIG["realtime_cache"],
-        key=lambda self, symbol=None: f"xueqiu_{symbol}",
-    )
-    def get_current_data(self) -> pd.DataFrame:
-        """获取雪球实时行情数据
-
-        Args:
-            symbol: 股票代码 ("600000")
-
-        Returns:
-            pd.DataFrame with columns:
-            - symbol: 股票代码
-            - price: 最新价
-            - change: 涨跌额
-            - pct_change: 涨跌幅(%)
-            - timestamp: 时间戳
-            - volume: 成交量(手)
-            - amount: 成交额(元)
-            - open: 今开
-            - high: 最高
-            - low: 最低
-            - prev_close: 昨收
-        """
-        raw_df = ak.stock_individual_spot_xq(symbol=convert_xieqiu_symbol(self.symbol))
-
-        # Transform to match standard format
-        data = {
-            "symbol": self.symbol,
-            "price": float(raw_df.loc[raw_df["item"] == "现价", "value"].values[0]),
-            "change": float(raw_df.loc[raw_df["item"] == "涨跌", "value"].values[0]),
-            "pct_change": float(
-                raw_df.loc[raw_df["item"] == "涨幅", "value"].values[0]
-            ),
-            "timestamp": pd.to_datetime(
-                raw_df.loc[raw_df["item"] == "时间", "value"].values[0]
-            )
-            .tz_localize("Asia/Shanghai")
-            .tz_convert("UTC"),
-            "volume": int(raw_df.loc[raw_df["item"] == "成交量", "value"].values[0])
-            / 100,
-            "amount": float(raw_df.loc[raw_df["item"] == "成交额", "value"].values[0]),
-            "open": float(raw_df.loc[raw_df["item"] == "今开", "value"].values[0]),
-            "high": float(raw_df.loc[raw_df["item"] == "最高", "value"].values[0]),
-            "low": float(raw_df.loc[raw_df["item"] == "最低", "value"].values[0]),
-            "prev_close": float(
-                raw_df.loc[raw_df["item"] == "昨收", "value"].values[0]
-            ),
-        }
-
-        return pd.DataFrame([data])
+from cachetools import cached
+import pandas as pd
+import akshare as ak
+from ..utils import convert_xieqiu_symbol
+from ..cache import CACHE_CONFIG
+from .base import RealtimeDataProvider
+
+
+class XueQiuRealtime(RealtimeDataProvider):
+    @cached(
+        cache=CACHE_CONFIG["realtime_cache"],
+        key=lambda self, symbol=None: f"xueqiu_{symbol}",
+    )
+    def get_current_data(self) -> pd.DataFrame:
+        """获取雪球实时行情数据
+
+        Args:
+            symbol: 股票代码 ("600000")
+
+        Returns:
+            pd.DataFrame with columns:
+            - symbol: 股票代码
+            - price: 最新价
+            - change: 涨跌额
+            - pct_change: 涨跌幅(%)
+            - timestamp: 时间戳
+            - volume: 成交量(手)
+            - amount: 成交额(元)
+            - open: 今开
+            - high: 最高
+            - low: 最低
+            - prev_close: 昨收
+        """
+        raw_df = ak.stock_individual_spot_xq(symbol=convert_xieqiu_symbol(self.symbol))
+
+        # Transform to match standard format
+        data = {
+            "symbol": self.symbol,
+            "price": float(raw_df.loc[raw_df["item"] == "现价", "value"].values[0]),
+            "change": float(raw_df.loc[raw_df["item"] == "涨跌", "value"].values[0]),
+            "pct_change": float(
+                raw_df.loc[raw_df["item"] == "涨幅", "value"].values[0]
+            ),
+            "timestamp": pd.to_datetime(
+                raw_df.loc[raw_df["item"] == "时间", "value"].values[0]
+            )
+            .tz_localize("Asia/Shanghai")
+            .tz_convert("UTC"),
+            "volume": int(raw_df.loc[raw_df["item"] == "成交量", "value"].values[0])
+            / 100,
+            "amount": float(raw_df.loc[raw_df["item"] == "成交额", "value"].values[0]),
+            "open": float(raw_df.loc[raw_df["item"] == "今开", "value"].values[0]),
+            "high": float(raw_df.loc[raw_df["item"] == "最高", "value"].values[0]),
+            "low": float(raw_df.loc[raw_df["item"] == "最低", "value"].values[0]),
+            "prev_close": float(
+                raw_df.loc[raw_df["item"] == "昨收", "value"].values[0]
+            ),
+        }
+
+        return pd.DataFrame([data])

akshare_one/modules/utils.py

@@ -1,10 +1,10 @@
-def convert_xieqiu_symbol(symbol: str) -> str:
-    """
-    Convert Symbol (600000) to XueQiu Symbol (SH600000)
-    """
-    if symbol.startswith("6"):
-        return f"SH{symbol}"
-    elif symbol.startswith("0") or symbol.startswith("3"):
-        return f"SZ{symbol}"
-    else:  # TODO: add more cases
-        return symbol
+def convert_xieqiu_symbol(symbol: str) -> str:
+    """
+    Convert Symbol (600000) to XueQiu Symbol (SH600000)
+    """
+    if symbol.startswith("6"):
+        return f"SH{symbol}"
+    elif symbol.startswith("0") or symbol.startswith("3"):
+        return f"SZ{symbol}"
+    else:  # TODO: add more cases
+        return symbol

akshare_one/news.py

@@ -1,27 +1,27 @@
-"""新闻数据模块
-
-包含股票新闻相关功能
-"""
-
-import pandas as pd
-from .modules.news.factory import NewsDataFactory
-
-
-def get_news_data(symbol: str, source: str = "eastmoney") -> "pd.DataFrame":
-    """获取个股新闻数据
-
-    Args:
-        symbol: 股票代码 (如 "300059")
-        source: 数据源 ('eastmoney')
-
-    Returns:
-        pd.DataFrame:
-        - keyword: 关键词
-        - title: 新闻标题
-        - content: 新闻内容
-        - publish_time: 发布时间
-        - source: 文章来源
-        - url: 新闻链接
-    """
-    provider = NewsDataFactory.get_provider(source, symbol=symbol)
-    return provider.get_news_data()
+"""新闻数据模块
+
+包含股票新闻相关功能
+"""
+
+import pandas as pd
+from .modules.news.factory import NewsDataFactory
+
+
+def get_news_data(symbol: str, source: str = "eastmoney") -> "pd.DataFrame":
+    """获取个股新闻数据
+
+    Args:
+        symbol: 股票代码 (如 "300059")
+        source: 数据源 ('eastmoney')
+
+    Returns:
+        pd.DataFrame:
+        - keyword: 关键词
+        - title: 新闻标题
+        - content: 新闻内容
+        - publish_time: 发布时间
+        - source: 文章来源
+        - url: 新闻链接
+    """
+    provider = NewsDataFactory.get_provider(source, symbol=symbol)
+    return provider.get_news_data()

akshare_one/stock.py

@@ -1,78 +1,78 @@
-"""股票市场数据模块
-
-包含股票历史数据和实时数据相关功能
-"""
-
-from typing import Optional
-import pandas as pd
-
-from akshare_one.modules.historical.factory import HistoricalDataFactory
-from akshare_one.modules.realtime.factory import RealtimeDataFactory
-
-
-def get_hist_data(
-    symbol: str,
-    interval: str = "day",
-    interval_multiplier: int = 1,
-    start_date: str = "1970-01-01",
-    end_date: str = "2030-12-31",
-    adjust: str = "none",
-    source: str = "eastmoney",
-) -> "pd.DataFrame":
-    """Get historical market data
-
-    Args:
-        symbol: 股票代码 (e.g. '600000')
-        interval: 时间间隔 ('minute','hour','day','week','month','year')
-        interval_multiplier: 时间间隔倍数 (e.g. 5 for 5 minutes)
-        start_date: 开始日期 (YYYY-MM-DD)
-        end_date: 结束日期 (YYYY-MM-DD)
-        adjust: 复权类型 ('none','qfq','hfq')
-        source: 数据源 ('eastmoney', 'eastmoney_direct', 'sina') (default: 'eastmoney')
-
-    Returns:
-        pd.DataFrame:
-        - timestamp: 时间戳(UTC时区)
-        - open: 开盘价
-        - high: 最高价
-        - low: 最低价
-        - close: 收盘价
-        - volume: 成交量
-    """
-    kwargs = {
-        "symbol": symbol,
-        "interval": interval,
-        "interval_multiplier": interval_multiplier,
-        "start_date": start_date,
-        "end_date": end_date,
-        "adjust": adjust,
-    }
-    provider = HistoricalDataFactory.get_provider(source, **kwargs)
-    return provider.get_hist_data()
-
-
-def get_realtime_data(
-    symbol: Optional[str] = None, source: str = "xueqiu"
-) -> "pd.DataFrame":
-    """Get real-time market quotes
-
-    Args:
-        symbol: 股票代码 (如 "600000")
-        source: 数据源 ('eastmoney', 'eastmoney_direct', 'xueqiu')
-
-    Returns:
-        pd.DataFrame:
-        - symbol: 股票代码
-        - price: 最新价
-        - change: 涨跌额
-        - pct_change: 涨跌幅(%)
-        - timestamp: 时间戳
-        - volume: 成交量(手)
-        - amount: 成交额(元)
-        - open: 今开
-        - high: 最高
-        - low: 最低
-        - prev_close: 昨收
-    """
-    provider = RealtimeDataFactory.get_provider(source, symbol=symbol)
-    return provider.get_current_data()
+"""股票市场数据模块
+
+包含股票历史数据和实时数据相关功能
+"""
+
+from typing import Optional
+import pandas as pd
+
+from akshare_one.modules.historical.factory import HistoricalDataFactory
+from akshare_one.modules.realtime.factory import RealtimeDataFactory
+
+
+def get_hist_data(
+    symbol: str,
+    interval: str = "day",
+    interval_multiplier: int = 1,
+    start_date: str = "1970-01-01",
+    end_date: str = "2030-12-31",
+    adjust: str = "none",
+    source: str = "eastmoney",
+) -> "pd.DataFrame":
+    """Get historical market data
+
+    Args:
+        symbol: 股票代码 (e.g. '600000')
+        interval: 时间间隔 ('minute','hour','day','week','month','year')
+        interval_multiplier: 时间间隔倍数 (e.g. 5 for 5 minutes)
+        start_date: 开始日期 (YYYY-MM-DD)
+        end_date: 结束日期 (YYYY-MM-DD)
+        adjust: 复权类型 ('none','qfq','hfq')
+        source: 数据源 ('eastmoney', 'eastmoney_direct', 'sina') (default: 'eastmoney')
+
+    Returns:
+        pd.DataFrame:
+        - timestamp: 时间戳(UTC时区)
+        - open: 开盘价
+        - high: 最高价
+        - low: 最低价
+        - close: 收盘价
+        - volume: 成交量
+    """
+    kwargs = {
+        "symbol": symbol,
+        "interval": interval,
+        "interval_multiplier": interval_multiplier,
+        "start_date": start_date,
+        "end_date": end_date,
+        "adjust": adjust,
+    }
+    provider = HistoricalDataFactory.get_provider(source, **kwargs)
+    return provider.get_hist_data()
+
+
+def get_realtime_data(
+    symbol: Optional[str] = None, source: str = "xueqiu"
+) -> "pd.DataFrame":
+    """Get real-time market quotes
+
+    Args:
+        symbol: 股票代码 (如 "600000")
+        source: 数据源 ('eastmoney', 'eastmoney_direct', 'xueqiu')
+
+    Returns:
+        pd.DataFrame:
+        - symbol: 股票代码
+        - price: 最新价
+        - change: 涨跌额
+        - pct_change: 涨跌幅(%)
+        - timestamp: 时间戳
+        - volume: 成交量(手)
+        - amount: 成交额(元)
+        - open: 今开
+        - high: 最高
+        - low: 最低
+        - prev_close: 昨收
+    """
+    provider = RealtimeDataFactory.get_provider(source, symbol=symbol)
+    return provider.get_current_data()