使用弹出窗口将Dash数据表保存到excel或csv

import dash
import dash_html_components as html
import dash_table
import pandas as pd

app = dash.Dash(__name__)

# 创建一个数据表
data = pd.DataFrame({'Column 1': [1, 2, 3, 4],
                     'Column 2': ['A', 'B', 'C', 'D']})

app.layout = html.Div([
    dash_table.DataTable(
        id='table',
        columns=[{"name": i, "id": i} for i in data.columns],
        data=data.to_dict('records')
    ),
    html.Button('保存为Excel', id='save-excel-button'),
    html.Button('保存为CSV', id='save-csv-button')
])

# 回调函数，处理保存按钮的点击事件
@app.callback(
    dash.dependencies.Output('save-excel-button', 'children'),
    dash.dependencies.Output('save-csv-button', 'children'),
    dash.dependencies.Input('save-excel-button', 'n_clicks'),
    dash.dependencies.Input('save-csv-button', 'n_clicks')
)
def save_table_to_file(excel_clicks, csv_clicks):
    ctx = dash.callback_context
    button_id = ctx.triggered[0]['prop_id'].split('.')[0]

    if button_id == 'save-excel-button':
        # 保存为Excel文件
        data.to_excel('table.xlsx', index=False)
        return '保存为Excel成功', '保存为CSV'
    elif button_id == 'save-csv-button':
        # 保存为CSV文件
        data.to_csv('table.csv', index=False)
        return '保存为Excel', '保存为CSV成功'
    else:
        return '保存为Excel', '保存为CSV'

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