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

akshare_one/__init__.py

@@ -1,31 +1,31 @@
-"""Akshare One - Unified interface for Chinese market data
-
-Provides standardized access to various financial data sources with:
-- Consistent symbol formats
-- Unified data schemas
-- Cleaned and normalized outputs
-
-Example:
-    >>> from akshare_one import get_hist_data, get_realtime_data
-    >>> # 获取股票历史数据
-    >>> df = get_hist_data("600000", interval="day")
-    >>> print(df.head())
-    >>> # 获取股票实时数据
-    >>> df = get_realtime_data(symbol="600000")
-"""
-
-from .stock import get_hist_data, get_realtime_data
-from .news import get_news_data
-from .insider import get_inner_trade_data
-from .financial import get_balance_sheet, get_income_statement, get_cash_flow
-
-
-__all__ = [
-    "get_hist_data",
-    "get_realtime_data",
-    "get_news_data",
-    "get_inner_trade_data",
-    "get_balance_sheet",
-    "get_income_statement",
-    "get_cash_flow",
-]
+"""Akshare One - Unified interface for Chinese market data
+
+Provides standardized access to various financial data sources with:
+- Consistent symbol formats
+- Unified data schemas
+- Cleaned and normalized outputs
+
+Example:
+    >>> from akshare_one import get_hist_data, get_realtime_data
+    >>> # 获取股票历史数据
+    >>> df = get_hist_data("600000", interval="day")
+    >>> print(df.head())
+    >>> # 获取股票实时数据
+    >>> df = get_realtime_data(symbol="600000")
+"""
+
+from .stock import get_hist_data, get_realtime_data
+from .news import get_news_data
+from .insider import get_inner_trade_data
+from .financial import get_balance_sheet, get_income_statement, get_cash_flow
+
+
+__all__ = [
+    "get_hist_data",
+    "get_realtime_data",
+    "get_news_data",
+    "get_inner_trade_data",
+    "get_balance_sheet",
+    "get_income_statement",
+    "get_cash_flow",
+]

akshare_one/financial.py

@@ -1,46 +1,46 @@
-"""财务报表数据模块
-
-包含资产负债表、利润表和现金流量表相关功能
-"""
-
-import pandas as pd
-from akshare_one.modules.financial.factory import FinancialDataFactory
-
-
-def get_balance_sheet(symbol: str, source: str = "sina") -> "pd.DataFrame":
-    """获取资产负债表数据
-
-    Args:
-        symbol: 股票代码 (如 "600600")
-        source: 数据源 ("sina")
-    """
-    if source == "sina":
-        provider = FinancialDataFactory.get_provider(source, symbol=symbol)
-        return provider.get_balance_sheet()
-    raise ValueError(f"Unsupported data source: {source}")
-
-
-def get_income_statement(symbol: str, source: str = "sina") -> "pd.DataFrame":
-    """获取利润表数据
-
-    Args:
-        symbol: 股票代码 (如 "600600")
-        source: 数据源 ("sina")
-    """
-    if source == "sina":
-        provider = FinancialDataFactory.get_provider(source, symbol=symbol)
-        return provider.get_income_statement()
-    raise ValueError(f"Unsupported data source: {source}")
-
-
-def get_cash_flow(symbol: str, source: str = "sina") -> "pd.DataFrame":
-    """获取现金流量表数据
-
-    Args:
-        symbol: 股票代码 (如 "600600")
-        source: 数据源 ("sina")
-    """
-    if source == "sina":
-        provider = FinancialDataFactory.get_provider(source, symbol=symbol)
-        return provider.get_cash_flow()
-    raise ValueError(f"Unsupported data source: {source}")
+"""财务报表数据模块
+
+包含资产负债表、利润表和现金流量表相关功能
+"""
+
+import pandas as pd
+from akshare_one.modules.financial.factory import FinancialDataFactory
+
+
+def get_balance_sheet(symbol: str, source: str = "sina") -> "pd.DataFrame":
+    """获取资产负债表数据
+
+    Args:
+        symbol: 股票代码 (如 "600600")
+        source: 数据源 ("sina")
+    """
+    if source == "sina":
+        provider = FinancialDataFactory.get_provider(source, symbol=symbol)
+        return provider.get_balance_sheet()
+    raise ValueError(f"Unsupported data source: {source}")
+
+
+def get_income_statement(symbol: str, source: str = "sina") -> "pd.DataFrame":
+    """获取利润表数据
+
+    Args:
+        symbol: 股票代码 (如 "600600")
+        source: 数据源 ("sina")
+    """
+    if source == "sina":
+        provider = FinancialDataFactory.get_provider(source, symbol=symbol)
+        return provider.get_income_statement()
+    raise ValueError(f"Unsupported data source: {source}")
+
+
+def get_cash_flow(symbol: str, source: str = "sina") -> "pd.DataFrame":
+    """获取现金流量表数据
+
+    Args:
+        symbol: 股票代码 (如 "600600")
+        source: 数据源 ("sina")
+    """
+    if source == "sina":
+        provider = FinancialDataFactory.get_provider(source, symbol=symbol)
+        return provider.get_cash_flow()
+    raise ValueError(f"Unsupported data source: {source}")

