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_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_bak_basic, sync_bak_basic
    >>> data = get_daily('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_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",
    # 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",
]