if True:
    import sys
    sys.path.append('../..')
    import dash
    import json
    from dash import html
    import feffery_antd_charts as fact
    from dash.dependencies import Input, Output

app = dash.Dash(__name__)

mock_data = {
    'name': 'root',
    'children': [
        {
            'name': '分类 1',
            'value': 560,
        },
        {
            'name': '分类 2',
            'value': 500,
        },
        {
            'name': '分类 3',
            'value': 150,
        },
        {
            'name': '分类 4',
            'value': 140,
        },
        {
            'name': '分类 5',
            'value': 115,
        },
        {
            'name': '分类 6',
            'value': 95,
        },
        {
            'name': '分类 7',
            'value': 90,
        },
        {
            'name': '分类 8',
            'value': 75,
        },
        {
            'name': '分类 9',
            'value': 98,
        },
        {
            'name': '分类 10',
            'value': 60,
        },
        {
            'name': '分类 11',
            'value': 45,
        },
        {
            'name': '分类 12',
            'value': 40,
        },
        {
            'name': '分类 13',
            'value': 40,
        },
        {
            'name': '分类 14',
            'value': 35,
        },
        {
            'name': '分类 15',
            'value': 40,
        },
        {
            'name': '分类 16',
            'value': 40,
        },
        {
            'name': '分类 17',
            'value': 40,
        },
        {
            'name': '分类 18',
            'value': 30,
        },
        {
            'name': '分类 19',
            'value': 28,
        },
        {
            'name': '分类 20',
            'value': 16,
        },
    ],
}

app.layout = html.Div(
    [
        html.Div(
            fact.AntdTreemap(
                id='treemap-demo',
                data=mock_data,
                colorField='name'
            ),
            style={
                'height': '400px'
            }
        ),
        html.Pre(
            id='output'
        )
    ],
    style={
        'padding': '50px 150px'
    }
)


if __name__ == '__main__':
    app.run(debug=True)