akshare-one 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

akshare_one/modules/insider/base.py

@@ -1,28 +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
+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/modules/insider/factory.py

@@ -1,44 +1,44 @@
-from .xueqiu import XueQiuInsider
-from .base import InsiderDataProvider
-
-
-class InsiderDataFactory:
-    """
-    Factory class for creating insider data providers
-    """
-
-    _providers = {
-        "xueqiu": XueQiuInsider,
-    }
-
-    @classmethod
-    def get_provider(cls, provider_name: str, **kwargs) -> InsiderDataProvider:
-        """
-        Get an insider data provider by name
-
-        Args:
-            provider_name: Name of the provider (e.g., 'xueqiu')
-            **kwargs: Additional arguments to pass to the provider's constructor
-
-        Returns:
-            InsiderDataProvider: 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 insider data provider: {provider_name}")
-
-        return provider_class(**kwargs)
-
-    @classmethod
-    def register_provider(cls, name: str, provider_class: type):
-        """
-        Register a new insider data provider
-
-        Args:
-            name: Name to associate with this provider
-            provider_class: The provider class to register
-        """
-        cls._providers[name.lower()] = provider_class
+from .xueqiu import XueQiuInsider
+from .base import InsiderDataProvider
+
+
+class InsiderDataFactory:
+    """
+    Factory class for creating insider data providers
+    """
+
+    _providers = {
+        "xueqiu": XueQiuInsider,
+    }
+
+    @classmethod
+    def get_provider(cls, provider_name: str, **kwargs) -> InsiderDataProvider:
+        """
+        Get an insider data provider by name
+
+        Args:
+            provider_name: Name of the provider (e.g., 'xueqiu')
+            **kwargs: Additional arguments to pass to the provider's constructor
+
+        Returns:
+            InsiderDataProvider: 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 insider data provider: {provider_name}")
+
+        return provider_class(**kwargs)
+
+    @classmethod
+    def register_provider(cls, name: str, provider_class: type):
+        """
+        Register a new insider 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/insider/xueqiu.py

@@ -1,115 +1,115 @@
-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
-
-
-class XueQiuInsider(InsiderDataProvider):
-    """Provider for XueQiu insider trading data"""
-
-    @cached(
-        cache=CACHE_CONFIG["financial_cache"],
-        key=lambda self, symbol=None: f"xueqiu_insider_{symbol if symbol else 'all'}",
-    )
-    def get_inner_trade_data(self) -> pd.DataFrame:
-        """获取雪球内部交易数据
-
-        Args:
-            symbol: 可选股票代码，如"600000"，不传则返回所有数据
-
-        Returns:
-            Standardized DataFrame with insider trading data:
-            - symbol: 股票代码
-            - name: 股票名称
-            - change_date: 变动日期
-            - insider: 变动人
-            - shares_changed: 变动股数
-            - avg_price: 成交均价
-            - shares_after: 变动后持股数
-            - relationship: 与董监高关系
-            - position: 董监高职务
-        """
-        raw_df = ak.stock_inner_trade_xq()
-        if self.symbol:
-            xueqiu_symbol = convert_xieqiu_symbol(self.symbol)
-            raw_df = raw_df[raw_df["股票代码"] == xueqiu_symbol]
-        return self._clean_insider_data(raw_df)
-
-    def _clean_insider_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
-        """清理和标准化内部交易数据
-
-        Args:
-            raw_df: Raw DataFrame from XueQiu API
-
-        Returns:
-            Standardized DataFrame with consistent columns
-        """
-        column_mapping = {
-            "股票代码": "symbol",
-            "股票名称": "issuer",
-            "变动人": "name",
-            "董监高职务": "title",
-            "变动日期": "transaction_date",
-            "变动股数": "transaction_shares",
-            "成交均价": "transaction_price_per_share",
-            "变动后持股数": "shares_owned_after_transaction",
-            "与董监高关系": "relationship",
-        }
-
-        df = raw_df.rename(columns=column_mapping)
-
-        # Convert symbol back to original format (remove SH/SZ prefix)
-        if "symbol" in df.columns:
-            df["symbol"] = df["symbol"].str.replace(r"^[A-Z]{2}", "", regex=True)
-
-        # Add is_board_director column
-        df["is_board_director"] = df["title"].str.contains("董事")
-
-        # Calculate transaction_value
-        if (
-            "transaction_shares" in df.columns
-            and "transaction_price_per_share" in df.columns
-        ):
-            df["transaction_value"] = (
-                df["transaction_shares"] * df["transaction_price_per_share"]
-            )
-
-        # Add shares_owned_before_transaction if possible
-        if (
-            "shares_owned_after_transaction" in df.columns
-            and "transaction_shares" in df.columns
-        ):
-            df["shares_owned_before_transaction"] = (
-                df["shares_owned_after_transaction"] - df["transaction_shares"]
-            )
-
-        # 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")
-            )
-
-        if "filing_date" in df.columns:
-            df["filing_date"] = (
-                pd.to_datetime(df["filing_date"])
-                .dt.tz_localize("Asia/Shanghai")
-                .dt.tz_convert("UTC")
-            )
-
-        # Convert numeric columns
-        numeric_cols = [
-            "transaction_shares",
-            "transaction_price_per_share",
-            "transaction_value",
-            "shares_owned_before_transaction",
-            "shares_owned_after_transaction",
-        ]
-        for col in numeric_cols:
-            if col in df.columns:
-                df[col] = pd.to_numeric(df[col], errors="coerce")
-
-        return df.reset_index(drop=True)
+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
+
+
+class XueQiuInsider(InsiderDataProvider):
+    """Provider for XueQiu insider trading data"""
+
+    @cached(
+        cache=CACHE_CONFIG["financial_cache"],
+        key=lambda self, symbol=None: f"xueqiu_insider_{symbol if symbol else 'all'}",
+    )
+    def get_inner_trade_data(self) -> pd.DataFrame:
+        """获取雪球内部交易数据
+
+        Args:
+            symbol: 可选股票代码，如"600000"，不传则返回所有数据
+
+        Returns:
+            Standardized DataFrame with insider trading data:
+            - symbol: 股票代码
+            - name: 股票名称
+            - change_date: 变动日期
+            - insider: 变动人
+            - shares_changed: 变动股数
+            - avg_price: 成交均价
+            - shares_after: 变动后持股数
+            - relationship: 与董监高关系
+            - position: 董监高职务
+        """
+        raw_df = ak.stock_inner_trade_xq()
+        if self.symbol:
+            xueqiu_symbol = convert_xieqiu_symbol(self.symbol)
+            raw_df = raw_df[raw_df["股票代码"] == xueqiu_symbol]
+        return self._clean_insider_data(raw_df)
+
+    def _clean_insider_data(self, raw_df: pd.DataFrame) -> pd.DataFrame:
+        """清理和标准化内部交易数据
+
+        Args:
+            raw_df: Raw DataFrame from XueQiu API
+
+        Returns:
+            Standardized DataFrame with consistent columns
+        """
+        column_mapping = {
+            "股票代码": "symbol",
+            "股票名称": "issuer",
+            "变动人": "name",
+            "董监高职务": "title",
+            "变动日期": "transaction_date",
+            "变动股数": "transaction_shares",
+            "成交均价": "transaction_price_per_share",
+            "变动后持股数": "shares_owned_after_transaction",
+            "与董监高关系": "relationship",
+        }
+
+        df = raw_df.rename(columns=column_mapping)
+
+        # Convert symbol back to original format (remove SH/SZ prefix)
+        if "symbol" in df.columns:
+            df["symbol"] = df["symbol"].str.replace(r"^[A-Z]{2}", "", regex=True)
+
+        # Add is_board_director column
+        df["is_board_director"] = df["title"].str.contains("董事")
+
+        # Calculate transaction_value
+        if (
+            "transaction_shares" in df.columns
+            and "transaction_price_per_share" in df.columns
+        ):
+            df["transaction_value"] = (
+                df["transaction_shares"] * df["transaction_price_per_share"]
+            )
+
+        # Add shares_owned_before_transaction if possible
+        if (
+            "shares_owned_after_transaction" in df.columns
+            and "transaction_shares" in df.columns
+        ):
+            df["shares_owned_before_transaction"] = (
+                df["shares_owned_after_transaction"] - df["transaction_shares"]
+            )
+
+        # 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")
+            )
+
+        if "filing_date" in df.columns:
+            df["filing_date"] = (
+                pd.to_datetime(df["filing_date"])
+                .dt.tz_localize("Asia/Shanghai")
+                .dt.tz_convert("UTC")
+            )
+
+        # Convert numeric columns
+        numeric_cols = [
+            "transaction_shares",
+            "transaction_price_per_share",
+            "transaction_value",
+            "shares_owned_before_transaction",
+            "shares_owned_after_transaction",
+        ]
+        for col in numeric_cols:
+            if col in df.columns:
+                df[col] = pd.to_numeric(df[col], errors="coerce")
+
+        return df.reset_index(drop=True)

akshare_one/modules/news/base.py

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

@@ -1,47 +1,47 @@
-from cachetools import cached
-import pandas as pd
-import akshare as ak
-
-from ..cache import CACHE_CONFIG
-from .base import NewsDataProvider
-
-
-class EastMoneyNews(NewsDataProvider):
-    @cached(
-        CACHE_CONFIG["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)
-
-        # Convert time to UTC
-        df["publish_time"] = (
-            pd.to_datetime(df["publish_time"])
-            .dt.tz_localize("Asia/Shanghai")
-            .dt.tz_convert("UTC")
-        )
-
-        required_columns = [
-            "keyword",
-            "title",
-            "content",
-            "publish_time",
-            "source",
-            "url",
-        ]
-        return df[required_columns]
+from cachetools import cached
+import pandas as pd
+import akshare as ak
+
+from ..cache import CACHE_CONFIG
+from .base import NewsDataProvider
+
+
+class EastMoneyNews(NewsDataProvider):
+    @cached(
+        CACHE_CONFIG["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)
+
+        # Convert time to UTC
+        df["publish_time"] = (
+            pd.to_datetime(df["publish_time"])
+            .dt.tz_localize("Asia/Shanghai")
+            .dt.tz_convert("UTC")
+        )
+
+        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:
+        """
+        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

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