WPS二次开发入门：使用Python API实现报表自动化与数据可视化

import win32com.client

try:
    # 尝试启动WPS表格应用
    wps_app = win32com.client.Dispatch("et.Application") # “et”代表WPS表格
    # 让WPS应用可见（调试时有用）
    wps_app.Visible = True
    print("WPS表格COM对象创建成功！")
    
    # 创建一个新工作簿
    workbook = wps_app.Workbooks.Add()
    # 获取第一个工作表
    sheet = workbook.Worksheets(1)
    # 在A1单元格写入测试文本
    sheet.Range("A1").Value = "Hello, WPS Python API!"
    
    # 暂停一下，让你看到效果
    input("按回车键继续并关闭工作簿...")
    
    # 不保存，关闭工作簿
    workbook.Close(SaveChanges=False)
    # 退出WPS表格应用
    wps_app.Quit()
    
    print("测试完成，WPS环境正常。")
    
except Exception as e:
    print(f"出现错误：{e}")
    print("请检查：1. WPS Office是否已安装。 2. 是否安装了pywin32库。")

import win32com.client
import os

# 1. 启动WPS表格，并设置为不可见（后台运行）
et_app = win32com.client.Dispatch("et.Application")
et_app.Visible = False # 后台运行，不显示界面
et_app.DisplayAlerts = False # 关闭警告提示，如“是否保存”

# 2. 打开一个已存在的工作簿
workbook_path = r"C:\Your\Data\Path\销售数据.xlsx"
if os.path.exists(workbook_path):
    workbook = et_app.Workbooks.Open(workbook_path)
else:
    # 如果文件不存在，创建一个新的
    workbook = et_app.Workbooks.Add()
    workbook.SaveAs(workbook_path)

# 3. 操作工作表
sheet = workbook.Worksheets("Sheet1") # 通过名称获取，或使用 Worksheets(1) 通过索引获取
# 或激活一个工作表
sheet.Activate()

# 4. 读写单元格数据
# 写入单个值
sheet.Range("A1").Value = "产品名称"
sheet.Range("B1").Value = "销售额"
# 写入一个列表（一行数据）
data_row = ["产品A", 12000]
sheet.Range("A2:B2").Value = data_row
# 写入一个二维列表（一个区域）
data_area = [["产品B", 15000], ["产品C", 9800], ["产品D", 21000]]
sheet.Range("A3:C5").Value = data_area # 注意：会自动扩展到相应大小

# 读取数据
cell_value = sheet.Range("A1").Value
print(f"A1单元格的值是：{cell_value}")
# 读取一个区域到Python列表
used_range = sheet.UsedRange # 获取已使用的区域
data_from_sheet = used_range.Value # 返回一个二维元组（tuple of tuples）
print(f"从工作表读取的数据：{data_from_sheet}")

# 5. 设置单元格格式
target_cell = sheet.Range("B1")
target_cell.Font.Bold = True # 字体加粗
target_cell.Font.Color = 0xFF0000 # 红色字体 (BGR格式)
target_cell.Interior.Color = 0xFFFF00 # 黄色背景
target_cell.HorizontalAlignment = -4108 # 居中（xlCenter）

# 6. 保存与关闭
workbook.Save() # 保存
# workbook.SaveAs(r"新路径\新文件名.xlsx") # 另存为
workbook.Close()
et_app.Quit()

print("基础操作执行完毕。")

import win32com.client
import pandas as pd
import matplotlib.pyplot as plt
from datetime import datetime
import os

# 步骤1: 使用pandas进行数据处理
def process_sales_data(csv_path):
    """读取CSV，清洗并汇总数据"""
    df = pd.read_csv(csv_path, encoding='utf-8')
    
    # 基础清洗
    df.drop_duplicates(inplace=True)
    df['销售额'] = df['单价'] * df['数量']
    df.fillna({'区域': '未知'}, inplace=True)
    
    # 数据透视：按类别和区域汇总销售额
    pivot_table = pd.pivot_table(df, 
                                 values='销售额', 
                                 index='产品类别', 
                                 columns='区域', 
                                 aggfunc='sum', 
                                 fill_value=0,
                                 margins=True, 
                                 margins_name='总计')
    
    # 生成图表数据
    category_sales = df.groupby('产品类别')['销售额'].sum().sort_values(ascending=False)
    
    return df, pivot_table, category_sales

