39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-#
|
|
# -------------------------------------------------------------------------------
|
|
# Name: MediaMSG.py
|
|
# Description: 负责处理语音数据库
|
|
# Author: xaoyaoo
|
|
# Date: 2024/04/15
|
|
# -------------------------------------------------------------------------------
|
|
from .dbbase import DatabaseBase
|
|
from .utils import silk2audio
|
|
|
|
|
|
class MediaHandler(DatabaseBase):
|
|
_class_name = "MediaMSG"
|
|
Media_required_tables = ["Media"]
|
|
|
|
def Media_add_index(self):
|
|
"""
|
|
添加索引, 加快查询速度
|
|
"""
|
|
if self.tables_exist("Media"):
|
|
self.execute("CREATE INDEX IF NOT EXISTS MsgSvrID ON Media(Reserved0)")
|
|
|
|
def get_audio(self, MsgSvrID, is_play=False, is_wave=False, save_path=None, rate=24000):
|
|
if not self.tables_exist("Media"):
|
|
return False
|
|
|
|
sql = "select Buf from Media where Reserved0=? "
|
|
DBdata = self.execute(sql, (MsgSvrID,))
|
|
if not DBdata:
|
|
return False
|
|
if len(DBdata) == 0:
|
|
return False
|
|
data = DBdata[0][0] # [1:] + b'\xFF\xFF'
|
|
try:
|
|
pcm_data = silk2audio(buf_data=data, is_play=is_play, is_wave=is_wave, save_path=save_path, rate=rate)
|
|
return pcm_data
|
|
except Exception as e:
|
|
return False
|