46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import os
|
||
import sys
|
||
import subprocess
|
||
import logging
|
||
|
||
logger = logging.getLogger("Opus")
|
||
|
||
|
||
def setup_opus():
|
||
libs_dir = ""
|
||
lib_path = ""
|
||
|
||
logger.info("正在加载 Opus 动态库，请耐心等待...")
|
||
|
||
if sys.platform == "win32":
|
||
libs_dir = os.path.dirname(os.path.abspath(__file__))
|
||
lib_path = os.path.join(libs_dir, "opus.dll")
|
||
elif sys.platform == "darwin":
|
||
result = subprocess.check_output(
|
||
"brew list opus | grep libopus.dylib", shell=True
|
||
)
|
||
lib_path = result.decode("utf-8").strip()
|
||
if not lib_path.endswith("libopus.dylib"):
|
||
raise RuntimeError("请先安装 Opus： brew install opus")
|
||
libs_dir = os.path.dirname(lib_path)
|
||
else:
|
||
raise RuntimeError(f"暂不支持 {sys.platform} 平台")
|
||
|
||
import ctypes.util
|
||
|
||
original_find_library = ctypes.util.find_library
|
||
|
||
def patched_find_library(name):
|
||
if name == "opus":
|
||
return lib_path
|
||
return original_find_library(name)
|
||
|
||
ctypes.util.find_library = patched_find_library
|
||
|
||
if hasattr(os, "add_dll_directory"):
|
||
os.add_dll_directory(libs_dir)
|
||
|
||
os.environ["PATH"] = libs_dir + os.pathsep + os.environ.get("PATH", "")
|
||
|
||
ctypes.CDLL(lib_path)
|