Python解析cyber record文件的示例详解

import argparse
import os
from cyber_py3 import record
from cyber_py3.record import RecordReader, RecordWriter, RecordMessage
 
def parse_record_file(file_path):
    """
    解析单个 Cyber Record 文件
    """
    try:
        reader = RecordReader(file_path)
        print(f"\n===== 正在解析文件: {os.path.basename(file_path)} =====")
        print(f"总消息数: {reader.get_messagenumber()}")
        print(f"开始时间: {reader.get_starttime()}")
        print(f"结束时间: {reader.get_endtime()}")
        print(f"通道列表: {reader.get_channellist()}\n")
 
        # 遍历所有消息
        for channel_name, msg, datatype, timestamp in reader.read_messages():
            print(f"[通道] {channel_name}")
            print(f"  时间戳: {timestamp}")
            print(f"  数据类型: {datatype}")
            print(f"  消息长度: {len(msg)} bytes")
            print("-" * 60)
 
    except Exception as e:
        print(f"解析文件 {file_path} 失败: {str(e)}")
 
def parse_record_directory(directory):
    """
    解析目录下的所有 Cyber Record 文件
    """
    if not os.path.isdir(directory):
        print(f"错误: {directory} 不是有效目录")
        return
 
    for filename in os.listdir(directory):
        file_path = os.path.join(directory, filename)
        
        # 通过文件头验证是否为合法 Record 文件
        if record.is_valid_record_file(file_path):
            parse_record_file(file_path)
        else:
            print(f"跳过非 Record 文件: {filename}")
 
if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Cyber Record 文件解析工具')
    parser.add_argument('path', type=str, help='文件路径或目录路径')
    args = parser.parse_args()
 
    target_path = args.path
 
    if os.path.isfile(target_path):
        if record.is_valid_record_file(target_path):
            parse_record_file(target_path)
        else:
            print(f"错误: {target_path} 不是有效的 Cyber Record 文件")
    elif os.path.isdir(target_path):
        parse_record_directory(target_path)
    else:
        print(f"错误: {target_path} 不存在")

# 解析单个文件
python parse_record.py /path/to/your.record
 
# 解析目录
python parse_record.py /path/to/record_dir/