import os
import tempfile
import pythoncom
import win32com.client
import threading
import shutil
import tkinter as tk
from tkinter import filedialog, ttk, messagebox, scrolledtext
from docx import Document
from PyPDF2 import PdfMerger, PdfReader, PdfWriter
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.lib.colors import red, black, white
from reportlab.platypus import Table, TableStyle
from io import BytesIO
from datetime import datetime
import openpyxl # 用于读取Excel文件
class PDFConverterApp:
def __init__(self, root):
self.root = root
self.root.title("audio_data")
self.root.geometry("800x650") # 增加窗口高度以容纳新控件
self.folders = []
self.log_messages = []
self.output_path = "" # 存储自定义输出路径
self.backup_mode = tk.BooleanVar(value=True) # 添加备份模式开关
self.point_22_mode = tk.BooleanVar(value=False) # 新增22号点位开关
self.output_filename = tk.StringVar(value="听筒磁干扰_Simulation_Result") # 默认文件名
self.create_widgets()
def create_widgets(self):
# 创建顶部框架
top_frame = ttk.Frame(self.root, padding=10)
top_frame.pack(fill=tk.X)
output_frame = ttk.LabelFrame(self.root, text="输出设置", padding=10)
output_frame.pack(fill=tk.X, padx=10, pady=(0, 5))
# 文件名输入框
ttk.Label(output_frame, text="文件名:").grid(row=0, column=0, sticky=tk.W, padx=(0, 5))
filename_entry = ttk.Entry(output_frame, textvariable=self.output_filename, width=30)
filename_entry.grid(row=0, column=1, sticky=tk.W, padx=5)
# 输出路径选择
ttk.Label(output_frame, text="输出路径:").grid(row=0, column=2, sticky=tk.W, padx=(20, 5))
self.path_entry = ttk.Entry(output_frame, width=40, state='readonly')
self.path_entry.grid(row=0, column=3, sticky=tk.EW, padx=5)
browse_btn = ttk.Button(output_frame, text="浏览...", command=self.choose_output_path)
browse_btn.grid(row=0, column=4, padx=(5, 0))
# 设置网格列权重
output_frame.columnconfigure(3, weight=1)
# 添加文件夹按钮
add_btn = ttk.Button(top_frame, text="添加文件夹", command=self.add_folder)
add_btn.pack(side=tk.LEFT, padx=5)
# 移除文件夹按钮
remove_btn = ttk.Button(top_frame, text="移除选中", command=self.remove_selected)
remove_btn.pack(side=tk.LEFT, padx=5)
# 清空列表按钮
clear_btn = ttk.Button(top_frame, text="清空列表", command=self.clear_list)
clear_btn.pack(side=tk.LEFT, padx=5)
# 处理按钮
process_btn = ttk.Button(top_frame, text="开始处理", command=self.start_processing)
process_btn.pack(side=tk.RIGHT, padx=5)
# 创建文件夹列表
list_frame = ttk.LabelFrame(self.root, text="待处理文件夹", padding=10)
list_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=5)
# 滚动条
scrollbar = ttk.Scrollbar(list_frame)
scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
backup_frame = ttk.Frame(output_frame)
backup_frame.grid(row=0, column=5, sticky=tk.W, padx=(20, 0))
# 在输出设置区域添加22号点位复选框
backup_frame = ttk.Frame(output_frame)
backup_frame.grid(row=0, column=5, sticky=tk.W, padx=(20, 0))
self.backup_check = ttk.Checkbutton(
backup_frame,
text="报告存档",
variable=self.backup_mode
)
self.backup_check.pack(side=tk.LEFT)
# 新增22号点位复选框
self.point_22_check = ttk.Checkbutton(
backup_frame,
text="22号点位",
variable=self.point_22_mode
)
self.point_22_check.pack(side=tk.LEFT, padx=(10, 0))
# 文件夹列表
self.folder_list = tk.Listbox(
list_frame,
selectmode=tk.EXTENDED,
yscrollcommand=scrollbar.set,
height=10
)
self.folder_list.pack(fill=tk.BOTH, expand=True)
scrollbar.config(command=self.folder_list.yview)
# 创建日志区域
log_frame = ttk.LabelFrame(self.root, text="处理日志", padding=10)
log_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=5)
# 日志文本框
self.log_text = scrolledtext.ScrolledText(
log_frame,
wrap=tk.WORD,
state=tk.DISABLED
)
self.log_text.pack(fill=tk.BOTH, expand=True)
# 进度条
self.progress = ttk.Progressbar(
self.root,
orient=tk.HORIZONTAL,
mode='determinate'
)
self.progress.pack(fill=tk.X, padx=10, pady=5)
def choose_output_path(self):
"""选择输出文件夹"""
path = filedialog.askdirectory(title="选择输出文件夹")
if path:
self.output_path = path
self.path_entry.config(state='normal')
self.path_entry.delete(0, tk.END)
self.path_entry.insert(0, path)
self.path_entry.config(state='readonly')
self.log(f"已设置输出路径: {path}")
def add_folder(self):
"""添加要处理的文件夹"""
folders = filedialog.askdirectory(
title="选择要处理的文件夹",
mustexist=True
)
if folders:
self.folders.append(folders)
self.folder_list.insert(tk.END, folders)
self.log(f"已添加文件夹: {folders}")
def remove_selected(self):
"""移除选中的文件夹"""
selected = self.folder_list.curselection()
for index in selected[::-1]:
folder = self.folder_list.get(index)
self.folder_list.delete(index)
self.folders.remove(folder)
self.log(f"已移除文件夹: {folder}")
def clear_list(self):
"""清空文件夹列表"""
self.folder_list.delete(0, tk.END)
self.folders = []
self.log("已清空文件夹列表")
def log(self, message):
"""向日志区域添加消息"""
timestamp = datetime.now().strftime("%H:%M:%S")
log_entry = f"[{timestamp}] {message}"
self.log_messages.append(log_entry)
self.log_text.config(state=tk.NORMAL)
self.log_text.insert(tk.END, log_entry + "\n")
self.log_text.config(state=tk.DISABLED)
self.log_text.yview(tk.END) # 自动滚动到底部
self.root.update_idletasks()
def start_processing(self):
"""启动处理过程"""
if not self.folders:
messagebox.showwarning("警告", "请先添加要处理的文件夹")
return
# 禁用处理按钮
self.root.title("Word 转 PDF 合并工具 - 处理中...")
self.progress["value"] = 0
# 在新线程中处理，避免界面冻结
thread = threading.Thread(target=self.process_folders)
thread.daemon = True
thread.start()
# +++ 修改方法：备份时获取校准数据 +++
def backup_data_files(self, folder_path, backup_dir):
"""递归查找并备份所有.xlsx和.csv文件并返回校准数据"""
self.log(f"开始在文件夹中搜索所有Excel和CSV文件: {folder_path}")
backup_count = 0
calibration_data = {'j2': None, 'j3': None} # 存储校准数据
for root, dirs, files in os.walk(folder_path):
for file in files:
if file.lower().endswith(('.xlsx', '.csv')):
file_path = os.path.join(root, file)
file_name = os.path.basename(file_path)
dest_path = os.path.join(backup_dir, file_name)
try:
os.makedirs(backup_dir, exist_ok=True)
shutil.copy2(file_path, dest_path)
backup_count += 1
self.log(f"备份成功: {file_path} → {dest_path}")
except Exception as e:
self.log(f"备份失败 {file_path}: {str(e)}")
self.log(f"共找到并备份 {backup_count} 个Excel和CSV文件")
def process_folders(self):
"""处理多个文件夹中的Word文件"""
try:
# 提前初始化 output_folder
if self.output_path:
output_folder = self.output_path
else:
output_folder = next((p for p in self.folders if os.path.isdir(p)), os.getcwd())
self.log(f"开始处理 {len(self.folders)} 个文件夹...")
# 获取所有文件夹中的Word文件
word_files = self.get_all_word_files(self.folders)
if not word_files:
self.log("没有找到任何Word文档")
return
self.log(f"共找到 {len(word_files)} 个Word文档")
self.progress["maximum"] = len(word_files) + 5 # 文件数 + 合并步骤
backup_root = os.path.join(output_folder, "报告存档") # 统一备份根目录
if self.backup_mode.get():
os.makedirs(backup_root, exist_ok=True)
# 创建临时目录存储转换后的PDF
with tempfile.TemporaryDirectory() as temp_dir:
pdf_files_with_header = []
toc_entries = []
all_tables = {}
calibration_data = {} # 存储所有校准数据
current_page = 1
# 处理每个Word文件
for i, word_file in enumerate(word_files):
self.progress["value"] = i + 1
file_name = os.path.splitext(os.path.basename(word_file))[0]
display_name = file_name
# 修改Word文档逻辑
modified_word_path = word_file
if self.point_22_mode.get() or "GSM" in file_name.upper():
# 创建临时副本进行修改
temp_word_path = os.path.join(temp_dir, os.path.basename(word_file))
shutil.copy2(word_file, temp_word_path)
if self.modify_word_spec(temp_word_path):
modified_word_path = temp_word_path
original_pdf = os.path.join(temp_dir, f"{file_name}_original.pdf")
pdf_with_header = os.path.join(temp_dir, f"{file_name}_with_header.pdf")
if self.backup_mode.get():
try:
# 为每个Word文件创建备份目录
dest_dir = os.path.join(backup_root, file_name)
os.makedirs(dest_dir, exist_ok=True)
# 备份Word文件
word_dest = os.path.join(dest_dir, os.path.basename(modified_word_path))
shutil.copy2(modified_word_path, word_dest)
self.log(f"Word文件备份成功: {word_file} → {word_dest}")
# +++ 备份数据文件并获取校准数据 +++
folder_path = os.path.dirname(word_file)
cal_data = self.backup_data_files(folder_path, dest_dir)
calibration_data[display_name] = cal_data
except OSError as e:
self.log(f"文件备份失败: {e}")
except Exception as e:
self.log(f"未知错误: {e}")
# 提取表格数据
tables = self.extract_spec_table(modified_word_path)
if tables:
all_tables[display_name] = tables
self.log(f"已从 {display_name} 中提取 {len(tables)} 个数据表格")
# 转换为PDF
if self.word_to_pdf(modified_word_path, original_pdf):
# 添加内联标题
if self.add_inline_header(original_pdf, display_name, pdf_with_header):
pdf_files_with_header.append(pdf_with_header)
toc_entries.append((display_name, current_page))
current_page += self.get_pdf_page_count(pdf_with_header)
else:
pdf_files_with_header.append(original_pdf)
toc_entries.append((display_name, current_page))
current_page += self.get_pdf_page_count(original_pdf)
else:
self.log(f"跳过 {display_name}，转换失败")
# 更新进度条
self.progress["value"] = len(word_files) + 1
if not pdf_files_with_header:
self.log("没有成功转换的PDF文件，无法进行合并")
return
# 获取输出路径
if self.output_path:
output_folder = self.output_path
else:
output_folder = next((p for p in self.folders if os.path.isdir(p)), os.getcwd())
# 获取文件名
report_name = self.output_filename.get().strip()
if not report_name:
report_name = self.get_folder_name_parts(self.folders[0]) # 使用默认规则
output_pdf = os.path.join(output_folder, f"{report_name}.pdf")
# 合并PDF
success = self.merge_pdfs_with_summary(
pdf_files_with_header,
toc_entries,
all_tables,
calibration_data, # +++ 新增校准数据参数 +++
output_pdf
)
self.progress["value"] = len(word_files) + 3
if success:
self.log(f"处理完成！输出文件: {output_pdf}")
messagebox.showinfo("完成", f"处理完成！输出文件: {output_pdf}")
else:
self.log("处理失败")
messagebox.showerror("错误", "处理过程中出现错误")
self.root.title("Word 转 PDF 合并工具")
except Exception as e:
self.log(f"处理过程中出现错误: {str(e)}")
messagebox.showerror("错误", f"处理过程中出现错误: {str(e)}")
self.root.title("Word 转 PDF 合并工具")
# 以下是原有的处理函数，保持不变但添加为类方法
def extract_spec_table(self, word_path):
"""从Word文档中提取SPEC(dB)、Simulation和Pass/Fail数据表格"""
try:
doc = Document(word_path)
tables = []
for table in doc.tables:
headers = [cell.text.strip() for cell in table.rows[0].cells]
if "SPEC(dB)" in headers and "Simulation" in headers and "Pass/Fail" in headers:
table_data = []
table_data.append(headers)
for row in table.rows[1:]:
row_data = [cell.text.strip() for cell in row.cells]
table_data.append(row_data)
tables.append(table_data)
return tables
except Exception as e:
self.log(f"提取 {os.path.basename(word_path)} 中的表格时出错: {str(e)}")
return []
def modify_word_spec(self, word_path):
try:
doc = Document(word_path)
filename = os.path.basename(word_path).upper()
has_gsm = "GSM" in filename # 移动到try块内部
# 确定SPEC基准值
if self.point_22_mode.get(): # 22号点位启用
spec_value = 22 if has_gsm else 20
else: # 22号点位未启用
spec_value = 20 if has_gsm else 18
modified = False # 初始化修改标志
# 遍历文档所有表格
for table in doc.tables:
headers = [cell.text.strip() for cell in table.rows[0].cells]
try:
spec_index = headers.index("SPEC(dB)") # 定位SPEC列
sim_index = headers.index("Simulation") # 定位Simulation列
pf_index = headers.index("Pass/Fail") # 定位Pass/Fail列
except ValueError:
continue # 跳过不含目标列的表
# 标记已找到可修改表格
modified = True
# 修改每行数据
for row in table.rows[1:]:
cells = row.cells
# 更新SPEC值
if spec_index < len(cells):
cells[spec_index].text = str(spec_value)
# 更新Pass/Fail状态
if sim_index < len(cells) and pf_index < len(cells):
try:
sim_value = float(cells[sim_index].text)
new_status = "PASS" if sim_value < spec_value else "FAIL"
cells[pf_index].text = new_status
except ValueError:
pass # 忽略格式错误
# 保存修改后的文档
if modified:
doc.save(word_path)
self.log(f"已修改 {os.path.basename(word_path)} 的SPEC值为{spec_value}")
return modified
except Exception as e:
self.log(f"修改 {os.path.basename(word_path)} 失败: {str(e)}")
return False
def add_inline_header(self, pdf_path, title, output_path):
"""在PDF的第一页顶部添加一行红色加粗的标题"""
try:
reader = PdfReader(pdf_path)
writer = PdfWriter()
if len(reader.pages) > 0:
first_page = reader.pages[0]
packet = BytesIO()
can = canvas.Canvas(packet, pagesize=letter)
width, height = letter
font_name = "Helvetica-Bold"
try:
pdfmetrics.registerFont(TTFont('SimSun', 'simsun.ttc'))
pdfmetrics.registerFont(TTFont('SimSun-Bold', 'simsun.ttc'))
font_name = "SimSun-Bold"
except:
pass
can.setFont(font_name, 14)
can.setFillColor(red)
can.drawString(50, height - 50, title)
can.save()
packet.seek(0)
title_reader = PdfReader(packet)
title_page = title_reader.pages[0]
first_page.merge_page(title_page)
writer.add_page(first_page)
for page in reader.pages[1:]:
writer.add_page(page)
with open(output_path, "wb") as f:
writer.write(f)
return True
return False
except Exception as e:
self.log(f"PDF添加标题失败: {str(e)}")
return False
# +++ 修改方法：创建Summary页（核心修改） +++
def create_summary_page(self, toc_entries, all_tables, calibration_data, output_path):
"""创建包含四列数据的Summary页（新增Calibration列）"""
try:
c = canvas.Canvas(output_path, pagesize=letter)
width, height = letter
font_name = "Helvetica"
try:
pdfmetrics.registerFont(TTFont('SimSun', 'simsun.ttc'))
font_name = "SimSun"
except:
pass
# Summary标题
c.setFont(font_name, 24)
c.setFillColor(red)
c.drawCentredString(width / 2.0, height - 50, "Summary")
c.setFillColor(black)
y_position = height - 100
# 添加数据汇总表格
if all_tables:
c.setFont(font_name, 16)
c.drawString(50, y_position, "Data Summary:")
y_position -= 30
c.setFont(font_name, 10)
table_width = width - 100
for doc_name, tables in all_tables.items():
c.setFont(font_name, 12)
c.setFillColor(red)
c.drawString(60, y_position, f"Document: {doc_name}")
y_position -= 20
c.setFillColor(black)
c.setFont(font_name, 10)
# 修复：处理每个表格
for table_data in tables:
# 确保表格有数据行
if len(table_data) < 2: # 至少包含表头+1行数据
continue
# 新表头格式
headers = ["SPEC(dB)", "Simulation", "Pass/Fail"] # 移除Calibration列
# 提取第一行原始数据（跳过表头）
data_row = table_data[1] if len(table_data) > 1 else ["N/A"] * 3
# 确保数据行有足够列
while len(data_row) < 3:
data_row.append("N/A")
# 获取当前文档的J3校准值
doc_cal_data = calibration_data.get(doc_name, {})
j3_value = doc_cal_data.get('j3', 'N/A')
# 创建数据行：前三列使用原始数据，第四列添加J3值
new_row = [
data_row[0], # SPEC(dB)值
data_row[1], # Simulation值
data_row[2], # Pass/Fail值
]
# 表格数据：表头+数据行
modified_table = [headers, new_row]
# 设置四列等宽布局
col_widths = [table_width / 3] * 3
table = Table(modified_table, colWidths=col_widths)
# 设置表格样式
style = TableStyle([
('BACKGROUND', (0, 0), (-1, 0), white),
('TEXTCOLOR', (0, 0), (-1, 0), black),
('ALIGN', (0, 0), (-1, -1), 'CENTER'),
('FONTNAME', (0, 0), (-1, 0), font_name),
('FONTNAME', (0, 1), (-1, -1), font_name),
('BOTTOMPADDING', (0, 0), (-1, 0), 12),
('BACKGROUND', (0, 1), (-1, -1), white),
('GRID', (0, 0), (-1, -1), 1, black)
])
table.setStyle(style)
# 计算表格高度并绘制
table_height = table.wrap(0, 0)[1]
if y_position - table_height < 50:
c.showPage()
y_position = height - 50
c.setFont(font_name, 24)
c.setFillColor(red)
c.drawCentredString(width / 2.0, y_position, "Summary")
y_position -= 50
c.setFillColor(black)
table.drawOn(c, 50, y_position - table_height)
y_position -= (table_height + 20)
c.save()
return output_path
except Exception as e:
self.log(f"创建Summary页失败: {str(e)}")
return None
def word_to_pdf(self, word_path, pdf_path):
"""将Word文档转换为PDF"""
pythoncom.CoInitialize()
try:
word = win32com.client.Dispatch("Word.Application")
word.Visible = False
doc = word.Documents.Open(os.path.abspath(word_path))
doc.SaveAs(os.path.abspath(pdf_path), FileFormat=17)
doc.Close()
word.Quit()
self.log(f"已将 {os.path.basename(word_path)} 转换为PDF")
return True
except Exception as e:
self.log(f"转换 {os.path.basename(word_path)} 时出错: {str(e)}")
return False
finally:
pythoncom.CoUninitialize()
def get_pdf_page_count(self, pdf_path):
"""获取PDF文件的页数"""
try:
reader = PdfReader(pdf_path)
return len(reader.pages)
except:
return 0
def merge_pdfs_with_summary(self, pdf_files, toc_entries, all_tables, output_path):
"""合并PDF文件并添加Summary页"""
try:
with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as summary_file:
summary_path = summary_file.name
# 修复：传递calibration_data参数
self.create_summary_page(toc_entries, all_tables, summary_path)
summary_page_count = self.get_pdf_page_count(summary_path)
updated_toc_entries = [(title, page_num + summary_page_count) for title, page_num in toc_entries]
merger = PdfMerger()
merger.append(summary_path)
current_page = summary_page_count
for pdf, (title, _) in zip(pdf_files, updated_toc_entries):
merger.append(pdf)
merger.add_outline_item(title, current_page)
current_page += self.get_pdf_page_count(pdf)
merger.write(output_path)
merger.close()
os.remove(summary_path)
self.log(f"已成功合并 {len(pdf_files)} 个PDF文件")
return True
except Exception as e:
self.log(f"合并PDF时出错: {str(e)}")
return False
def get_all_word_files(self, folder_paths):
"""获取所有Word文件"""
word_extensions = ['.docx', '.doc']
word_files = []
for folder_path in folder_paths:
if not os.path.isdir(folder_path):
continue
for file in os.listdir(folder_path):
file_ext = os.path.splitext(file)[1].lower()
if file_ext in word_extensions:
word_path = os.path.join(folder_path, file)
word_files.append(word_path)
return word_files
def get_folder_name_parts(self, folder_paths):
"""生成报告文件名"""
if not folder_paths:
return "听筒磁干扰仿真报告"
folder_path = folder_paths[0]
norm_path = os.path.normpath(folder_path)
parts = [p for p in norm_path.split(os.sep) if p]
if len(parts) >= 3:
return f"{parts[-3]}_{parts[-2]}_{parts[-1]}"
elif len(parts) == 2:
return f"{parts[-2]}_{parts[-1]}"
elif len(parts) == 1:
return parts[0]
return "听筒磁干扰仿真报告"
if __name__ == "__main__":
root = tk.Tk()
app = PDFConverterApp(root)
root.mainloop() # 添加这行启动事件循环 [15:52:40] 处理过程中出现错误: PDFConverterApp.merge_pdfs_with_summary() takes 5 positional arguments but 6 were given
最新发布
本文介绍如何安装文件对比工具WinMerge及配置xdocdiff插件的方法。首先从指定网址下载32位版本的WinMerge并完成安装。接着通过另一网址下载xdocdiff插件，按步骤解压并复制必要文件到WinMerge安装目录中。
扫一扫
2220
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
