feat: 开放小爱音箱接入小智 AI 演示源代码
This commit is contained in:
Del Wang
@@ -0,0 +1,256 @@
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Any, Optional
|
||||
import threading
|
||||
import requests
|
||||
import socket
|
||||
import uuid
|
||||
|
||||
logger = logging.getLogger("ConfigManager")
|
||||
|
||||
|
||||
class ConfigManager:
|
||||
"""配置管理器 - 单例模式"""
|
||||
|
||||
_instance = None
|
||||
_lock = threading.Lock()
|
||||
CONFIG_FILE = Path(os.getcwd()) / "xiaozhi.json"
|
||||
|
||||
# 默认配置
|
||||
DEFAULT_CONFIG = {
|
||||
"CLIENT_ID": None,
|
||||
"DEVICE_ID": None,
|
||||
"NETWORK": {
|
||||
"OTA_VERSION_URL": "https://api.tenclass.net/xiaozhi/ota/",
|
||||
"WEBSOCKET_URL": "wss://api.tenclass.net/xiaozhi/v1/",
|
||||
"WEBSOCKET_ACCESS_TOKEN": "test-token",
|
||||
},
|
||||
"MQTT_INFO": None,
|
||||
}
|
||||
|
||||
def __new__(cls):
|
||||
"""确保单例模式"""
|
||||
if cls._instance is None:
|
||||
cls._instance = super().__new__(cls)
|
||||
return cls._instance
|
||||
|
||||
def __init__(self):
|
||||
"""初始化配置管理器"""
|
||||
self.logger = logger
|
||||
if hasattr(self, "_initialized"):
|
||||
return
|
||||
self._initialized = True
|
||||
|
||||
# 加载配置
|
||||
self._config = self._load_config()
|
||||
self._initialize_client_id()
|
||||
self._initialize_device_id()
|
||||
self._initialize_mqtt_info()
|
||||
|
||||
def _load_config(self) -> Dict[str, Any]:
|
||||
"""加载配置文件，如果不存在则创建"""
|
||||
try:
|
||||
if self.CONFIG_FILE.exists():
|
||||
config = json.loads(self.CONFIG_FILE.read_text(encoding="utf-8"))
|
||||
return self._merge_configs(self.DEFAULT_CONFIG, config)
|
||||
else:
|
||||
self._save_config(self.DEFAULT_CONFIG)
|
||||
return self.DEFAULT_CONFIG.copy()
|
||||
except Exception as e:
|
||||
logger.error(f"Error loading config: {e}")
|
||||
return self.DEFAULT_CONFIG.copy()
|
||||
|
||||
def _save_config(self, config: dict) -> bool:
|
||||
"""保存配置到文件"""
|
||||
try:
|
||||
self.CONFIG_FILE.write_text(
|
||||
json.dumps(config, indent=2, ensure_ascii=False), encoding="utf-8"
|
||||
)
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"Error saving config: {e}")
|
||||
return False
|
||||
|
||||
@staticmethod
|
||||
def _merge_configs(default: dict, custom: dict) -> dict:
|
||||
"""递归合并配置字典"""
|
||||
result = default.copy()
|
||||
for key, value in custom.items():
|
||||
if (
|
||||
key in result
|
||||
and isinstance(result[key], dict)
|
||||
and isinstance(value, dict)
|
||||
):
|
||||
result[key] = ConfigManager._merge_configs(result[key], value)
|
||||
else:
|
||||
result[key] = value
|
||||
return result
|
||||
|
||||
def get_client_id(self) -> str:
|
||||
"""获取客户端ID"""
|
||||
return self._config["CLIENT_ID"]
|
||||
|
||||
def get_device_id(self) -> Optional[str]:
|
||||
"""获取设备ID"""
|
||||
return self._config.get("DEVICE_ID")
|
||||
|
||||
def get_network_config(self) -> dict:
|
||||
"""获取网络配置"""
|
||||
return self._config["NETWORK"]
|
||||
|
||||
def get_config(self, path: str, default: Any = None) -> Any:
|
||||
"""
|
||||
通过路径获取配置值
|
||||
"""
|
||||
try:
|
||||
value = self._config
|
||||
for key in path.split("."):
|
||||
value = value[key]
|
||||
return value
|
||||
except (KeyError, TypeError):
|
||||
return default
|
||||
|
||||
def update_config(self, path: str, value: Any) -> bool:
|
||||
"""
|
||||
更新特定配置项
|
||||
"""
|
||||
try:
|
||||
current = self._config
|
||||
*parts, last = path.split(".")
|
||||
for part in parts:
|
||||
current = current.setdefault(part, {})
|
||||
current[last] = value
|
||||
return self._save_config(self._config)
|
||||
except Exception as e:
|
||||
logger.error(f"Error updating config {path}: {e}")
|
||||
return False
|
||||
|
||||
@classmethod
|
||||
def instance(cls):
|
||||
"""获取配置管理器实例（线程安全）"""
|
||||
with cls._lock:
|
||||
if cls._instance is None:
|
||||
cls._instance = cls()
|
||||
return cls._instance
|
||||
|
||||
def get_mac_address(self):
|
||||
mac = uuid.UUID(int=uuid.getnode()).hex[-12:]
|
||||
return ":".join([mac[i : i + 2] for i in range(0, 12, 2)])
|
||||
|
||||
def generate_uuid(self) -> str:
|
||||
return str(uuid.uuid4())
|
||||
|
||||
def get_local_ip(self):
|
||||
try:
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
s.connect(("8.8.8.8", 80))
|
||||
ip = s.getsockname()[0]
|
||||
s.close()
|
||||
return ip
|
||||
except Exception:
|
||||
return "127.0.0.1"
|
||||
|
||||
def _initialize_client_id(self):
|
||||
"""确保存在客户端ID"""
|
||||
if not self._config["CLIENT_ID"]:
|
||||
client_id = self.generate_uuid()
|
||||
success = self.update_config("CLIENT_ID", client_id)
|
||||
if success:
|
||||
logger.info(f"Generated new CLIENT_ID: {client_id}")
|
||||
else:
|
||||
logger.error("Failed to save new CLIENT_ID")
|
||||
|
||||
def _initialize_device_id(self):
|
||||
"""确保存在设备ID"""
|
||||
if not self._config["DEVICE_ID"]:
|
||||
try:
|
||||
device_hash = self.get_mac_address()
|
||||
success = self.update_config("DEVICE_ID", device_hash)
|
||||
if success:
|
||||
logger.info(f"Generated new DEVICE_ID: {device_hash}")
|
||||
else:
|
||||
logger.error("Failed to save new DEVICE_ID")
|
||||
except Exception as e:
|
||||
logger.error(f"Error generating DEVICE_ID: {e}")
|
||||
|
||||
def _initialize_mqtt_info(self):
|
||||
try:
|
||||
mqtt_info = self._get_ota_version()
|
||||
if mqtt_info:
|
||||
self.update_config("MQTT_INFO", mqtt_info)
|
||||
self.logger.info("MQTT信息已成功更新")
|
||||
return mqtt_info
|
||||
else:
|
||||
self.logger.warning("获取MQTT信息失败，使用已保存的配置")
|
||||
return self.get_config("MQTT_INFO")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"初始化MQTT信息失败: {e}")
|
||||
return self.get_config("MQTT_INFO")
|
||||
|
||||
def _get_ota_version(self):
|
||||
"""获取OTA服务器的MQTT信息"""
|
||||
MAC_ADDR = self.get_device_id()
|
||||
OTA_VERSION_URL = self.get_config("NETWORK.OTA_VERSION_URL")
|
||||
headers = {"Device-Id": MAC_ADDR, "Content-Type": "application/json"}
|
||||
|
||||
# 构建设备信息payload
|
||||
payload = {
|
||||
"flash_size": 16777216, # 闪存大小 (16MB)
|
||||
"minimum_free_heap_size": 8318916, # 最小可用堆内存
|
||||
"mac_address": MAC_ADDR, # 设备MAC地址
|
||||
"chip_model_name": "esp32s3", # 芯片型号
|
||||
"chip_info": {"model": 9, "cores": 2, "revision": 2, "features": 18},
|
||||
"application": {
|
||||
"name": "xiaozhi",
|
||||
"version": "1.1.2",
|
||||
"idf_version": "v5.3.2-dirty",
|
||||
},
|
||||
"partition_table": [],
|
||||
"ota": {"label": "factory"},
|
||||
"board": {
|
||||
"type": "bread-compact-wifi",
|
||||
"ip": self.get_local_ip(),
|
||||
"mac": MAC_ADDR,
|
||||
},
|
||||
}
|
||||
|
||||
try:
|
||||
# 发送请求到OTA服务器
|
||||
response = requests.post(
|
||||
OTA_VERSION_URL,
|
||||
headers=headers,
|
||||
json=payload,
|
||||
timeout=10,
|
||||
)
|
||||
|
||||
# 检查HTTP状态码
|
||||
if response.status_code != 200:
|
||||
self.logger.error(f"OTA服务器错误: HTTP {response.status_code}")
|
||||
raise ValueError(f"OTA服务器返回错误状态码: {response.status_code}")
|
||||
|
||||
# 解析JSON数据
|
||||
response_data = response.json()
|
||||
|
||||
# 调试信息：打印完整的OTA响应
|
||||
self.logger.debug(
|
||||
f"OTA服务器返回数据: {json.dumps(response_data, indent=4, ensure_ascii=False)}"
|
||||
)
|
||||
|
||||
# 确保"mqtt"信息存在
|
||||
if "mqtt" in response_data:
|
||||
self.logger.info(f"MQTT服务器信息已更新")
|
||||
return response_data["mqtt"]
|
||||
else:
|
||||
self.logger.error("OTA服务器返回的数据无效: MQTT信息缺失")
|
||||
raise ValueError("OTA服务器返回的数据无效，请检查服务器状态或MAC地址！")
|
||||
|
||||
except requests.Timeout:
|
||||
self.logger.error("OTA请求超时，请检查网络或服务器状态")
|
||||
raise ValueError("OTA请求超时！请稍后重试。")
|
||||
|
||||
except requests.RequestException as e:
|
||||
self.logger.error(f"OTA请求失败: {e}")
|
||||
raise ValueError("无法连接到OTA服务器，请检查网络连接！")
|
||||
@@ -0,0 +1,30 @@
|
||||
import logging
|
||||
|
||||
|
||||
def setup_logging():
|
||||
"""配置日志系统"""
|
||||
|
||||
# 创建根日志记录器
|
||||
root_logger = logging.getLogger()
|
||||
root_logger.setLevel(logging.INFO) # 设置根日志级别
|
||||
|
||||
# 清除已有的处理器（避免重复添加）
|
||||
if root_logger.handlers:
|
||||
root_logger.handlers.clear()
|
||||
|
||||
# 创建控制台处理器
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setLevel(logging.INFO)
|
||||
|
||||
# 创建格式化器
|
||||
formatter = logging.Formatter(
|
||||
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||||
)
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# 添加处理器到根日志记录器
|
||||
root_logger.addHandler(console_handler)
|
||||
|
||||
# 设置特定模块的日志级别
|
||||
logging.getLogger("XiaoZhi").setLevel(logging.INFO)
|
||||
logging.getLogger("WebsocketProtocol").setLevel(logging.INFO)
|
||||
Reference in New Issue
Block a user