akshare-one 0.1.2__py3-none-any.whl → 0.2.0__py3-none-any.whl

akshare_one/modules/news/base.py

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

akshare_one/modules/news/eastmoney.py

@@ -0,0 +1,48 @@
+from cachetools import cached
+import pandas as pd
+import akshare as ak
+
+from ..cache import CACHE_CONFIG
+from .base import NewsDataProvider, validate_news_data
+
+
+class EastMoneyNews(NewsDataProvider):
+    @validate_news_data
+    @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

@@ -0,0 +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

akshare_one/modules/realtime/base.py

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

akshare_one/modules/realtime/eastmoney.py

@@ -0,0 +1,58 @@
+from cachetools import cached
+import pandas as pd
+import akshare as ak
+
+from ..cache import CACHE_CONFIG
+from .base import RealtimeDataProvider, validate_realtime_data
+
+
+class EastmoneyRealtime(RealtimeDataProvider):
+    @validate_realtime_data
+    @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/factory.py

@@ -0,0 +1,46 @@
+from .eastmoney import EastmoneyRealtime
+from .xueqiu import XueQiuRealtime
+from .base import RealtimeDataProvider
+
+
+class RealtimeDataFactory:
+    """
+    Factory class for creating realtime data providers
+    """
+
+    _providers = {
+        "eastmoney": EastmoneyRealtime,
+        "xueqiu": XueQiuRealtime,
+    }
+
+    @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

@@ -0,0 +1,61 @@
+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, validate_realtime_data
+
+
+class XueQiuRealtime(RealtimeDataProvider):
+    @validate_realtime_data
+    @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

@@ -0,0 +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

akshare_one/news.py

@@ -4,7 +4,7 @@
 """
 
 import pandas as pd
-from .adapters import EastMoneyAdapter
+from .modules.news.factory import NewsDataFactory
 
 
 def get_news_data(symbol: str, source: str = "eastmoney") -> "pd.DataFrame":
@@ -23,6 +23,5 @@ def get_news_data(symbol: str, source: str = "eastmoney") -> "pd.DataFrame":
         - source: 文章来源
         - url: 新闻链接
     """
-    if source == "eastmoney":
-        return EastMoneyAdapter().get_news_data(symbol=symbol)
-    raise ValueError(f"Unsupported data source: {source}")
+    provider = NewsDataFactory.get_provider(source, symbol=symbol)
+    return provider.get_news_data()

akshare_one/stock.py

@@ -5,12 +5,14 @@
 
 from typing import Optional
 import pandas as pd
