akshare-one 0.3.7__py3-none-any.whl → 0.3.9__py3-none-any.whl

akshare_one/modules/news/eastmoney.py

@@ -1,43 +1,43 @@
-import pandas as pd
-import akshare as ak
-
-from ..cache import cache
-from .base import NewsDataProvider
-
-
-class EastMoneyNews(NewsDataProvider):
-    @cache(
-        "news_cache",
-        key=lambda self: f"eastmoney_news_{self.symbol}",
-    )
-    def get_news_data(self) -> pd.DataFrame:
-        """获取东方财富个股新闻数据"""
-        raw_df = ak.stock_news_em(symbol=self.symbol)
-        return self._clean_news_data(raw_df)
-
-    def _clean_news_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
-        """清理和标准化新闻数据"""
-        column_mapping = {
-            "关键词": "keyword",
-            "新闻标题": "title",
-            "新闻内容": "content",
-            "发布时间": "publish_time",
-            "文章来源": "source",
-            "新闻链接": "url",
-        }
-
-        df = raw_df.rename(columns=column_mapping)
-
-        df["publish_time"] = pd.to_datetime(df["publish_time"]).dt.tz_localize(
-            "Asia/Shanghai"
-        )
-
-        required_columns = [
-            "keyword",
-            "title",
-            "content",
-            "publish_time",
-            "source",
-            "url",
-        ]
-        return df[required_columns]
+import pandas as pd
+import akshare as ak  # type: ignore
+
+from ..cache import cache
+from .base import NewsDataProvider
+
+
+class EastMoneyNews(NewsDataProvider):
+    @cache(
+        "news_cache",
+        key=lambda self: f"eastmoney_news_{self.symbol}",
+    )
+    def get_news_data(self) -> pd.DataFrame:
+        """获取东方财富个股新闻数据"""
+        raw_df = ak.stock_news_em(symbol=self.symbol)
+        return self._clean_news_data(raw_df)
+
+    def _clean_news_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
+        """清理和标准化新闻数据"""
+        column_mapping = {
+            "关键词": "keyword",
+            "新闻标题": "title",
+            "新闻内容": "content",
+            "发布时间": "publish_time",
+            "文章来源": "source",
+            "新闻链接": "url",
+        }
+
+        df = raw_df.rename(columns=column_mapping)
+
+        df["publish_time"] = pd.to_datetime(df["publish_time"]).dt.tz_localize(
+            "Asia/Shanghai"
+        )
+
+        required_columns = [
+            "keyword",
+            "title",
+            "content",
+            "publish_time",
+            "source",
+            "url",
+        ]
+        return df[required_columns]

akshare_one/modules/news/factory.py

@@ -1,44 +1,44 @@
-from .eastmoney import EastMoneyNews
-from .base import NewsDataProvider
-
-
-class NewsDataFactory:
-    """
-    Factory class for creating news data providers
-    """
-
-    _providers = {
-        "eastmoney": EastMoneyNews,
-    }
-
-    @classmethod
-    def get_provider(cls, provider_name: str, **kwargs) -> NewsDataProvider:
-        """
-        Get a news 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:
-            NewsDataProvider: 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 news data provider: {provider_name}")
-
-        return provider_class(**kwargs)
-
-    @classmethod
-    def register_provider(cls, name: str, provider_class: type):
-        """
-        Register a new news 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 EastMoneyNews
+from .base import NewsDataProvider
+
+
+class NewsDataFactory:
+    """
+    Factory class for creating news data providers
+    """
+
+    _providers = {
+        "eastmoney": EastMoneyNews,
+    }
+
+    @classmethod
+    def get_provider(cls, provider_name: str, **kwargs) -> NewsDataProvider:  # type: ignore
+        """
+        Get a news 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:
+            NewsDataProvider: 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 news data provider: {provider_name}")
+
+        return provider_class(**kwargs)
+
+    @classmethod
+    def register_provider(cls, name: str, provider_class: type) -> None:
+        """
+        Register a new news data provider
+
+        Args:
+            name: Name to associate with this provider
+            provider_class: The provider class to register
+        """
+        cls._providers[name.lower()] = provider_class  # type: ignore

akshare_one/modules/realtime/base.py

@@ -1,27 +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
+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/modules/realtime/eastmoney.py

@@ -1,53 +1,53 @@
-import pandas as pd
-import akshare as ak
-
-from ..cache import cache
-from .base import RealtimeDataProvider
-
-
-class EastmoneyRealtime(RealtimeDataProvider):
-    @cache(
-        "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)
-
-        df = df.assign(timestamp=lambda x: pd.Timestamp.now(tz="Asia/Shanghai"))
-
-        required_columns = [
-            "symbol",
-            "price",
-            "change",
-            "pct_change",
-            "timestamp",
-            "volume",
-            "amount",
-            "open",
-            "high",
-            "low",
-            "prev_close",
-        ]
-        return df[required_columns]
+import pandas as pd
+import akshare as ak  # type: ignore
+
+from ..cache import cache
+from .base import RealtimeDataProvider
+
+
+class EastmoneyRealtime(RealtimeDataProvider):
+    @cache(
+        "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)
+
+        df = df.assign(timestamp=lambda x: pd.Timestamp.now(tz="Asia/Shanghai"))
+
+        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,36 +1,36 @@
-import pandas as pd
-from .base import RealtimeDataProvider
-from ..cache import cache
-from akshare_one.eastmoney.client import EastMoneyClient
-from akshare_one.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()
-
-    @cache(
-        "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 .base import RealtimeDataProvider
+from ..cache import cache
+from akshare_one.eastmoney.client import EastMoneyClient
+from akshare_one.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()
+
+    @cache(
+        "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,71 @@
-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: dict[str, type["RealtimeDataProvider"]] = {
+        "eastmoney": EastmoneyRealtime,
+        "xueqiu": XueQiuRealtime,
+        "eastmoney_direct": EastMoneyDirectRealtime,
+    }
+
+    @classmethod
+    def get_provider(cls, provider_name: str, **kwargs: object) -> "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}")
+
+        # Extract symbol from kwargs if present
+        symbol = kwargs.get("symbol", "")
+        if not isinstance(symbol, str):
+            raise ValueError("symbol must be a string")
+        
+        return provider_class(symbol=symbol)
+        """
+        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) -> None:
+        """
+        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,57 +1,57 @@
-import pandas as pd
-import akshare as ak
-from ..utils import convert_xieqiu_symbol
-from ..cache import cache
-from .base import RealtimeDataProvider
-
-
-class XueQiuRealtime(RealtimeDataProvider):
-    @cache(
-        "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"),
-            "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])
+import pandas as pd
+import akshare as ak  # type: ignore
+from ..utils import convert_xieqiu_symbol
+from ..cache import cache
+from .base import RealtimeDataProvider
+
+
+class XueQiuRealtime(RealtimeDataProvider):
+    @cache(
+        "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"),
+            "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])