This commit is contained in:
cjw
2026-02-15 00:54:58 +08:00
parent 0ad995a335
commit 3a8a1ff3cc
2 changed files with 15 additions and 9 deletions
+2 -1
View File
@@ -342,7 +342,8 @@ async def process_file_core(
html_file_path = html_generator.generate_and_save_visualization( html_file_path = html_generator.generate_and_save_visualization(
geometry_data, geometry_data,
Path(file_path).name Path(file_path).name,
cavity_data=detailed_cavity_json
) )
# 保存HTML文件信息 # 保存HTML文件信息
+13 -8
View File
@@ -275,25 +275,30 @@ class HTMLGenerator:
raise raise
def generate_and_save_visualization( def generate_and_save_visualization(
self, self,
geometry_data: Dict[str, Any], geometry_data: Dict[str, Any],
stp_filename: str stp_filename: str,
cavity_data: Optional[Dict[str, Any]] = None
) -> str: ) -> str:
"""生成并保存可视化HTML文件""" """生成并保存可视化HTML文件"""
try: try:
# 生成HTML内容 # 生成HTML内容
html_content = self.generate_3d_viewer_html(geometry_data, stp_filename) html_content = self.generate_3d_viewer_html(
geometry_data,
stp_filename,
cavity_data=cavity_data
)
# 创建文件名 # 创建文件名
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
safe_filename = stp_filename.replace('.', '_').replace(' ', '_') safe_filename = stp_filename.replace('.', '_').replace(' ', '_')
html_filename = f"{safe_filename}_{timestamp}.html" html_filename = f"{safe_filename}_{timestamp}.html"
# 保存文件 # 保存文件
file_path = self.save_html_file(html_content, html_filename) file_path = self.save_html_file(html_content, html_filename)
return file_path return file_path
except Exception as e: except Exception as e:
logger.error(f"生成可视化文件失败: {e}") logger.error(f"生成可视化文件失败: {e}")
raise raise