310 lines
11 KiB
Python
310 lines
11 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
SUWood数据桌面保存器
|
||
将实时接收到的SUWood服务器数据转换成JSON格式，保存到桌面001.json文件
|
||
"""
|
||
|
||
from data_listener import create_listener
|
||
import os
|
||
import sys
|
||
import json
|
||
import time
|
||
import threading
|
||
from datetime import datetime
|
||
from typing import Dict, Any, List
|
||
|
||
# 添加当前目录到路径
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
if current_dir not in sys.path:
|
||
sys.path.insert(0, current_dir)
|
||
|
||
|
||
class DesktopDataSaver:
|
||
"""桌面数据保存器"""
|
||
|
||
def __init__(self, server_host="127.0.0.1", server_port=7999):
|
||
self.listener = None
|
||
self.server_host = server_host
|
||
self.server_port = server_port
|
||
|
||
# 获取桌面路径
|
||
self.desktop_path = self._get_desktop_path()
|
||
self.json_file_path = os.path.join(self.desktop_path, "001.json")
|
||
|
||
# 数据存储
|
||
self.collected_data = []
|
||
self.last_save_time = None
|
||
|
||
# 保存控制
|
||
self.auto_save_interval = 2 # 2秒自动保存一次
|
||
self.max_data_count = 1000 # 最多保存1000条数据
|
||
|
||
print(f"📁 JSON文件路径: {self.json_file_path}")
|
||
|
||
def _get_desktop_path(self) -> str:
|
||
"""获取桌面路径"""
|
||
# Windows
|
||
if os.name == 'nt':
|
||
desktop = os.path.join(os.path.expanduser('~'), 'Desktop')
|
||
if os.path.exists(desktop):
|
||
return desktop
|
||
# 中文系统可能是"桌面"
|
||
desktop_cn = os.path.join(os.path.expanduser('~'), '桌面')
|
||
if os.path.exists(desktop_cn):
|
||
return desktop_cn
|
||
|
||
# macOS/Linux
|
||
desktop = os.path.join(os.path.expanduser('~'), 'Desktop')
|
||
if os.path.exists(desktop):
|
||
return desktop
|
||
|
||
# 如果都找不到，使用当前目录
|
||
print("⚠️ 未找到桌面目录，将保存到当前目录")
|
||
return os.getcwd()
|
||
|
||
def start_monitoring(self):
|
||
"""开始监控并保存数据"""
|
||
print(f"🚀 开始监控SUWood服务器数据: {self.server_host}:{self.server_port}")
|
||
print(f"💾 数据将保存到: {self.json_file_path}")
|
||
|
||
# 创建监听器
|
||
self.listener = create_listener(self.server_host, self.server_port)
|
||
|
||
# 添加数据处理回调
|
||
self.listener.add_callback(self.on_data_received)
|
||
|
||
# 开始监听
|
||
self.listener.start_listening()
|
||
|
||
# 启动自动保存线程
|
||
self._start_auto_save_thread()
|
||
|
||
def stop_monitoring(self):
|
||
"""停止监控"""
|
||
if self.listener:
|
||
self.listener.stop_listening()
|
||
|
||
# 最后保存一次数据
|
||
self._save_to_json()
|
||
print("⛔ 数据监控已停止")
|
||
|
||
def on_data_received(self, data: Dict[str, Any]):
|
||
"""处理接收到的数据"""
|
||
# 添加时间戳
|
||
processed_data = {
|
||
"timestamp": datetime.now().isoformat(),
|
||
"raw_data": data
|
||
}
|
||
|
||
# 添加到收集列表
|
||
self.collected_data.append(processed_data)
|
||
|
||
# 保持数据量在限制范围内
|
||
if len(self.collected_data) > self.max_data_count:
|
||
self.collected_data.pop(0) # 移除最旧的数据
|
||
|
||
# 打印接收信息
|
||
data_size = len(json.dumps(data, ensure_ascii=False))
|
||
print(
|
||
f"📥 [{datetime.now().strftime('%H:%M:%S')}] 收到数据: {data_size} 字节，总计: {len(self.collected_data)} 条")
|
||
|
||
# 如果数据包含重要信息，立即保存
|
||
if self._is_important_data(data):
|
||
print("⚡ 检测到重要数据，立即保存...")
|
||
self._save_to_json()
|
||
|
||
def _is_important_data(self, data: Dict[str, Any]) -> bool:
|
||
"""判断是否为重要数据（需要立即保存）"""
|
||
# 检查是否包含几何数据
|
||
if 'data' in data:
|
||
data_content = data['data']
|
||
important_fields = ['meshes', 'objects',
|
||
'vertices', 'faces', 'cmds']
|
||
return any(field in data_content for field in important_fields)
|
||
|
||
# 检查是否为成功的命令响应
|
||
if data.get('ret') == 1:
|
||
return True
|
||
|
||
return False
|
||
|
||
def _start_auto_save_thread(self):
|
||
"""启动自动保存线程"""
|
||
def auto_save_loop():
|
||
while self.listener and self.listener.running:
|
||
time.sleep(self.auto_save_interval)
|
||
if self.collected_data:
|
||
self._save_to_json()
|
||
|
||
save_thread = threading.Thread(target=auto_save_loop, daemon=True)
|
||
save_thread.start()
|
||
print(f"🔄 自动保存线程已启动（间隔: {self.auto_save_interval}秒）")
|
||
|
||
def _save_to_json(self):
|
||
"""保存数据到JSON文件"""
|
||
if not self.collected_data:
|
||
return
|
||
|
||
try:
|
||
# 准备保存的数据结构
|
||
save_data = {
|
||
"save_info": {
|
||
"save_time": datetime.now().isoformat(),
|
||
"total_records": len(self.collected_data),
|
||
"data_source": f"{self.server_host}:{self.server_port}",
|
||
"file_version": "1.0"
|
||
},
|
||
"data_records": self.collected_data
|
||
}
|
||
|
||
# 写入JSON文件
|
||
with open(self.json_file_path, 'w', encoding='utf-8') as f:
|
||
json.dump(save_data, f, ensure_ascii=False, indent=2)
|
||
|
||
self.last_save_time = datetime.now()
|
||
file_size = os.path.getsize(self.json_file_path)
|
||
|
||
print(
|
||
f"💾 数据已保存: {len(self.collected_data)} 条记录，文件大小: {file_size/1024:.1f}KB")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 保存JSON文件失败: {e}")
|
||
|
||
def get_statistics(self) -> Dict[str, Any]:
|
||
"""获取统计信息"""
|
||
file_exists = os.path.exists(self.json_file_path)
|
||
file_size = os.path.getsize(self.json_file_path) if file_exists else 0
|
||
|
||
stats = {
|
||
"collected_records": len(self.collected_data),
|
||
"json_file_path": self.json_file_path,
|
||
"json_file_exists": file_exists,
|
||
"json_file_size_kb": file_size / 1024 if file_size > 0 else 0,
|
||
"last_save_time": self.last_save_time.isoformat() if self.last_save_time else None,
|
||
"listener_running": self.listener.running if self.listener else False
|
||
}
|
||
|
||
if self.listener:
|
||
listener_stats = self.listener.get_statistics()
|
||
stats.update({
|
||
"total_received": listener_stats.get("total_received", 0),
|
||
"queue_size": listener_stats.get("queue_size", 0),
|
||
"is_connected": listener_stats.get("is_connected", False)
|
||
})
|
||
|
||
return stats
|
||
|
||
def manually_save(self):
|
||
"""手动保存数据"""
|
||
print("🖱️ 手动保存数据...")
|
||
self._save_to_json()
|
||
|
||
def clear_data(self):
|
||
"""清空收集的数据"""
|
||
self.collected_data.clear()
|
||
print("🗑️ 已清空收集的数据")
|
||
|
||
def load_existing_data(self) -> bool:
|
||
"""加载已存在的JSON文件数据"""
|
||
if not os.path.exists(self.json_file_path):
|
||
print("📄 桌面上没有找到001.json文件")
|
||
return False
|
||
|
||
try:
|
||
with open(self.json_file_path, 'r', encoding='utf-8') as f:
|
||
existing_data = json.load(f)
|
||
|
||
if 'data_records' in existing_data:
|
||
self.collected_data = existing_data['data_records']
|
||
print(f"📂 已加载现有数据: {len(self.collected_data)} 条记录")
|
||
return True
|
||
else:
|
||
print("⚠️ 现有JSON文件格式不匹配")
|
||
return False
|
||
|
||
except Exception as e:
|
||
print(f"❌ 加载现有数据失败: {e}")
|
||
return False
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("💾 SUWood数据桌面保存器")
|
||
print("=" * 50)
|
||
print("功能: 实时接收SUWood服务器数据并保存到桌面001.json")
|
||
print()
|
||
|
||
# 创建数据保存器
|
||
saver = DesktopDataSaver()
|
||
|
||
# 询问是否加载现有数据
|
||
if os.path.exists(saver.json_file_path):
|
||
response = input(
|
||
"📂 发现桌面已存在001.json文件，是否加载现有数据？(y/n): ").strip().lower()
|
||
if response == 'y':
|
||
saver.load_existing_data()
|
||
|
||
try:
|
||
# 开始监控
|
||
saver.start_monitoring()
|
||
|
||
print()
|
||
print("⌨️ 控制命令:")
|
||
print(" - 按 Enter 查看统计信息")
|
||
print(" - 输入 's' 手动保存数据")
|
||
print(" - 输入 'c' 清空收集的数据")
|
||
print(" - 输入 'q' 或 Ctrl+C 退出")
|
||
print()
|
||
print("📡 等待SUWood服务器数据...")
|
||
|
||
# 主循环
|
||
while True:
|
||
try:
|
||
user_input = input().strip().lower()
|
||
|
||
if user_input == 'q':
|
||
break
|
||
elif user_input == 's':
|
||
saver.manually_save()
|
||
elif user_input == 'c':
|
||
saver.clear_data()
|
||
else:
|
||
# 显示统计信息
|
||
stats = saver.get_statistics()
|
||
print()
|
||
print("📊 统计信息:")
|
||
print(f" 📥 收集记录: {stats['collected_records']} 条")
|
||
print(f" 📡 接收总数: {stats.get('total_received', 0)} 条")
|
||
print(f" 📁 文件大小: {stats['json_file_size_kb']:.1f} KB")
|
||
print(f" 💾 最后保存: {stats['last_save_time'] or '未保存'}")
|
||
print(
|
||
f" 🔗 连接状态: {'已连接' if stats.get('is_connected') else '未连接'}")
|
||
print()
|
||
|
||
except EOFError:
|
||
# Ctrl+D
|
||
break
|
||
|
||
except KeyboardInterrupt:
|
||
print("\n⛔ 收到中断信号...")
|
||
|
||
finally:
|
||
# 停止监控并保存数据
|
||
print("💾 正在保存最终数据...")
|
||
saver.stop_monitoring()
|
||
|
||
# 显示最终统计
|
||
final_stats = saver.get_statistics()
|
||
print()
|
||
print("📈 最终统计:")
|
||
print(f" 📥 总收集记录: {final_stats['collected_records']} 条")
|
||
print(f" 📁 JSON文件: {final_stats['json_file_path']}")
|
||
print(f" 💾 文件大小: {final_stats['json_file_size_kb']:.1f} KB")
|
||
print()
|
||
print("👋 程序结束，数据已保存到桌面001.json")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|