107 lines
4.1 KiB
Python
107 lines
4.1 KiB
Python
import asyncio
|
|
from typing import Dict, Optional
|
|
import time
|
|
from sqlalchemy import select, update, insert
|
|
from database.models import DeviceConfig
|
|
from services.database_service_base import DatabaseServiceBase
|
|
from utils.logger import session_logger
|
|
|
|
class DeviceVolumeManager(DatabaseServiceBase):
|
|
def __init__(self):
|
|
super().__init__(service_name="device_volume")
|
|
self.device_volumes: Dict[str, int] = {} # 缓存设备音量
|
|
self.lock = asyncio.Lock()
|
|
self._db_initialized = False
|
|
self.default_volume = 70 # 默认音量值
|
|
|
|
async def get_volume(self, device_id: str) -> int:
|
|
"""获取设备音量设置"""
|
|
await self._init_database()
|
|
|
|
if device_id in self.device_volumes:
|
|
return self.device_volumes[device_id]
|
|
|
|
db_session = await self.get_session()
|
|
try:
|
|
query = select(DeviceConfig).where(DeviceConfig.device_id == device_id)
|
|
result = await db_session.execute(query)
|
|
config = result.scalars().first()
|
|
|
|
if config and hasattr(config, "volume") and config.volume is not None:
|
|
volume = config.volume
|
|
else:
|
|
volume = self.default_volume
|
|
|
|
self.device_volumes[device_id] = volume
|
|
return volume
|
|
except Exception as e:
|
|
session_logger.error(device_id, "volume", f"获取音量信息失败: {str(e)}")
|
|
return self.default_volume # 出错时返回默认值
|
|
finally:
|
|
await db_session.close()
|
|
|
|
async def get_configured_volume(self, device_id: str) -> Optional[int]:
|
|
"""获取已持久化的期望音量；未配置时返回 None。"""
|
|
await self._init_database()
|
|
|
|
db_session = await self.get_session()
|
|
try:
|
|
query = select(DeviceConfig.volume).where(DeviceConfig.device_id == device_id)
|
|
result = await db_session.execute(query)
|
|
volume = result.scalar_one_or_none()
|
|
if volume is None:
|
|
return None
|
|
|
|
normalized_volume = int(volume)
|
|
self.device_volumes[device_id] = normalized_volume
|
|
return normalized_volume
|
|
except Exception as e:
|
|
session_logger.error(device_id, "volume", f"获取期望音量失败: {str(e)}")
|
|
raise
|
|
finally:
|
|
await db_session.close()
|
|
|
|
async def set_volume(self, device_id: str, volume: int) -> bool:
|
|
"""设置设备音量并保存到数据库"""
|
|
if volume < 0 or volume > 100:
|
|
raise ValueError("音量值必须在0-100范围内")
|
|
|
|
await self._init_database()
|
|
|
|
self.device_volumes[device_id] = volume
|
|
|
|
db_session = await self.get_session()
|
|
try:
|
|
query = select(DeviceConfig).where(DeviceConfig.device_id == device_id)
|
|
result = await db_session.execute(query)
|
|
config = result.scalars().first()
|
|
|
|
if config:
|
|
update_stmt = (
|
|
update(DeviceConfig)
|
|
.where(DeviceConfig.device_id == device_id)
|
|
.values(volume=volume, last_update_time=time.time())
|
|
)
|
|
await db_session.execute(update_stmt)
|
|
else:
|
|
from config import settings
|
|
insert_stmt = insert(DeviceConfig).values(
|
|
device_id=device_id,
|
|
selected_role_key=settings.selected_role_key,
|
|
volume=volume,
|
|
last_update_time=time.time()
|
|
)
|
|
await db_session.execute(insert_stmt)
|
|
|
|
await db_session.commit()
|
|
session_logger.info(device_id, "volume", f"设备音量已更新为 {volume}")
|
|
return True
|
|
except Exception as e:
|
|
await db_session.rollback()
|
|
session_logger.error(device_id, "volume", f"更新音量配置失败: {str(e)}")
|
|
raise
|
|
finally:
|
|
await db_session.close()
|
|
|
|
device_volume_manager = DeviceVolumeManager()
|