-from .adapters import EastMoneyAdapter, SinaAdapter
+
+from akshare_one.modules.historical.factory import HistoricalDataFactory
+from akshare_one.modules.realtime.factory import RealtimeDataFactory
 
 
 def get_hist_data(
     symbol: str,
-    interval: str,
+    interval: str = "day",
     interval_multiplier: int = 1,
     start_date: str = "1970-01-01",
     end_date: str = "2030-12-31",
@@ -37,35 +39,26 @@ def get_hist_data(
         - close: 收盘价
         - volume: 成交量
     """
-    if source == "eastmoney":
-        return EastMoneyAdapter().get_hist_data(
-            symbol=symbol,
-            interval=interval,
-            interval_multiplier=interval_multiplier,
-            start_date=start_date,
-            end_date=end_date,
-            adjust=adjust,
-        )
-    elif source == "sina":
-        return SinaAdapter().get_hist_data(
-            symbol=symbol,
-            interval=interval,
-            interval_multiplier=interval_multiplier,
-            start_date=start_date,
-            end_date=end_date,
-            adjust=adjust,
-        )
-    raise ValueError(f"Unsupported data source: {source}")
+    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(
-    source: str = "eastmoney", symbol: Optional[str] = None
+    symbol: Optional[str] = None, source: str = "xueqiu"
 ) -> "pd.DataFrame":
     """Get real-time market quotes
 
     Args:
         symbol: 股票代码 (如 "600000")
-        source: 数据源 ('eastmoney')
+        source: 数据源 ('eastmoney', 'xueqiu')
 
     Returns:
         pd.DataFrame:
@@ -81,6 +74,5 @@ def get_realtime_data(
         - low: 最低
         - prev_close: 昨收
     """
-    if source == "eastmoney":
-        return EastMoneyAdapter().get_realtime_data(symbol=symbol)
-    raise ValueError(f"Unsupported data source: {source}")
+    provider = RealtimeDataFactory.get_provider(source, symbol=symbol)
+    return provider.get_current_data()

{akshare_one-0.1.2.dist-info → akshare_one-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: akshare-one
-Version: 0.1.2
+Version: 0.2.0
 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.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: akshare>=1.16.64
+Requires-Dist: akshare>=1.16.94
 Requires-Dist: cachetools>=5.5.2
 Dynamic: license-file
 

akshare_one-0.2.0.dist-info/RECORD

@@ -0,0 +1,29 @@
+akshare_one/__init__.py,sha256=M4eXCnBzGqa5FihT-q7DHaluTvidnqwVF7AgPgCikKU,878
+akshare_one/financial.py,sha256=XAsonRzGK8akKtW2Q7LUrew4OFRnRAfZm0nw0JY73Jc,1426
+akshare_one/insider.py,sha256=fnbVT--7jGCvULhsiKNqLmZggo1VmR-eNOr4iq5BI7I,1179
+akshare_one/news.py,sha256=yrYeCaKTgCGP-TSyOfOou9gMw8185qiWrg380fD9-f8,669
+akshare_one/stock.py,sha256=K8KOHxHszHoWmdhgO4sj-6zfIUSMS--0PdzgsAOGk88,2203
+akshare_one/modules/cache.py,sha256=47A80Xtfr4gkKvEfBQrT8Dz8hNFR769rBa2gU6ew25s,373
+akshare_one/modules/utils.py,sha256=H4nrGf8m4_ezTiW5-OcNPxpV-neTYfffEfaOLDFLY9Y,323
+akshare_one/modules/financial/base.py,sha256=JRZJhzl7kP16KEVOXP9mq1DjXW6rga2FJ8-sI5FH814,1107
+akshare_one/modules/financial/factory.py,sha256=GqzFp6LoHWj7t5VwtZJkLFxBuaAW_0b__bduBrmlcOg,1301
+akshare_one/modules/financial/sina.py,sha256=gHrbmykselWZC5tqR1Zkyc3oK0TcCCdCn_1z_8C2y_U,11026
+akshare_one/modules/historical/base.py,sha256=hIk-DKxaDXfx9y9VWx-b918ntv6pCeqxjHpAiS7A5ZQ,1895
+akshare_one/modules/historical/eastmoney.py,sha256=8lTZjbQ-P713vBvJxXTBmn6hPpNslWf5uAW8R6HJtyo,8292
+akshare_one/modules/historical/factory.py,sha256=adxNNo_-PT180QXRUNwkdiN5x0lDJpPCt6rNFZcJ0ps,1389
+akshare_one/modules/historical/sina.py,sha256=d-AWr4C72nvPYOC9IETqUXDoksofztWyoe9Ie9TC9Z0,7293
+akshare_one/modules/insider/base.py,sha256=_TDvHr6TZa0sTynE_ZL0njnms-cb5Wd8hfqR7pdfvrU,2728
+akshare_one/modules/insider/factory.py,sha256=om51vX-GTXknpptZGvmDD9m10FfgEVmy402xZ3bBsNM,1280
+akshare_one/modules/insider/xueqiu.py,sha256=8ZBRn8tW4r0aTEsoaQuYzcNQ-q2oQ64Cn7VG_Hs8hhE,4021
+akshare_one/modules/news/base.py,sha256=L55BLQionqTZGmrHZDpmNKuhcWQdIN8ZKR9jsMj1Bgo,1550
+akshare_one/modules/news/eastmoney.py,sha256=iuZgpvub9zKDJNtGIShFXBQRHIG10oEvdMiYFDvbsdI,1377
+akshare_one/modules/news/factory.py,sha256=Em6c7m46K760iwIX3GZ15HdFu7pXT2w-n4QsjwHezjY,1264
+akshare_one/modules/realtime/base.py,sha256=Yztkh9IhCr0Y-N8SH81APL9uNr4ZMd10B5KfcC6Ekcg,2093
+akshare_one/modules/realtime/eastmoney.py,sha256=PPfRC3LUIH0HWbjkgfHgoER5Aq8CtqGFbazjJdYFqcQ,1702
+akshare_one/modules/realtime/factory.py,sha256=SxDvJJvp6VutRqLUGYRnrUBtJEOE0N7vVh5Rw-vJ6NY,1373
+akshare_one/modules/realtime/xueqiu.py,sha256=WnSRktk4eyFV8_8c9SpCtL6UN08vdZdzFpCyhMyLRPM,2293
+akshare_one-0.2.0.dist-info/licenses/LICENSE,sha256=Gg6A1GNSJCZWQ73aHJ7TXOa0i8RQ3FejZCTZ6Db07cU,1066
+akshare_one-0.2.0.dist-info/METADATA,sha256=glFdSHMMfGgYlg1DIZvQ1r0xKnMCQHFgV215lM2pmbo,1865
+akshare_one-0.2.0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+akshare_one-0.2.0.dist-info/top_level.txt,sha256=kNiucyLVAGa89wmUSpXbBLWD7pF_RuahuiaOfLHZSyw,12
+akshare_one-0.2.0.dist-info/RECORD,,

{akshare_one-0.1.2.dist-info → akshare_one-0.2.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

akshare_one/adapters/__init__.py

@@ -1,7 +0,0 @@
-# Initialize adapters package
-from .eastmoney import EastMoneyAdapter
-from .sina import SinaAdapter
-from .xueqiu import XueQiuAdapter
-from .cache.cache import CACHE_CONFIG
-
-__all__ = ['EastMoneyAdapter', 'SinaAdapter', 'XueQiuAdapter', 'CACHE_CONFIG']

akshare_one/adapters/cache/cache.py

@@ -1,9 +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小时
-}