ProStock/src/data/api_wrappers/__init__.py

"""Tushare API wrapper modules.

This package contains simplified interfaces for fetching data from Tushare API.
All wrapper files follow the naming convention: api_{data_type}.py

Available APIs:
    - api_daily: Daily market data (日线行情)
    - api_pro_bar: Pro Bar universal market data (通用行情，后复权)
    - api_stock_basic: Stock basic information (股票基本信息)
    - api_trade_cal: Trading calendar (交易日历)
    - api_namechange: Stock name change history (股票曾用名)
    - api_bak_basic: Stock historical list (股票历史列表)

Example:
    >>> from src.data.api_wrappers import get_daily, get_stock_basic, get_trade_cal, get_bak_basic
    >>> from src.data.api_wrappers import get_pro_bar, sync_pro_bar
    >>> data = get_daily('000001.SZ', start_date='20240101', end_date='20240131')
    >>> pro_data = get_pro_bar('000001.SZ', start_date='20240101', end_date='20240131')
    >>> stocks = get_stock_basic()
    >>> calendar = get_trade_cal('20240101', '20240131')
    >>> bak_basic = get_bak_basic(trade_date='20240101')
"""

from src.data.api_wrappers.api_daily import (
    get_daily,
    sync_daily,
    preview_daily_sync,
    DailySync,
)
from src.data.api_wrappers.api_pro_bar import (
    get_pro_bar,
    sync_pro_bar,
    preview_pro_bar_sync,
    ProBarSync,
)
from src.data.api_wrappers.financial_data.api_income import (
    get_income,
    sync_income,
    IncomeSync,
)
from src.data.api_wrappers.api_bak_basic import get_bak_basic, sync_bak_basic
from src.data.api_wrappers.api_namechange import get_namechange, sync_namechange
from src.data.api_wrappers.api_stock_basic import get_stock_basic, sync_all_stocks
from src.data.api_wrappers.api_trade_cal import (
    get_trade_cal,
    get_trading_days,
    get_first_trading_day,
    get_last_trading_day,
    sync_trade_cal_cache,
)

__all__ = [
    # Daily market data
    "get_daily",
    "sync_daily",
    "preview_daily_sync",
    "DailySync",
    # Pro Bar (universal market data)
    "get_pro_bar",
    "sync_pro_bar",
    "preview_pro_bar_sync",
    "ProBarSync",
    # Income statement
    "get_income",
    "sync_income",
    "IncomeSync",
    # Historical stock list
    "get_bak_basic",
    "sync_bak_basic",
    # Namechange
    "get_namechange",
    "sync_namechange",
    # Stock basic information
    "get_stock_basic",
    "sync_all_stocks",
    # Trade calendar
    "get_trade_cal",
    "get_trading_days",
    "get_first_trading_day",
    "get_last_trading_day",
    "sync_trade_cal_cache",
]