akshare_one/indicators.py

@@ -0,0 +1,395 @@
+"""Technical indicators module
+
+Provides common technical analysis indicators like:
+- Simple Moving Average (SMA)
+- Exponential Moving Average (EMA)
+- Relative Strength Index (RSI)
+- Moving Average Convergence Divergence (MACD)
+- Bollinger Bands (BBANDS)
+- Stochastic Oscillator (STOCH)
+- Average True Range (ATR)
+- Commodity Channel Index (CCI)
+- Average Directional Index (ADX)
+- Williams' %R (WILLR)
+- Chaikin A/D Line (AD)
+- Chaikin A/D Oscillator (ADOSC)
+- On Balance Volume (OBV)
+- Momentum (MOM)
+- Parabolic SAR (SAR)
+- Time Series Forecast (TSF)
+- Absolute Price Oscillator (APO)
+- Aroon (AROON)
+- Aroon Oscillator (AROONOSC)
+- Balance of Power (BOP)
+- Chande Momentum Oscillator (CMO)
+- Directional Movement Index (DX)
+- Money Flow Index (MFI)
+- Minus Directional Indicator (MINUS_DI)
+- Minus Directional Movement (MINUS_DM)
+- Plus Directional Indicator (PLUS_DI)
+- Plus Directional Movement (PLUS_DM)
+- Percentage Price Oscillator (PPO)
+- Rate of change (ROC)
+- Rate of change Percentage (ROCP)
+- Rate of change ratio (ROCR)
+- Rate of change ratio 100 scale (ROCR100)
+- 1-day Rate of Change (ROC) of a Triple Smooth EMA (TRIX)
+- Ultimate Oscillator (ULTOSC)
+"""
+
+import pandas as pd
+from .modules.indicators.factory import IndicatorFactory
+
+
+def get_sma(
+    df: pd.DataFrame, window: int = 20, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Simple Moving Average
+
+    Args:
+        df: DataFrame with 'close' column
+        window: Lookback window size
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_sma(df, window)
+
+
+def get_ema(
+    df: pd.DataFrame, window: int = 20, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Exponential Moving Average
+
+    Args:
+        df: DataFrame with 'close' column
+        window: Lookback window size
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_ema(df, window)
+
+
+def get_rsi(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Relative Strength Index
+
+    Args:
+        df: DataFrame with 'close' column
+        window: Lookback window size
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_rsi(df, window)
+
+
+def get_macd(
+    df: pd.DataFrame,
+    fast: int = 12,
+    slow: int = 26,
+    signal: int = 9,
+    calculator_type: str = "talib",
+) -> pd.DataFrame:
+    """Calculate MACD
+
+    Args:
+        df: DataFrame with 'close' column
+        fast: Fast period
+        slow: Slow period
+        signal: Signal period
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_macd(df, fast, slow, signal)
+
+
+def get_bollinger_bands(
+    df: pd.DataFrame,
+    window: int = 20,
+    std: int = 2,
+    calculator_type: str = "talib",
+) -> pd.DataFrame:
+    """Calculate Bollinger Bands
+
+    Args:
+        df: DataFrame with 'close' column
+        window: Lookback window size
+        std: Standard deviation
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_bollinger_bands(df, window, std)
+
+
+def get_stoch(
+    df: pd.DataFrame,
+    window: int = 14,
+    smooth_d: int = 3,
+    smooth_k: int = 3,
+    calculator_type: str = "talib",
+) -> pd.DataFrame:
+    """Calculate Stochastic Oscillator
+
+    Args:
+        df: DataFrame with 'high', 'low', 'close' columns
+        window: Lookback window size
+        smooth_d: Smoothing for D line
+        smooth_k: Smoothing for K line
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_stoch(df, window, smooth_d, smooth_k)
+
+
+def get_atr(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Average True Range
+
+    Args:
+        df: DataFrame with 'high', 'low', 'close' columns
+        window: Lookback window size
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_atr(df, window)
+
+
+def get_cci(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Commodity Channel Index
+
+    Args:
+        df: DataFrame with 'high', 'low', 'close' columns
+        window: Lookback window size
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_cci(df, window)
+
+
+def get_adx(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Average Directional Index
+
+    Args:
+        df: DataFrame with 'high', 'low', 'close' columns
+        window: Lookback window size
+        calculator_type: ('talib', 'simple')
+    """
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_adx(df, window)
+
+
+def get_willr(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Williams' %R"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_willr(df, window)
+
+
+def get_ad(df: pd.DataFrame, calculator_type: str = "talib") -> pd.DataFrame:
+    """Calculate Chaikin A/D Line"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_ad(df)
+
+
+def get_adosc(
+    df: pd.DataFrame,
+    fast_period: int = 3,
+    slow_period: int = 10,
+    calculator_type: str = "talib",
+) -> pd.DataFrame:
+    """Calculate Chaikin A/D Oscillator"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_adosc(df, fast_period, slow_period)
+
+
+def get_obv(df: pd.DataFrame, calculator_type: str = "talib") -> pd.DataFrame:
+    """Calculate On Balance Volume"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_obv(df)
+
+
+def get_mom(
+    df: pd.DataFrame, window: int = 10, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Momentum"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_mom(df, window)
+
+
+def get_sar(
+    df: pd.DataFrame,
+    acceleration: float = 0.02,
+    maximum: float = 0.2,
+    calculator_type: str = "talib",
+) -> pd.DataFrame:
+    """Calculate Parabolic SAR"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_sar(df, acceleration, maximum)
+
+
+def get_tsf(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Time Series Forecast"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_tsf(df, window)
+
+
+def get_apo(
+    df: pd.DataFrame,
+    fast_period: int = 12,
+    slow_period: int = 26,
+    ma_type: int = 0,
+    calculator_type: str = "talib",
+) -> pd.DataFrame:
+    """Calculate Absolute Price Oscillator"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_apo(df, fast_period, slow_period, ma_type)
+
+
+def get_aroon(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Aroon"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_aroon(df, window)
+
+
+def get_aroonosc(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Aroon Oscillator"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_aroonosc(df, window)
+
+
+def get_bop(df: pd.DataFrame, calculator_type: str = "talib") -> pd.DataFrame:
+    """Calculate Balance of Power"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_bop(df)
+
+
+def get_cmo(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Chande Momentum Oscillator"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_cmo(df, window)
+
+
+def get_dx(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Directional Movement Index"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_dx(df, window)
+
+
+def get_mfi(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Money Flow Index"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_mfi(df, window)
+
+
+def get_minus_di(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Minus Directional Indicator"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_minus_di(df, window)
+
+
+def get_minus_dm(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Minus Directional Movement"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_minus_dm(df, window)
+
+
+def get_plus_di(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Plus Directional Indicator"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_plus_di(df, window)
+
+
+def get_plus_dm(
+    df: pd.DataFrame, window: int = 14, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Plus Directional Movement"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_plus_dm(df, window)
+
+
+def get_ppo(
+    df: pd.DataFrame,
+    fast_period: int = 12,
+    slow_period: int = 26,
+    ma_type: int = 0,
+    calculator_type: str = "talib",
+) -> pd.DataFrame:
+    """Calculate Percentage Price Oscillator"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_ppo(df, fast_period, slow_period, ma_type)
+
+
+def get_roc(
+    df: pd.DataFrame, window: int = 10, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Rate of change"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_roc(df, window)
+
+
+def get_rocp(
+    df: pd.DataFrame, window: int = 10, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Rate of change Percentage"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_rocp(df, window)
+
+
+def get_rocr(
+    df: pd.DataFrame, window: int = 10, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Rate of change ratio"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_rocr(df, window)
+
+
+def get_rocr100(
+    df: pd.DataFrame, window: int = 10, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate Rate of change ratio 100 scale"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_rocr100(df, window)
+
+
+def get_trix(
+    df: pd.DataFrame, window: int = 30, calculator_type: str = "talib"
+) -> pd.DataFrame:
+    """Calculate 1-day Rate of Change (ROC) of a Triple Smooth EMA"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_trix(df, window)
+
+
+def get_ultosc(
+    df: pd.DataFrame,
+    window1: int = 7,
+    window2: int = 14,
+    window3: int = 28,
+    calculator_type: str = "talib",
+) -> pd.DataFrame:
+    """Calculate Ultimate Oscillator"""
+    calculator = IndicatorFactory.get_calculator(calculator_type)
+    return calculator.calculate_ultosc(df, window1, window2, window3)

akshare_one/insider.py

@@ -1,33 +1,33 @@
-"""内部交易数据模块
-
-包含上市公司内部交易相关功能
-"""
-
-import pandas as pd
-from .modules.insider.factory import InsiderDataFactory
-
-
-def get_inner_trade_data(symbol: str, source: str = "xueqiu") -> "pd.DataFrame":
-    """获取雪球内部交易数据
-
-    Args:
-        source: 数据源 (目前支持 "xueqiu")
-        symbol: 股票代码，如"600000"
-
-    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: 变动前持股数
-    """
-    provider = InsiderDataFactory.get_provider(source, symbol=symbol)
-    return provider.get_inner_trade_data()
+"""内部交易数据模块
+
+包含上市公司内部交易相关功能
+"""
+
+import pandas as pd
+from .modules.insider.factory import InsiderDataFactory
+
+
+def get_inner_trade_data(symbol: str, source: str = "xueqiu") -> "pd.DataFrame":
+    """获取雪球内部交易数据
+
+    Args:
+        source: 数据源 (目前支持 "xueqiu")
+        symbol: 股票代码，如"600000"
+
+    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: 变动前持股数
+    """
+    provider = InsiderDataFactory.get_provider(source, symbol=symbol)
+    return provider.get_inner_trade_data()

akshare_one/modules/cache.py

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