chentianrui
269 lines
8.7 KiB
Python
269 lines
8.7 KiB
Python
import os
|
|
import subprocess
|
|
import time
|
|
from gradio_client import file
|
|
import requests
|
|
import json
|
|
import shutil
|
|
import threading
|
|
import atexit
|
|
|
|
# ==================== 配置区 ====================
|
|
JAVA_WORKING_DIR = "D:/eclipseworkspace/bwyAnalysis2.3.2/analysis-server"
|
|
JAR_NAME = "booway-analysis-server-null.jar"
|
|
SERVER_URL = "http://localhost:8090/api/doAnalysis"
|
|
# ==============================================
|
|
|
|
# 全局变量：Java 进程
|
|
java_process = None
|
|
|
|
|
|
def start_java_server():
|
|
"""启动 Java 服务"""
|
|
global java_process
|
|
|
|
if java_process is not None and java_process.poll() is None:
|
|
print("✅ Java 服务已在运行。")
|
|
return True
|
|
|
|
try:
|
|
jar_path = os.path.join(JAVA_WORKING_DIR, JAR_NAME)
|
|
if not os.path.exists(jar_path):
|
|
print(f"❌ JAR 文件不存在: {jar_path}")
|
|
return False
|
|
if not os.path.exists(JAVA_WORKING_DIR):
|
|
print(f"❌ Java 工作目录不存在: {JAVA_WORKING_DIR}")
|
|
return False
|
|
|
|
print(f"📁 启动 Java 服务: {jar_path}")
|
|
cmd = ["java", "-Dfile.encoding=UTF-8", "-jar", JAR_NAME]
|
|
java_process = subprocess.Popen(
|
|
cmd,
|
|
cwd=JAVA_WORKING_DIR,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
encoding="utf-8",
|
|
errors="replace",
|
|
)
|
|
|
|
def log_output(pipe, prefix):
|
|
for line in iter(pipe.readline, ""):
|
|
if line.strip():
|
|
print(f"{prefix}{line.strip()}")
|
|
|
|
threading.Thread(target=log_output, args=(java_process.stdout, "【OUT】 "), daemon=True).start()
|
|
threading.Thread(target=log_output, args=(java_process.stderr, "【ERR】 "), daemon=True).start()
|
|
|
|
print("⏳ 等待服务启动（等待 20 秒）...")
|
|
time.sleep(20)
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 启动 Java 服务失败: {e}")
|
|
return False
|
|
|
|
|
|
def check_server_ready(timeout=30):
|
|
"""检查服务是否可用"""
|
|
print("🔍 检测服务是否就绪...")
|
|
start_time = time.time()
|
|
while time.time() - start_time < timeout:
|
|
try:
|
|
requests.get("http://localhost:8090/", timeout=3)
|
|
print("✅ 服务已就绪！")
|
|
return True
|
|
except:
|
|
time.sleep(1)
|
|
print("❌ 服务启动超时。")
|
|
return False
|
|
|
|
|
|
def convert_json_to_readable(input_file, output_file=None):
|
|
"""
|
|
将JSON文件转换为可读格式并保存
|
|
|
|
参数:
|
|
input_file: 输入JSON文件路径
|
|
output_file: 输出文件路径，如果为None则自动生成
|
|
"""
|
|
try:
|
|
with open(input_file, "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
|
|
if output_file is None:
|
|
base_name = os.path.splitext(input_file)[0]
|
|
output_file = f"{base_name}.json"
|
|
|
|
with open(output_file, "w", encoding="utf-8") as f:
|
|
json.dump(data, f, ensure_ascii=False, indent=4)
|
|
|
|
print(f"✅ JSON美化成功: {output_file}")
|
|
return output_file
|
|
except Exception as e:
|
|
print(f"❌ 转换JSON失败 {input_file}: {e}")
|
|
return None
|
|
|
|
|
|
def convert_project_to_json(input_folder, output_folder):
|
|
"""
|
|
将工程文件夹转换为JSON，并移动到输出文件夹（含美化）
|
|
|
|
参数:
|
|
input_folder: 工程文件所在目录（包含 .zwzj 文件）
|
|
output_folder: 转换后的 JSON 文件输出目录
|
|
|
|
返回:
|
|
tuple(bool, int): (是否成功, 成功转移并美化的文件数量)
|
|
"""
|
|
transferred_count = 0 # 记录成功转移的文件数
|
|
success = False # 整体是否成功
|
|
|
|
# --- 路径预处理 ---
|
|
try:
|
|
input_folder = os.path.abspath(input_folder)
|
|
output_folder = os.path.abspath(output_folder)
|
|
except Exception as e:
|
|
print(f"❌ 路径解析失败: {e}")
|
|
return False, 0
|
|
|
|
print(f"📁 输入工程目录: {input_folder}")
|
|
print(f"📁 输出 JSON 目录: {output_folder}")
|
|
# -----------------
|
|
|
|
# 1. 检查输入目录
|
|
if not os.path.exists(input_folder):
|
|
print(f"❌ 输入目录不存在: {input_folder}")
|
|
return False, 0
|
|
if not os.path.isdir(input_folder):
|
|
print(f"❌ 输入路径不是目录: {input_folder}")
|
|
return False, 0
|
|
|
|
# 2. 确保输出目录存在
|
|
try:
|
|
os.makedirs(output_folder, exist_ok=True)
|
|
output_folder = os.path.abspath(output_folder)
|
|
except Exception as e:
|
|
print(f"❌ 创建输出目录失败: {e}")
|
|
return False, 0
|
|
|
|
# 3. 启动 Java 服务
|
|
if not start_java_server():
|
|
return False, 0
|
|
if not check_server_ready():
|
|
return False, 0
|
|
|
|
# 4. 调用 API 开始分析
|
|
params = {"filePath": input_folder.replace("\\", "/")}
|
|
print(f"🌐 调用 API: {SERVER_URL}?{requests.compat.urlencode(params)}")
|
|
|
|
try:
|
|
response = requests.get(SERVER_URL, params=params, timeout=180)
|
|
print(f"📡 响应状态码: {response.status_code}")
|
|
if response.status_code != 200:
|
|
print(f"❌ API 调用失败: {response.text}")
|
|
return False, 0
|
|
result = response.json()
|
|
if not result.get("success", False):
|
|
print(f"❌ 服务返回失败: {result.get('msg', '未知错误')}")
|
|
return False, 0
|
|
except Exception as e:
|
|
print(f"❌ 调用 API 出错: {e}")
|
|
return False, 0
|
|
|
|
# 5. 等待并检测 JSON 文件生成完成（改进版）
|
|
print("⏳ 正在等待 JSON 文件生成完成...")
|
|
json_files = []
|
|
max_wait_time = 600
|
|
check_interval = 2
|
|
|
|
start_time = time.time()
|
|
while time.time() - start_time < max_wait_time:
|
|
# 获取目录中的所有文件
|
|
all_files = os.listdir(input_folder)
|
|
# 获取JSON文件列表
|
|
current_jsons = [f for f in all_files if f.endswith(".json")]
|
|
# 获取非JSON文件列表
|
|
non_json_files = [f for f in all_files if not f.endswith(".json")]
|
|
|
|
# 计算文件数量
|
|
json_count = len(current_jsons)
|
|
non_json_count = len(non_json_files)
|
|
|
|
print(
|
|
f" ├─ 第 {int(time.time()-start_time)} 秒: 发现 {json_count} 个 JSON 文件, {non_json_count} 个非 JSON 文件"
|
|
)
|
|
|
|
# 如果JSON文件和非JSON文件数量相等且都大于0，认为处理完成
|
|
if json_count > 0 and json_count == non_json_count:
|
|
json_files = current_jsons
|
|
print(f"✅ 文件生成完成，JSON文件和非JSON文件数量相等（各{json_count}个）。")
|
|
break
|
|
|
|
time.sleep(check_interval)
|
|
else:
|
|
json_files = [f for f in os.listdir(input_folder) if f.endswith(".json")]
|
|
if json_files:
|
|
print(f"⚠️ 警告：等待超时，但已发现 {len(json_files)} 个 JSON 文件，继续处理。")
|
|
else:
|
|
print(f"❌ 在 {max_wait_time} 秒内未生成任何 JSON 文件。")
|
|
return False, 0
|
|
|
|
# 6. 查找所有 .json 文件
|
|
json_files = [f for f in os.listdir(input_folder) if f.endswith(".json")]
|
|
if not json_files:
|
|
print(f"❌ 未在 {input_folder} 中生成 JSON 文件。")
|
|
return False, 0
|
|
|
|
print(f"📄 发现 {len(json_files)} 个 JSON 文件: {json_files}")
|
|
|
|
# 7. 转换为可读格式并移动到输出目录
|
|
transferred_count = 0
|
|
for file_name in json_files:
|
|
src_path = os.path.join(input_folder, file_name)
|
|
readable_path = os.path.join(output_folder, f"{os.path.splitext(file_name)[0]}.json")
|
|
|
|
if convert_json_to_readable(src_path, readable_path):
|
|
transferred_count += 1
|
|
else:
|
|
print(f"⚠️ 跳过文件: {file_name}")
|
|
|
|
print(f"✅ 共转移并美化 {transferred_count}/{len(json_files)} 个 JSON 文件到: {output_folder}")
|
|
|
|
# ✅ 返回成功状态和实际转移数量
|
|
success = True
|
|
return success, transferred_count
|
|
|
|
|
|
def cleanup():
|
|
"""清理 Java 进程"""
|
|
global java_process
|
|
if java_process and java_process.poll() is None:
|
|
print("\n🛑 正在关闭 Java 服务...")
|
|
java_process.terminate()
|
|
try:
|
|
java_process.wait(timeout=5)
|
|
except subprocess.TimeoutExpired:
|
|
java_process.kill()
|
|
print("✅ Java 服务已关闭。")
|
|
|
|
|
|
# 注册退出钩子
|
|
atexit.register(cleanup)
|
|
|
|
|
|
# ==================== 使用示例 ====================
|
|
if __name__ == "__main__":
|
|
# 示例调用
|
|
input_dir = r"project2json/uploads"
|
|
output_dir = r"project2json/outputs/json"
|
|
|
|
success = convert_project_to_json(input_dir, output_dir)
|
|
|
|
if success:
|
|
print("\n🎉 ✅ 全部流程执行成功！")
|
|
else:
|
|
print("\n❌ 流程执行失败，请检查日志。")
|
|
|
|
input("\n按回车键退出...")
|