如何用Python 3.x抓取WebSocket流媒体？解决Base64解码乱码问题

核心问题分析 #

你拿到的Base64编码内容是二进制流媒体数据，不是文本，用UTF-8解码肯定会乱码——UTF-8是文本编码，用来解码二进制流（比如视频帧、压缩后的媒体包）完全不适用。

具体实现步骤 #

• 第一步：正确解码Base64到二进制数据
  用Python的base64模块解码成bytes对象，而非直接转字符串：

  import base64
  
  # 假设ws_recv_data是你从WebSocket收到的Base64字符串
  binary_data = base64.b64decode(ws_recv_data)
  

• 第二步：识别流媒体协议
  气象服务商的视频流大概率采用标准流媒体协议，你需要先分析二进制数据的头部特征：

  • 若开头是FLV字节（十六进制46 4C 56），即为FLV格式，可用flvlib或ffmpeg-python处理
  • 若为H.264的NALU单元（开头是00 00 00 01或00 00 01），可使用opencv-python或av库解码帧
  • 若为压缩后的媒体包，可尝试用zlib.decompress()解包

• 第三步：选择合适的库处理流数据

  • 轻量级处理：用av（PyAV）——FFmpeg的Python绑定，支持几乎所有流媒体格式，可直接处理二进制流：

    import av
    
    container = av.open(binary_data, format='flv')  # 根据实际格式调整
    for frame in container.decode(video=0):
        # 处理每一帧，比如转成numpy数组保存或展示
        img = frame.to_ndarray(format='bgr24')
        # 可添加保存到文件或显示的逻辑
    

  • 实时推流/保存：用ffmpeg-python将二进制流导入FFmpeg处理，比如保存为MP4：

    import ffmpeg
    
    process = (
        ffmpeg
        .input('pipe:', format='flv')  # 对应实际格式
        .output('output.mp4', vcodec='copy')
        .run_async(pipe_stdin=True)
    )
    process.stdin.write(binary_data)
    process.stdin.close()
    process.wait()
    

• 第四步：验证WebSocket消息结构
  部分服务商的WebSocket消息会带头部（比如消息类型、长度标识），需先剥离头部再处理二进制数据，可用struct模块解析：

  import struct
  
  # 假设头部是4字节的长度字段
  header = binary_data[:4]
  data_length = struct.unpack('!I', header)[0]
  real_media_data = binary_data[4:4+data_length]
  

关键注意事项 #

• 禁止用文本编码（UTF-8、GBK等）处理二进制媒体数据，必须保持bytes对象操作
• 优先选用FFmpeg生态的库（PyAV、ffmpeg-python），兼容性最强，无需自行实现解码逻辑
• 若不确定流媒体格式，可通过ffmpeg -i -命令将二进制数据通过stdin传入，让FFmpeg自动识别

内容的提问来源于stack exchange，提问作者Lucas016