From 9f2d358c57aea4a53a28477dd88a984c3eee4e9c Mon Sep 17 00:00:00 2001 From: jinye_huang Date: Sat, 10 May 2025 21:43:10 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E4=BA=86=E4=BF=9D=E5=AD=98?= =?UTF-8?q?=E7=9A=84=E6=96=87=E6=A1=A3=E7=BB=93=E6=9E=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/extract_and_render.py | 73 +++++++---------------------------- 1 file changed, 14 insertions(+), 59 deletions(-) diff --git a/scripts/extract_and_render.py b/scripts/extract_and_render.py index faa6a33..7366dae 100644 --- a/scripts/extract_and_render.py +++ b/scripts/extract_and_render.py @@ -71,7 +71,6 @@ def process_result_directory(source_dir, output_dir, run_id=None): "ArticleJsonPath", "OutputTxtPath", "PosterPath", - "CollagePath", "AdditionalImagesCount", "Status", "Details" @@ -107,7 +106,6 @@ def process_result_directory(source_dir, output_dir, run_id=None): "ArticleJsonPath": "", "OutputTxtPath": "", "PosterPath": "", - "CollagePath": "", "AdditionalImagesCount": 0, "Status": "Processing", "Details": "" @@ -140,11 +138,6 @@ def process_result_directory(source_dir, output_dir, run_id=None): with open(txt_path, 'w', encoding='utf-8') as f_txt: f_txt.write(txt_content) print(f" - 成功写入文本文件: {txt_path}") - - # 同时保存原始JSON - json_output_path = os.path.join(output_entry_path, "article.json") - shutil.copy2(json_path, json_output_path) - print(f" - 复制原始JSON文件: {json_output_path}") except Exception as e: record["Status"] = "Partial" record["Details"] += f"写入文本文件失败: {e}; " @@ -164,13 +157,6 @@ def process_result_directory(source_dir, output_dir, run_id=None): try: shutil.copy2(poster_jpg_path, output_poster_path) print(f" - 成功复制海报图片: {output_poster_path}") - - # 复制元数据 - poster_metadata_path = os.path.join(poster_dir, "poster_metadata.json") - if os.path.exists(poster_metadata_path): - output_poster_metadata = os.path.join(output_entry_path, "poster_metadata.json") - shutil.copy2(poster_metadata_path, output_poster_metadata) - print(f" - 复制海报元数据: {output_poster_metadata}") except Exception as e: record["Status"] = "Partial" record["Details"] += f"复制海报图片失败: {e}; " @@ -180,33 +166,7 @@ def process_result_directory(source_dir, output_dir, run_id=None): record["Details"] += "海报图片不存在; " print(f" - 警告: {record['Details']}") - # 3. 处理拼贴图 - collage_dir = os.path.join(entry_path, "collage_img") - collage_path = os.path.join(collage_dir, "collage.png") - output_collage_path = os.path.join(output_entry_path, "collage.png") - record["CollagePath"] = output_collage_path - - if os.path.exists(collage_path): - try: - shutil.copy2(collage_path, output_collage_path) - print(f" - 成功复制拼贴图: {output_collage_path}") - - # 复制元数据 - collage_metadata_path = os.path.join(collage_dir, "collage_metadata.json") - if os.path.exists(collage_metadata_path): - output_collage_metadata = os.path.join(output_entry_path, "collage_metadata.json") - shutil.copy2(collage_metadata_path, output_collage_metadata) - print(f" - 复制拼贴图元数据: {output_collage_metadata}") - except Exception as e: - record["Status"] = "Partial" - record["Details"] += f"复制拼贴图失败: {e}; " - print(f" - 错误: {record['Details']}") - else: - record["Status"] = "Partial" - record["Details"] += "拼贴图不存在; " - print(f" - 警告: {record['Details']}") - - # 4. 处理额外图片 + # 3. 处理额外图片 image_dir = os.path.join(entry_path, "image") output_image_dir = os.path.join(output_entry_path, "additional_images") @@ -223,13 +183,6 @@ def process_result_directory(source_dir, output_dir, run_id=None): # 复制图片 shutil.copy2(source_file, dest_file) image_count += 1 - - # 复制相关元数据 - metadata_filename = filename.replace(".jpg", "_metadata.json") - metadata_path = os.path.join(image_dir, metadata_filename) - if os.path.exists(metadata_path): - dest_metadata = os.path.join(output_image_dir, metadata_filename) - shutil.copy2(metadata_path, dest_metadata) record["AdditionalImagesCount"] = image_count print(f" - 复制了 {image_count} 张额外图片到: {output_image_dir}") @@ -264,22 +217,24 @@ def process_result_directory(source_dir, output_dir, run_id=None): print(f"结果保存在: {output_dir}") def main(): - parser = argparse.ArgumentParser(description="从TravelContentCreator结果目录提取内容并渲染到指定目录") - parser.add_argument("--source", type=str, required=True, help="源目录路径") - parser.add_argument("--output", type=str, required=True, help="输出目录路径") - parser.add_argument("--run-id", type=str, help="自定义运行ID") - - args = parser.parse_args() + # parser = argparse.ArgumentParser(description="从TravelContentCreator结果目录提取内容并渲染到指定目录") + # parser.add_argument("--source", type=str, required=True, help="源目录路径") + # parser.add_argument("--output", type=str, required=True, help="输出目录路径") + # parser.add_argument("--run-id", type=str, help="自定义运行ID") + # args = parser.parse_args() + source = "" + output = "" + run_id = datetime.now().strftime("%Y%m%d_%H%M%S") print("-" * 60) print(f"开始提取和渲染流程") - print(f"源目录: {args.source}") - print(f"输出目录: {args.output}") - if args.run_id: - print(f"运行ID: {args.run_id}") + print(f"源目录: {source}") + print(f"输出目录: {output}") + if run_id: + print(f"运行ID: {run_id}") print("-" * 60) - process_result_directory(args.source, args.output, args.run_id) + process_result_directory(source, output, run_id) print("\n脚本执行完毕.")