add banbanmini backend

talkingq-url/database/connection.py

@@ -0,0 +1,71 @@
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
+from sqlalchemy.orm import sessionmaker
+from config import settings
+from utils.logger import session_logger
+import asyncio
+import urllib.parse
+
+class DatabaseManager:
+    _instance = None
+    _lock = asyncio.Lock()
+    
+    def __init__(self):
+        self.engine = None
+        self.async_session = None
+        self._initialized = False
+    
+    @classmethod
+    async def get_instance(cls):
+        async with cls._lock:
+            if cls._instance is None:
+                cls._instance = DatabaseManager()
+            
+            if not cls._instance._initialized:
+                await cls._instance.initialize()
+                
+            return cls._instance
+    
+    async def initialize(self):
+        if self._initialized:
+            return
+            
+        try:
+            encoded_password = urllib.parse.quote_plus(settings.db_password)
+            connection_string = f"mysql+aiomysql://{settings.db_user}:{encoded_password}@{settings.db_host}:{settings.db_port}/{settings.db_name}?charset=utf8mb4"
+            self.engine = create_async_engine(
+                connection_string,
+                echo=settings.db_echo,
+                pool_pre_ping=True,
+                pool_recycle=3600,
+            )
+            
+            self.async_session = sessionmaker(
+                self.engine, expire_on_commit=False, class_=AsyncSession
+            )
+            
+            self._initialized = True
+            session_logger.info("system", "database", f"数据库连接初始化成功: {settings.db_host}:{settings.db_port}/{settings.db_name}")
+        except Exception as e:
+            session_logger.error("system", "database", f"数据库连接初始化失败: {str(e)}")
+            raise
+    
+    async def get_session(self):
+        """获取数据库会话"""
+        if not self._initialized:
+            await self.initialize()
+        return self.async_session()
+    
+    async def close(self):
+        """关闭数据库连接"""
+        if self.engine:
+            await self.engine.dispose()
+            self._initialized = False
+            session_logger.info("system", "database", "数据库连接已关闭")
+
+db_manager = None
+
+async def get_db_manager():
+    global db_manager
+    if db_manager is None:
+        db_manager = await DatabaseManager.get_instance()
+    return db_manager

talkingq-url/database/init_db.py

@@ -0,0 +1,24 @@
+import asyncio
+from sqlalchemy.ext.asyncio import create_async_engine
+from database.models import Base
+from config import settings
+from utils.logger import session_logger
+import urllib.parse
+
+async def init_db():
+    """初始化数据库，创建所有表"""
+    try:
+        encoded_password = urllib.parse.quote_plus(settings.db_password)
+        connection_string = f"mysql+aiomysql://{settings.db_user}:{encoded_password}@{settings.db_host}:{settings.db_port}/{settings.db_name}"
+        engine = create_async_engine(
+            connection_string,
+            echo=settings.db_echo
+        )
+        async with engine.begin() as conn:
+            await conn.run_sync(Base.metadata.create_all)
+        await engine.dispose()
+        session_logger.info("system", "database", "数据库表已成功创建")
+        return True
+    except Exception as e:
+        session_logger.error("system", "database", f"创建数据库表失败: {str(e)}")
+        return False

talkingq-url/database/models.py

