skills/ppt-station-skill/ppt_station/template/chart_presets.py

"""
预设图表配置库

包含常用的图表配置，可直接用于模板替换
"""

from pptx.enum.chart import XL_LEGEND_POSITION
from ..chart_builder.layout import ChartLayoutConfig, LegendConfig, ValueAxisConfig
from ..chart_builder.styles import StyleConfig
from ..chart_builder.date_axis import DateAxisConfig


# ============================================================================
# 预设图表配置
# ============================================================================

def get_chart1_config(df):
    """
    图表1：当年收益率走势（柱状图+折线图）
    
    特点：
    - 柱状图（沪深300指数，右轴）
    - 折线图（组合收益率，左轴）
    - 日期轴：2024/01 格式
    - 双轴配置
    """
    label_interval = len(df) // 7
    
    return {
        "df": df,
        "categories_col": "分类",
        "series_config": [
            {
                'name': '沪深300指数(收盘价)',
                'key': '沪深300指数(收盘价)',
                'type': 'bar',
                'axis': 'secondary'
            },
            {
                'name': '组合收益率（左轴）',
                'key': '组合收益率（左轴）',
                'type': 'line',
                'axis': 'primary'
            },
        ],
        "style_config": StyleConfig(
            color_scheme="aim00",
            line_width_pt=2.0,
            marker_style="none",
        ),
        "layout_config": ChartLayoutConfig(
            title="组合2024年以来收益率走势图",
            legend_config=LegendConfig(
                position=XL_LEGEND_POSITION.TOP,
                font_size_pt=9,
                font_name="黑体",
            ),
            value_axis_config=ValueAxisConfig(
                number_format="0%",
                font_size_pt=9,
                font_name="黑体",
                has_major_gridlines=False,
            ),
            secondary_value_axis_config=ValueAxisConfig(
                number_format="#,##0",
                font_size_pt=9,
                font_name="黑体",
                has_major_gridlines=False,
                min_value=2950,
                max_value=4450,
                major_unit=200,
            ),
            date_axis_config=DateAxisConfig(
                major_unit=label_interval,
                number_format='yyyy/mm',
            ),
        ),
    }


def get_chart2_config(df):
    """
    图表2：组合成立以来收益率走势（面积图+折线图）
    
    特点：
    - 面积图（组合规模，右轴）
    - 折线图（累计收益率，左轴）
    - 长期数据（2013-2024，2800+个点）
    - 日期轴自动间隔
    """
    label_interval = len(df) // 7
    
    return {
        "df": df,
        "categories_col": "分类",
        "series_config": [
            {
                'name': '组合规模(万元)',
                'key': '组合规模(万元)',
                'type': 'area',
                'axis': 'secondary'
            },
            {
                'name': '累计收益率',
                'key': '累计收益率',
                'type': 'line',
                'axis': 'primary'
            },
        ],
        "style_config": StyleConfig(
            color_scheme="aim00",
            line_width_pt=2.0,
            marker_style="none",
        ),
        "layout_config": ChartLayoutConfig(
            title="组合成立以来收益率走势",
            legend_config=LegendConfig(
                position=XL_LEGEND_POSITION.TOP,
                font_size_pt=9,
                font_name="黑体",
            ),
            value_axis_config=ValueAxisConfig(
                number_format="0%",
                font_size_pt=9,
                font_name="黑体",
                has_major_gridlines=True,
            ),
            secondary_value_axis_config=ValueAxisConfig(
                number_format="#,##0",
                font_size_pt=9,
                font_name="黑体",
                has_major_gridlines=False,
                min_value=0,
                max_value=160000,
                major_unit=24000,
            ),
            date_axis_config=DateAxisConfig(
                major_unit=label_interval,
                number_format='yyyy/mm',
            ),
        ),
    }