# 步骤2: 将数据与图表写入WPS表格
def create_excel_report(workbook, raw_df, pivot_df, chart_data):
    """在WPS工作簿中创建数据表和图表"""
    # 2.1 写入原始数据
    raw_sheet = workbook.Worksheets.Add()
    raw_sheet.Name = "原始订单数据"
    # 将DataFrame的列标题写入第一行
    headers = raw_df.columns.tolist()
    for i, header in enumerate(headers):
        raw_sheet.Cells(1, i+1).Value = header
    # 将DataFrame数据写入（从第二行开始）
    # 注意：win32com写入大量数据较慢，对于大数据量，可考虑先转为列表再一次性写入
    for r_idx, row in raw_df.iterrows():
        for c_idx, value in enumerate(row):
            raw_sheet.Cells(r_idx+2, c_idx+1).Value = value
    
    # 2.2 写入数据透视汇总表
    pivot_sheet = workbook.Worksheets.Add(Before=raw_sheet) # 插入在原始数据表之前
    pivot_sheet.Name = "销售汇总"
    # 将pivot_table的索引和列标题写入
    # 这里需要处理MultiIndex，简化操作：将pivot_df重置索引
    pivot_df_reset = pivot_df.reset_index()
    # 使用更高效的数组写入方式（将DataFrame转换为列表）
    data_to_write = [pivot_df_reset.columns.tolist()] + pivot_df_reset.values.tolist()
    pivot_sheet.Range(pivot_sheet.Cells(1,1), 
                       pivot_sheet.Cells(len(data_to_write), len(data_to_write[0]))).Value = data_to_write
    
    # 2.3 创建图表
    # 首先，将图表数据写入到一个新区域（例如从H列开始）
    chart_sheet = pivot_sheet # 在汇总表上添加图表
    start_row = 2
    start_col = pivot_df_reset.shape[1] + 3 # 隔开几列
    chart_sheet.Cells(start_row, start_col).Value = "产品类别"
    chart_sheet.Cells(start_row, start_col+1).Value = "销售额"
    
    for i, (category, sales) in enumerate(chart_data.items()):
        chart_sheet.Cells(start_row+1+i, start_col).Value = category
        chart_sheet.Cells(start_row+1+i, start_col+1).Value = sales
    
    # 创建饼图
    chart_range = chart_sheet.Range(
        chart_sheet.Cells(start_row, start_col+1),
        chart_sheet.Cells(start_row+len(chart_data), start_col+1)
    )
    category_range = chart_sheet.Range(
        chart_sheet.Cells(start_row, start_col),
        chart_sheet.Cells(start_row+len(chart_data), start_col)
    )
    
    chart = workbook.Charts.Add()
    chart.ChartType = 5 # xlPie
    chart.SetSourceData(Source=chart_range)
    # 设置图表数据标签和标题（WPS API中设置可能略有不同，此为通用思路）
    # 实际中可能需要更详细的属性设置
    chart.HasTitle = True
    chart.ChartTitle.Text = "各产品类别销售额占比"
    chart.Location(Where=2, Name=pivot_sheet.Name) # 将图表作为对象嵌入到指定工作表
    
    # 2.4 美化格式：设置汇总表标题行样式
    header_range = pivot_sheet.Range("A1").CurrentRegion.Rows(1)
    header_range.Font.Bold = True
    header_range.Interior.Color = 0x4F81BD # 浅蓝色
    header_range.Font.Color = 0xFFFFFF # 白色字体
    # 自动调整列宽
    pivot_sheet.Columns.AutoFit()

# 步骤3: 填充WPS文字报告模板
def fill_word_report(template_path, output_path, summary_data):
    """根据汇总数据，填充预定义的Word报告模板"""
    try:
        wps_app = win32com.client.Dispatch("wps.Application")
        wps_app.Visible = False
        doc = wps_app.Documents.Open(template_path)
        
        # 假设模板中有预定义的书签(Bookmark)或特定的占位文本
        # 方式A：替换占位文本（如果模板中有如`{{销售总额}}`的文本）
        # 遍历故事范围(StoryRanges)进行替换（简化示例）
        for story in [0, 1, 4, 5, 6]: # wdMainTextStory, wdFootnotesStory等
            try:
                content = doc.StoryRanges(story)
                content.Find.Execute(FindText="{{销售总额}}", ReplaceWith=f"{summary_data['total_sales']:,.2f}")
                content.Find.Execute(FindText="{{最高品类}}", ReplaceWith=summary_data['top_category'])
            except:
                pass
        
        # 方式B：如果模板是表格形式，可以直接定位单元格写入
        # tables = doc.Tables
        # if tables.Count > 0:
        #     tables(1).Cell(2,2).Range.Text = str(summary_data['total_sales'])
        
        doc.SaveAs(output_path)
        doc.Close()
        wps_app.Quit()
        print(f"Word报告已生成：{output_path}")
    except Exception as e:
        print(f"生成Word报告时出错：{e}")

# 主函数：串联整个流程
def main():
    today_str = datetime.now().strftime("%Y%m%d")
    csv_file = f"./data/sales_orders_{today_str}.csv"
    excel_output = f"./reports/销售日报_{today_str}.xlsx"
    word_template = "./templates/日报模板.docx"
    word_output = f"./reports/销售分析报告_{today_str}.docx"
    
    # 1. 数据处理
    if not os.path.exists(csv_file):
        print(f"数据文件{csv_file}不存在，请检查。")
        return
    raw_data, pivot_data, chart_data = process_sales_data(csv_file)
    
    # 准备传递给Word的摘要数据
    summary_info = {
        'total_sales': pivot_data.loc['总计', '总计'], # 假设透视表有总计行列
        'top_category': chart_data.index[0] if not chart_data.empty else "无"
    }
    
    # 2. 创建Excel报表
    et_app = win32com.client.Dispatch("et.Application")
    et_app.Visible = False
    workbook = et_app.Workbooks.Add()
    
    create_excel_report(workbook, raw_data, pivot_data, chart_data)
    
    workbook.SaveAs(excel_output)
    workbook.Close()
    et_app.Quit()
    print(f"Excel报表已生成：{excel_output}")
    
    # 3. 生成Word报告
    if os.path.exists(word_template):
        fill_word_report(word_template, word_output, summary_info)
    else:
        print("Word模板不存在，跳过Word报告生成。")

if __name__ == "__main__":
    main()

# 假设已经有一个包含透视表的工作表 pivot_sheet 和透视表对象 pivot_table
# ... 创建透视表的代码 ...
# 添加切片器（WPS API可能不完全支持所有Excel对象，此为概念性代码）
# 通常需要引用正确的透视表对象，并调用Slicers.Add方法
# slicer_cache = workbook.SlicerCaches.Add(PivotTable=pivot_table, Field="产品类别")
# slicer = slicer_cache.Slicers.Add(SlicerDestination=pivot_sheet, Name="品类筛选")