@@ -0,0 +1,128 @@
+from sqlalchemy import Column, String, Text, Float, DateTime, Integer, Boolean, func, ForeignKey, JSON
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.schema import UniqueConstraint
+import datetime
+
+Base = declarative_base()
+
+class DeviceConfig(Base):
+    __tablename__ = "device_configs"
+    
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    device_id = Column(String(64), unique=True, index=True, nullable=False)
+    selected_role_key = Column(String(64), nullable=False)
+    preferred_language = Column(String(10), nullable=True)
+    volume = Column(Integer, nullable=True)
+    last_update_time = Column(Float, nullable=False, default=lambda: datetime.datetime.now().timestamp())
+    created_at = Column(DateTime, nullable=False, server_default=func.now())
+    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
+    
+    def to_dict(self):
+        """将配置转换为可序列化的字典"""
+        return {
+            "selected_role_key": self.selected_role_key,
+            "preferred_language": self.preferred_language,
+            "volume": self.volume if hasattr(self, 'volume') else None,
+        }
+
+class ConversationHistory(Base):
+    __tablename__ = "conversation_histories"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    device_id = Column(String(64), nullable=False, index=True)
+    role_key = Column(String(64), nullable=False, index=True)
+    last_interaction_time = Column(Float, nullable=False, default=lambda: datetime.datetime.now().timestamp())
+    created_at = Column(DateTime, nullable=False, server_default=func.now())
+    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
+    __table_args__ = (
+        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
+    )
+
+class ConversationMessage(Base):
+    __tablename__ = "conversation_messages"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    conversation_id = Column(Integer, ForeignKey("conversation_histories.id", ondelete="CASCADE"), nullable=False, index=True)
+    is_user = Column(Boolean, nullable=False, default=False)  # True表示用户消息，False表示助手消息
+    content = Column(Text, nullable=False)
+    timestamp = Column(Float, nullable=False, default=lambda: datetime.datetime.now().timestamp())
+    created_at = Column(DateTime, nullable=False, server_default=func.now())
+    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
+    __table_args__ = (
+        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
+    )
+
+class Role(Base):
+    __tablename__ = "roles"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    role_key = Column(String(64), unique=True, nullable=False, index=True)
+    name = Column(String(128), nullable=False)
+    description = Column(Text, nullable=True)
+    content = Column(Text, nullable=False)
+    default_language = Column(String(10), nullable=True)
+    volcano_model_id = Column(String(64), nullable=True)
+    minimax_voice_id = Column(String(64), nullable=True)
+    url = Column(String(255), nullable=True)
+    homophones = Column(JSON, nullable=True)
+    enabled = Column(Boolean, default=True, nullable=False)
+    created_at = Column(DateTime, nullable=False, server_default=func.now())
+    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
+    
+    __table_args__ = (
+        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
+    )
+
+class RoleLanguage(Base):
+    __tablename__ = "role_languages"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    role_id = Column(Integer, ForeignKey("roles.id", ondelete="CASCADE"), nullable=False)
+    language_code = Column(String(10), nullable=False)
+    name = Column(String(128), nullable=True)
+    content = Column(Text, nullable=True)
+    minimax_voice_id = Column(String(64), nullable=True)
+    url = Column(String(255), nullable=True)
+    created_at = Column(DateTime, nullable=False, server_default=func.now())
+    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
+    
+    __table_args__ = (
+        UniqueConstraint('role_id', 'language_code', name='uix_role_language'),
+        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
+    )
+
+class DeviceAuth(Base):
+    __tablename__ = "device_auth"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    device_id = Column(String(64), nullable=False, unique=True, index=True)
+    serial_number = Column(String(64), nullable=False)
+    batch_id = Column(String(20), nullable=True, index=True)
+    is_active = Column(Boolean, default=True, nullable=False)
+    created_at = Column(DateTime, nullable=False, server_default=func.now())
+    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
+    
+    __table_args__ = (
+        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
+    )
+
+class DeviceFirmwareUpdate(Base):
+    __tablename__ = "device_firmware_update"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    device_id = Column(String(64), nullable=False, unique=True, index=True)
+    firmware_version = Column(String(64), nullable=False)
+    update_status = Column(String(32), nullable=False, default="success")     # 更新状态，如 updating/success/failed
+    created_at = Column(DateTime, nullable=False, server_default=func.now())
+    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
+    progress = Column(Float, nullable=True, default=0.0)  # 更新进度
+
+    __table_args__ = (
+        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
+    )
+
+class SystemConfig(Base):
+    __tablename__ = "system_config"
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    config_key = Column(String(128), unique=True, nullable=False, index=True)
+    config_value = Column(Text, nullable=True)
+    created_at = Column(DateTime, nullable=False, server_default=func.now())
+    updated_at = Column(DateTime, nullable=False, server_default=func.now(), onupdate=func.now())
+
+    __table_args__ = (
+        {'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
+    )