def get_chart3_config(df):
    """
    图表3：权益仓位走势（面积图+折线图）
    
    特点：
    - 面积图（沪深300指数，右轴）
    - 折线图（权益仓位，左轴）
    - 百分比格式
    """
    label_interval = len(df) // 7
    
    return {
        "df": df,
        "categories_col": "分类",
        "series_config": [
            {
                'name': '沪深300指数(收盘价)',
                'key': '沪深300指数(收盘价)',
                'type': 'area',
                'axis': 'secondary'
            },
            {
                'name': '权益仓位（人社部口径）',
                'key': '权益仓位（人社部口径）',
                'type': 'line',
                'axis': 'primary'
            },
        ],
        "style_config": StyleConfig(
            color_scheme="aim00",
            line_width_pt=2.0,
            marker_style="none",
        ),
        "layout_config": ChartLayoutConfig(
            title="组合权益仓位走势图",
            legend_config=LegendConfig(
                position=XL_LEGEND_POSITION.TOP,
                font_size_pt=9,
                font_name="黑体",
            ),
            value_axis_config=ValueAxisConfig(
                number_format="0%",
                font_size_pt=9,
                font_name="黑体",
                has_major_gridlines=False,
                min_value=0.15,
                max_value=0.30,
            ),
            secondary_value_axis_config=ValueAxisConfig(
                number_format="#,##0",
                font_size_pt=9,
                font_name="黑体",
                has_major_gridlines=False,
                min_value=2950,
                max_value=4450,
                major_unit=200,
            ),
            date_axis_config=DateAxisConfig(
                major_unit=label_interval,
                number_format='yyyy/mm',
            ),
        ),
    }


def get_chart4_config(df):
    """
    图表4：久期走势（柱状图+折线图）
    
    特点：
    - 柱状图（中债指数，右轴）
    - 折线图（久期，左轴）
    - 两位小数格式
    """
    label_interval = len(df) // 7
    
    # 动态计算右轴范围
    bond_index_data = df['中债新综合总财富指数(收盘价)']
    bond_min = bond_index_data.min()
    bond_max = bond_index_data.max()
    bond_range = bond_max - bond_min
    bond_unit = bond_range / 6
    
    return {
        "df": df,
        "categories_col": "分类",
        "series_config": [
            {
                'name': '中债新综合总财富指数(收盘价)',
                'key': '中债新综合总财富指数(收盘价)',
                'type': 'bar',
                'axis': 'secondary'
            },
            {
                'name': '久期',
                'key': '久期',
                'type': 'line',
                'axis': 'primary'
            },
        ],
        "style_config": StyleConfig(
            color_scheme="aim00",
            line_width_pt=2.0,
            marker_style="none",
        ),
        "layout_config": ChartLayoutConfig(
            title="组合久期走势图",
            legend_config=LegendConfig(
                position=XL_LEGEND_POSITION.TOP,
                font_size_pt=9,
                font_name="黑体",
            ),
            value_axis_config=ValueAxisConfig(
                number_format="0.00",
                font_size_pt=9,
                font_name="黑体",
                has_major_gridlines=False,
                min_value=0.2,
                max_value=1.0,
            ),
            secondary_value_axis_config=ValueAxisConfig(
                number_format="#,##0.00",
                font_size_pt=9,
                font_name="黑体",
                has_major_gridlines=False,
                min_value=int(bond_min - bond_unit),
                max_value=int(bond_max + bond_unit),
                major_unit=int(bond_unit),
            ),
            date_axis_config=DateAxisConfig(
                major_unit=label_interval,
                number_format='yyyy/mm',
            ),
        ),
    }


# ============================================================================
# 预设图表映射
# ============================================================================

CHART_PRESET_FUNCTIONS = {
    "图表1 - 当年收益率走势": get_chart1_config,
    "图表2 - 成立以来收益率": get_chart2_config,
    "图表3 - 权益仓位走势": get_chart3_config,
    "图表4 - 久期走势": get_chart4_config,
}


def get_chart_config(chart_key: str, df):
    """
    获取图表配置
    
    Args:
        chart_key: 图表键名
        df: DataFrame
    
    Returns:
        图表配置字典
    """
    if chart_key in CHART_PRESET_FUNCTIONS:
        return CHART_PRESET_FUNCTIONS[chart_key](df)
    else:
        raise ValueError(f"未找到图表配置: {chart_key}")